updated traceroutes to install tcptraceroute if it doesn't exist

[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 3c5df41..023629b 100755 (executable)
--- a/qaapi/qa/tests/slice/nmapitest.py
+++ b/qaapi/qa/tests/slice/nmapitest.py
@@ -5,14 +5,15 @@
 # Author: acb@cs.princeton.edu/sapanb@cs.princeton.edu
 
 import sys
+import os
 
 from xmlrpclib import ServerProxy
 
-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";
+homedir=os.environ['HOME']
+slice_name = homedir.rsplit('/')[0]
+
+nodemanager = ServerProxy('http://127.0.0.1:812/')
+try:
+    nodemanager.SetLoans(slice_name, [])
+except:
+    print "[FAILED] SetLoans didn't work, probably because the packets of this connection are not getting tagged.\n";