def test_get_roles(self, server_iface, DBSession, test_data): # noqa: ignore=N803 dbsession = DBSession() ogcserver_accesscontrol = OGCServerAccessControl( server_iface, "qgisserver1", "no_project", 21781, lambda: dbsession ) set_request_parameters(server_iface, {"USER_ID": "0"}) assert "ROOT" == ogcserver_accesscontrol.get_roles(dbsession) test_users = test_data["users"] test_roles = test_data["roles"] for user_name, expected_role_names in ( ("user1", ("role1",)), ("user12", ("role1", "role2")), ): set_request_parameters( server_iface, { "USER_ID": str(test_users[user_name]["id"]), "ROLE_IDS": ",".join([str(test_roles[r]["id"]) for r in expected_role_names]), }, ) expected_roles = { test_roles[expected_role_name]["id"] for expected_role_name in expected_role_names } assert expected_roles == {role.id for role in ogcserver_accesscontrol.get_roles(dbsession)}
def test_get_roles(self, server_iface, dbsession, test_data): ogcserver_accesscontrol = OGCServerAccessControl( server_iface, "qgisserver1", 21781, dbsession) set_request_parameters(server_iface, {"USER_ID": "0"}) assert "ROOT" == ogcserver_accesscontrol.get_roles() test_users = test_data["users"] test_roles = test_data["roles"] for user_name, expected_role_names in (("user1", ("role1", )), ("user12", ("role1", "role2"))): set_request_parameters(server_iface, {"USER_ID": str(test_users[user_name].id)}) expected_roles = { test_roles[expected_role_name] for expected_role_name in expected_role_names } assert expected_roles == set(ogcserver_accesscontrol.get_roles())
def test_get_roles(self, server_iface, dbsession, test_data): ogcserver_accesscontrol = OGCServerAccessControl( server_iface, 'qgisserver1', 21781, dbsession) set_request_parameters(server_iface, {'USER_ID': '0'}) assert 'ROOT' == ogcserver_accesscontrol.get_roles() test_users = test_data['users'] test_roles = test_data['roles'] for user_name, expected_role_names in ( ('user1', ('role1', )), ('user12', ('role1', 'role2')), ): set_request_parameters(server_iface, {'USER_ID': str(test_users[user_name].id)}) expected_roles = { test_roles[expected_role_name] for expected_role_name in expected_role_names } assert expected_roles == set(ogcserver_accesscontrol.get_roles())