X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=system%2FTestSite.py;h=f4d6cf296777aeb4d03f4fe83d6131dfbd96ceeb;hb=c8e2f4e3a327181b29521583770a6f0ff68ca6eb;hp=324861eb41e7742f5d6acf8cbf00573613e28763;hpb=65c93a296f2e481a312e7846f518eb2eaa2fc080;p=tests.git diff --git a/system/TestSite.py b/system/TestSite.py index 324861e..f4d6cf2 100644 --- a/system/TestSite.py +++ b/system/TestSite.py @@ -13,9 +13,9 @@ from TestKey import TestKey class TestSite: - def __init__ (self,test_plc,site_spec): - self.test_plc=test_plc - self.site_spec=site_spec + def __init__ (self, test_plc, site_spec): + self.test_plc = test_plc + self.site_spec = site_spec def name(self): return self.site_spec['site_fields']['login_base'] @@ -24,38 +24,39 @@ class TestSite: print self.test_plc.auth_root() self.test_plc.apiserver.AddSite(self.test_plc.auth_root(), self.site_spec['site_fields']) - self.test_plc.apiserver.AddSiteAddress(self.test_plc.auth_root(),self.name(), + self.test_plc.apiserver.AddSiteAddress(self.test_plc.auth_root(), + self.name(), self.site_spec['address_fields']) def create_users (self): for user_spec in self.site_spec['users']: - test_user=TestUser(self.test_plc,self,user_spec) + test_user = TestUser(self.test_plc, self, user_spec) test_user.create_user() test_user.add_keys() def delete_site (self): print self.test_plc.auth_root() - self.test_plc.apiserver.DeleteSite(self.test_plc.auth_root(),self.name()) + self.test_plc.apiserver.DeleteSite(self.test_plc.auth_root(), self.name()) return True def delete_users(self): for user_spec in self.site_spec['users']: - test_user=TestUser(self.test_plc,self,user_spec) + test_user = TestUser(self.test_plc, self, user_spec) test_user.delete_user() - def locate_user (self,username): + def locate_user (self, username): for user in self.site_spec['users']: if user['name'] == username: return user if user['user_fields']['email'] == username: return user - raise Exception,"Cannot locate user %s"%username + raise Exception,"Cannot locate user {}".format(username) - def locate_node (self,nodename): + def locate_node (self, nodename): for node in self.site_spec['nodes']: if node['name'] == nodename: return node - raise Exception,"Cannot locate node %s"%nodename + raise Exception,"Cannot locate node {}".format(nodename)