new: Method to set the default role

pull/382/head
Raphaël Vinot 2019-04-04 16:39:17 +02:00
parent 3fb54e62b2
commit b1b9f95501
2 changed files with 12 additions and 1 deletions

View File

@ -553,3 +553,8 @@ class ExpandedPyMISP(PyMISP):
me.from_dict(**e_meta)
to_return.append(me)
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)

View File

@ -52,8 +52,9 @@ class TestComprehensive(unittest.TestCase):
organisation = MISPOrganisation()
organisation.name = 'Test Org'
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
# TODO & FIXME: set the default role to User is not already set - MISP/MISP #4423
user = MISPUser()
user.email = 'testusr@user.local'
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()
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__':
unittest.main()