Repeatedly attempt to use the specified users credentials to access the spefified...

diff --git a/qaapi/qa/modules/plc/access_slice.py b/qaapi/qa/modules/plc/access_slice.py
new file mode 100644 (file)
index 0000000..d1d7fec
--- /dev/null
+++ b/qaapi/qa/modules/plc/access_slice.py
@@ -0,0 +1,68 @@
+import traceback
+import time
+from qa.Test import Test
+from qa import utils
+
+class access_slice(Test):
+    """
+    Repeatedly attempt to use the specified users credentials to 
+    access the specified node on the specified slice.    
+    """        
+
+    def call(self, email, slice_name, hostname, timeout=3):
+       api = self.config.api
+       auth = self.config.auth
+       email_parts = email.split("@")
+        keys_filename = email_parts[0]
+        keys_path = self.config.KEYS_PATH
+        private_key_path = keys_path + os.sep + keys_filename
+        public_key_path = private_key_path + ".pub"
+
+       # Validate slice
+       slices = api.GetSlices(auth, [slice_name], ['name', 'slice_id', 'node_ids'])
+       if not slices:
+           raise Exception, "No such slice %(slice_name)s" % locals()
+       slice = slices[0]
+
+       # Validate node
+       nodes = api.GetNodes(auth, [hostname], ['hostname', 'node_id', 'slice_ids'])
+       if not nodes:
+           raise Exception, "No such node %(hostname)s" % locals()
+       node = nodes[0]
+       if slice['slice_id'] not in node['slice_ids']:
+           raise Exception, "%(slice_name)s not on %(hostname)s" % locals()  
+
+       # Validate user
+       persons = api.GetPersons(auth, ['email'], ['person_id', 'key_ids', 'slice_ids'])
+       if not persons:
+           raise Exception, "No such person %(email)s" % locals()
+       person = persons[0]
+       if slice['slice_id'] not in person['slice_ids']:
+           raise Exception, "%(email)s not in slice %(slice_name)s" % locals()
+
+       # get keys
+       if not os.path.isfile(private_key_path) or \
+           not os.path.isfile(public_key_path):
+           # keys dont exist, call api.sync_user_key()
+           from qa.modules.api.sync_user_key import sync_user_key
+           sync_user_key()(email)
+
+       # attempt to access slice
+       start_time = time.time()
+       end_time = start_time + timeout*60
+       sleep = 30
+       while time.time() <  endtime:
+           if self.config.verbose:
+               utils.header("Trying to connect to %(slice_name)s@%(hostname)s" % locals())
+           ssh_command = "ssh -i %(private_key_path)s %(slice_name)s@%(hostname)s" % locals()     
+           host_check = os.system(ssh_command + " hostname ")
+           if host_check == 0:
+               if self.config.verbose:
+                   utils.header("connecteed to %(slice_name)s@%(hostname)s" % locals()
+               return 1
+           else:
+               if self.config.verbose:
+                   utils.header("failed to connect to %(slice_name)s@%(hostname)s" % locals())         
+           time.sleep(sleep)           
+               
+       return 0