mirror of https://github.com/MISP/PyMISP
new: Method to set the default role
parent
3fb54e62b2
commit
b1b9f95501
|
@ -553,3 +553,8 @@ class ExpandedPyMISP(PyMISP):
|
||||||
me.from_dict(**e_meta)
|
me.from_dict(**e_meta)
|
||||||
to_return.append(me)
|
to_return.append(me)
|
||||||
return to_return
|
return to_return
|
||||||
|
|
||||||
|
def set_default_role(self, role_id: int):
|
||||||
|
url = urljoin(self.root_url, f'/admin/roles/set_default/{role_id}')
|
||||||
|
response = self._prepare_request('POST', url)
|
||||||
|
return self._check_response(response)
|
||||||
|
|
|
@ -52,8 +52,9 @@ class TestComprehensive(unittest.TestCase):
|
||||||
organisation = MISPOrganisation()
|
organisation = MISPOrganisation()
|
||||||
organisation.name = 'Test Org'
|
organisation.name = 'Test Org'
|
||||||
cls.test_org = cls.admin_misp_connector.add_organisation(organisation)
|
cls.test_org = cls.admin_misp_connector.add_organisation(organisation)
|
||||||
|
# Set the refault role (id 3 on the VM)
|
||||||
|
cls.admin_misp_connector.set_default_role(3)
|
||||||
# Creates a user
|
# Creates a user
|
||||||
# TODO & FIXME: set the default role to User is not already set - MISP/MISP #4423
|
|
||||||
user = MISPUser()
|
user = MISPUser()
|
||||||
user.email = 'testusr@user.local'
|
user.email = 'testusr@user.local'
|
||||||
user.org_id = cls.test_org.id
|
user.org_id = cls.test_org.id
|
||||||
|
@ -1128,6 +1129,11 @@ class TestComprehensive(unittest.TestCase):
|
||||||
missing_acls = self.admin_misp_connector.get_live_query_acl()
|
missing_acls = self.admin_misp_connector.get_live_query_acl()
|
||||||
self.assertEqual(missing_acls, [], msg=missing_acls)
|
self.assertEqual(missing_acls, [], msg=missing_acls)
|
||||||
|
|
||||||
|
def test_roles(self):
|
||||||
|
role = self.admin_misp_connector.set_default_role(4)
|
||||||
|
self.assertEqual(role['message'], 'Default role set.')
|
||||||
|
self.admin_misp_connector.set_default_role(3)
|
||||||
|
|
||||||
|
|
||||||
if __name__ == '__main__':
|
if __name__ == '__main__':
|
||||||
unittest.main()
|
unittest.main()
|
||||||
|
|
Loading…
Reference in New Issue