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())