From f32295351f7c2b732dbfb53f7a4d2d1a32b0c7bf Mon Sep 17 00:00:00 2001
From: Tony Mack <tmack@cs.princeton.edu>
Date: Fri, 18 Jan 2008 22:09:36 +0000
Subject: [PATCH] Repeatedly attempt to use the specified users credentials to
access the spefified slice on the specified node
---
qaapi/qa/modules/plc/access_slice.py | 68 ++++++++++++++++++++++++++++
1 file changed, 68 insertions(+)
create mode 100644 qaapi/qa/modules/plc/access_slice.py
diff --git a/qaapi/qa/modules/plc/access_slice.py b/qaapi/qa/modules/plc/access_slice.py
new file mode 100644
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
--
2.47.0