From b1b9f95501819062f76d1e86112cae92c6cdc40f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Thu, 4 Apr 2019 16:39:17 +0200 Subject: [PATCH] new: Method to set the default role --- pymisp/aping.py | 5 +++++ tests/testlive_comprehensive.py | 8 +++++++- 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/pymisp/aping.py b/pymisp/aping.py index fdc2baa..d9ee80f 100644 --- a/pymisp/aping.py +++ b/pymisp/aping.py @@ -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) diff --git a/tests/testlive_comprehensive.py b/tests/testlive_comprehensive.py index 25f37a9..f12ba8f 100644 --- a/tests/testlive_comprehensive.py +++ b/tests/testlive_comprehensive.py @@ -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()