mirror of https://github.com/MISP/PyMISP
chg: Add in test case for searching against orgs and users
parent
43d8cdff4a
commit
1248dd459f
|
@ -1771,6 +1771,33 @@ class TestComprehensive(unittest.TestCase):
|
||||||
organisation = self.admin_misp_connector.update_organisation(organisation, pythonify=True)
|
organisation = self.admin_misp_connector.update_organisation(organisation, pythonify=True)
|
||||||
self.assertEqual(organisation.name, 'blah', organisation)
|
self.assertEqual(organisation.name, 'blah', organisation)
|
||||||
|
|
||||||
|
def test_org_search(self):
|
||||||
|
orgs = self.admin_misp_connector.organisations(pythonify=True)
|
||||||
|
org_name = 'ORGNAME'
|
||||||
|
# Search by the org name
|
||||||
|
orgs = self.admin_misp_connector.organisations(search=org_name, pythonify=True)
|
||||||
|
# There should be one org returned
|
||||||
|
self.assertTrue(len(orgs) == 1)
|
||||||
|
|
||||||
|
# This org should have the name ORGNAME
|
||||||
|
self.assertEqual(orgs[0].name, org_name)
|
||||||
|
|
||||||
|
def test_user_search(self):
|
||||||
|
users = self.admin_misp_connector.users(pythonify=True)
|
||||||
|
emailAddr = users[0].email
|
||||||
|
|
||||||
|
users = self.admin_misp_connector.users(search=emailAddr)
|
||||||
|
self.assertTrue(len(users) == 1)
|
||||||
|
self.assertEqual(users[0]['User']['email'], emailAddr)
|
||||||
|
|
||||||
|
users = self.admin_misp_connector.users(
|
||||||
|
search=emailAddr,
|
||||||
|
organisation=users[0]['Organisation']['id'],
|
||||||
|
pythonify=True
|
||||||
|
)
|
||||||
|
self.assertTrue(len(users) == 1)
|
||||||
|
self.assertEqual(users[0].email, emailAddr)
|
||||||
|
|
||||||
def test_attribute(self):
|
def test_attribute(self):
|
||||||
first = self.create_simple_event()
|
first = self.create_simple_event()
|
||||||
second = self.create_simple_event()
|
second = self.create_simple_event()
|
||||||
|
|
Loading…
Reference in New Issue