Cleaned up nmapitest.py

[tests.git] / qaapi / qa / tests / slice / nmapitest.py

diff --git a/qaapi/qa/tests/slice/nmapitest.py b/qaapi/qa/tests/slice/nmapitest.py
index f934eb1..4cad38e 100755 (executable)
--- a/qaapi/qa/tests/slice/nmapitest.py
+++ b/qaapi/qa/tests/slice/nmapitest.py
@@ -4,7 +4,15 @@
 # Connect to the node manager
 # Author: acb@cs.princeton.edu
 
+import sys
+
 from xmlrpclib import ServerProxy
 
-nodemanager = ServerProxy('http://127.0.0.1:812/')
-nodemanager.SetLoans('pl_sirius', [])
+if (len(sys.argv)!=2):
+    print "[FAILED] Please pass the name of the slice this script runs in as the first argument.\n";
+else:
+    nodemanager = ServerProxy('http://127.0.0.1:812/')
+    try:
+        nodemanager.SetLoans(sys.argv[1], [])
+    except:
+        print "[FAILED] SetLoans didn't work, probably because the packets of this connection are not getting tagged.\n";