Test framework reworked for

[tests.git] / system / TestUser.py

diff --git a/system/TestUser.py b/system/TestUser.py
new file mode 100644 (file)
index 0000000..6646e04
--- /dev/null
+++ b/system/TestUser.py
@@ -0,0 +1,50 @@
+import os, sys, time
+import xmlrpclib
+import pprint
+
+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 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],
+                }
+
+    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.server.AddPerson(auth,fields)
+        self.test_plc.server.UpdatePerson(auth,fields['email'],{'enabled': True})
+        for role in user_spec['roles']:
+            self.test_plc.server.AddRoleToPerson(auth,role,fields['email'])
+        self.test_plc.server.AddPersonToSite(auth,
+                                             self.name(),
+                                             self.test_site.name())
+
+    def delete_user(self):
+        auth=self.test_plc.auth_root()
+        self.test_plc.server.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.server.AddPersonKey(auth,self.name(), key['key_fields'])
+            
+
+