Merge branch 'feature/multiple-hosts' into 'master'

Support multiple hosts

See merge request remotewebapp/rwa.support.sessionservice!7

1 files changed, 5 insertions, 5 deletions

diff --git a/test_client.py b/test_client.py
index 9e2f15c..ee49bae 100755
--- a/test_client.py
+++ b/test_client.py
@@ -39,16 +39,16 @@ def cli():
 
 
 @cli.command()
-@click.argument("host")
+@click.argument("host", type=int)
 def start(host: int):
     """Start a session on the RWA.Support.WebApp host with index HOST."""
     click.echo(f"Sending D-Bus request 'start': {host}")
-    response = req.start()
+    response = req.start(host)
     click.echo(f"Your response is: {response}")
 
 
 @cli.command()
-@click.argument("pid")
+@click.argument("pid", type=int)
 def stop(pid: int):
     """Stop the session with the pid PID."""
     click.echo(f"Sending D-Bus request 'stop' with PID {pid}")
@@ -57,7 +57,7 @@ def stop(pid: int):
 
 
 @cli.command()
-@click.argument("pid")
+@click.argument("pid", type=int)
 def status(pid: int):
     """Get the status of the session with the pid PID."""
     click.echo(f"Sending D-Bus request 'status' with PID {pid}")
@@ -66,7 +66,7 @@ def status(pid: int):
 
 
 @cli.command()
-@click.argument("pid")
+@click.argument("pid", type=int)
 def refresh_status(pid: int):
     """Refresh and get the status of the session with the pid PID."""
     click.echo(f"Sending D-Bus request 'refresh_status' with PID {pid}")