def create_user(cls, name="guest", role="guest", group="group", password="******", accessible_datasets=None, process_num_limit=1000, process_time_limit=6000): auth = bytes('%s:%s' % (name, password), "utf-8") # We need to create an HTML basic authorization header cls.auth_header[role] = Headers() cls.auth_header[role].add('Authorization', 'Basic ' + base64.b64encode(auth).decode()) # Make sure the user database is empty user = ActiniaUser(name) if user.exists(): user.delete() # Create a user in the database user = ActiniaUser.create_user(name, group, password, user_role=role, accessible_datasets=accessible_datasets, process_num_limit=process_num_limit, process_time_limit=process_time_limit) user.add_accessible_modules(["uname", "sleep"]) cls.users_list.append(user) return name, group, cls.auth_header[role]
def test_create_update_user(self): """Test the creation and update of a user in the redis database """ user = ActiniaUser(self.user_id) if user.exists(): print("Delete existing user") user.delete() # Create a new user user = ActiniaUser.create_user(self.user_id, self.user_group, self.password, user_role="admin", accessible_datasets={"nc_spm_08": ["PERMANENT", "user1"]}, accessible_modules=["g.region", "g.mapset", "r.slope.aspect"], cell_limit=1000, process_num_limit=3, process_time_limit=30) print(user) self.assertEqual(user.get_role(), "admin") self.assertEqual(user.get_id(), self.user_id) self.assertEqual(user.get_group(), self.user_group) self.assertEqual(user.get_cell_limit(), 1000) self.assertEqual(user.get_process_num_limit(), 3) self.assertEqual(user.get_process_time_limit(), 30) datasets = user.get_accessible_datasets() self.assertTrue("nc_spm_08" in datasets) self.assertTrue("PERMANENT" in datasets["nc_spm_08"]) user = ActiniaUser(user_id=self.user_id) user.read_from_db() user.set_role("user") user.set_cell_limit(1000000) user.set_process_num_limit(10) user.set_process_time_limit(50) user.update() print(user) self.assertEqual(user.get_role(), "user") self.assertEqual(user.get_cell_limit(), 1000000) self.assertEqual(user.get_process_num_limit(), 10) self.assertEqual(user.get_process_time_limit(), 50) user = ActiniaUser(user_id=self.user_id) user.read_from_db() user.add_accessible_dataset("utm32n", ["PERMANENT"]) user.add_accessible_modules(["i.vi", ]) user.update() print(user) datasets = user.get_accessible_datasets() modules = user.get_accessible_modules() self.assertTrue("nc_spm_08" in datasets) self.assertTrue("PERMANENT" in datasets["nc_spm_08"]) self.assertTrue("g.region" in modules) self.assertTrue("utm32n" in datasets) self.assertTrue("PERMANENT" in datasets["utm32n"]) self.assertTrue("i.vi" in modules) user = ActiniaUser(user_id=self.user_id) user.read_from_db() user.remove_location("utm32n") user.remove_mapsets_from_location("nc_spm_08", ["user1", ]) user.remove_accessible_modules(["i.vi", ]) user.update() print(user) datasets = user.get_accessible_datasets() modules = user.get_accessible_modules() self.assertTrue("nc_spm_08" in datasets) self.assertTrue("PERMANENT" in datasets["nc_spm_08"]) self.assertFalse("user1" in datasets["nc_spm_08"]) self.assertTrue("g.region" in modules) self.assertFalse("utm32n" in datasets) self.assertFalse("i.vi" in modules)