Fix click implementation in test client

1 files changed, 4 insertions, 4 deletions

diff --git a/test_client.py b/test_client.py
index deb5252..ee49bae 100755
--- a/test_client.py
+++ b/test_client.py
@@ -39,7 +39,7 @@ 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}")
@@ -48,7 +48,7 @@ def start(host: int):
 
 
 @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}")