diff --git a/product/TIDStorage/tests/testTIDServer.py b/product/TIDStorage/tests/testTIDServer.py index ef4d15db032317cd57e7759ea44a9ea64e64d892..b3e42ea11c56a46a580cea792ea90e55798a88c8 100755 --- a/product/TIDStorage/tests/testTIDServer.py +++ b/product/TIDStorage/tests/testTIDServer.py @@ -6,11 +6,6 @@ import socket import time import sys -assert len(sys.argv) == 3, 'Requires exactly 2 arguments: <address> <port>' - -address = sys.argv[1] -port = int(sys.argv[2]) - class TIDClient: def __init__(self, address): self._to_server = socket.socket() @@ -283,6 +278,12 @@ class TestTIDServerV2: sys.stdout.write(message) if __name__ == '__main__': + + assert len(sys.argv) == 3, 'Requires exactly 2 arguments: <address> <port>' + + address = sys.argv[1] + port = int(sys.argv[2]) + test = TestTIDServerV2(address, port) test.run()