# Copyright (C) 2010 INRIA
#
import os, sys, time
-import xmlrpclib
+import xmlrpc.client
import utils
class TestUser:
- def __init__ (self,test_plc,test_site,user_spec):
- self.test_plc=test_plc
- self.test_site=test_site
- self.user_spec=user_spec
+ def __init__ (self, test_plc, test_site, user_spec):
+ self.test_plc = test_plc
+ self.test_site = test_site
+ self.user_spec = user_spec
def name(self):
return self.user_spec['user_fields']['email']
def auth (self):
- person=self.user_spec['user_fields']
- return {'Username':person['email'],
- 'AuthMethod':'password',
- 'AuthString':person['password'],
- 'Role':self.user_spec['roles'][0],
+ person = self.user_spec['user_fields']
+ return {'Username' : person['email'],
+ 'AuthMethod' : 'password',
+ 'AuthString' : person['password'],
+ 'Role' : self.user_spec['roles'][0],
}
def create_user (self):
- user_spec=self.user_spec
- fields=user_spec['user_fields']
- auth=self.test_plc.auth_root()
- utils.header('Adding user %s - roles %r'%(fields['email'],user_spec['roles']))
- self.test_plc.apiserver.AddPerson(auth,fields)
- self.test_plc.apiserver.UpdatePerson(auth,fields['email'],{'enabled': True})
+ user_spec = self.user_spec
+ fields = user_spec['user_fields']
+ auth = self.test_plc.auth_root()
+ utils.header('Adding user {} - roles {}'.format(fields['email'], user_spec['roles']))
+ self.test_plc.apiserver.AddPerson(auth, fields)
+ self.test_plc.apiserver.UpdatePerson(auth, fields['email'], {'enabled': True})
for role in user_spec['roles']:
self.test_plc.apiserver.AddRoleToPerson(auth,role,fields['email'])
self.test_plc.apiserver.AddPersonToSite(auth,
- self.name(),
- self.test_site.name())
+ self.name(),
+ self.test_site.name())
def delete_user(self):
- auth=self.test_plc.auth_root()
+ auth = self.test_plc.auth_root()
self.test_plc.apiserver.DeletePerson(auth,self.name())
def add_keys (self):
- user_spec=self.user_spec
- for keyname in user_spec['keynames']:
- key=self.test_plc.locate_key(keyname)
- auth=self.auth()
- self.test_plc.apiserver.AddPersonKey(auth,self.name(), key['key_fields'])
-
-
-
+ user_spec = self.user_spec
+ for key_name in user_spec['key_names']:
+ key_spec = self.test_plc.locate_key(key_name)
+ auth = self.auth()
+ self.test_plc.apiserver.AddPersonKey(auth,self.name(), key_spec['key_fields'])