From bc37a031749412ebd7adf2ab240e19454a77cb31 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Rapha=C3=ABl=20Vinot?= Date: Tue, 30 Nov 2021 11:41:53 +0100 Subject: [PATCH] fix: Update live tests to support proper format of SGs --- pymisp/mispevent.py | 4 ++-- tests/testlive_comprehensive.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/pymisp/mispevent.py b/pymisp/mispevent.py index 8ed34c3..f67322c 100644 --- a/pymisp/mispevent.py +++ b/pymisp/mispevent.py @@ -186,8 +186,8 @@ class MISPSharingGroup(AbstractMISP): def __repr__(self) -> str: if hasattr(self, 'name'): - return f'<{self.__class__.__name__}(name={self.name})' - return f'<{self.__class__.__name__}(NotInitialized)' + return f'<{self.__class__.__name__}(name={self.name})>' + return f'<{self.__class__.__name__}(NotInitialized)>' def _to_feed(self) -> Dict: to_return = super()._to_feed() diff --git a/tests/testlive_comprehensive.py b/tests/testlive_comprehensive.py index 0004a9c..b6dc153 100644 --- a/tests/testlive_comprehensive.py +++ b/tests/testlive_comprehensive.py @@ -2227,8 +2227,8 @@ class TestComprehensive(unittest.TestCase): self.assertEqual(sharing_group.name, 'Testcases SG') # Check we have the org field present and the first org is our org - self.assertTrue(isinstance(getattr(sharing_group, "orgs"), list)) - self.assertEqual(sharing_group.orgs[0].id, self.test_org.id) + self.assertTrue(isinstance(getattr(sharing_group, "sgorgs"), list)) + self.assertEqual(sharing_group.sgorgs[0].org_id, self.test_org.id) finally: self.admin_misp_connector.delete_sharing_group(sharing_group.id)