From 8db9891c838ba49e7709653a849a5c82344e2ce9 Mon Sep 17 00:00:00 2001 From: chrisr3d Date: Fri, 10 Jan 2020 15:12:52 +0100 Subject: [PATCH] fix: Updated ipasn test following the latest changes on the module --- tests/test_expansions.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/tests/test_expansions.py b/tests/test_expansions.py index 93ee69d..c6d5944 100644 --- a/tests/test_expansions.py +++ b/tests/test_expansions.py @@ -237,9 +237,7 @@ class TestExpansions(unittest.TestCase): def test_ipasn(self): query = {"module": "ipasn", "ip-dst": "1.1.1.1"} response = self.misp_modules_post(query) - key = list(self.get_values(response)['response'].keys())[0] - entry = self.get_values(response)['response'][key]['asn'] - self.assertEqual(entry, '13335') + self.assertEqual(self.get_object(response), 'asn') def test_macaddess_io(self): module_name = 'macaddress_io'