From 97d4950f823ac62264decd874057dad922aa7853 Mon Sep 17 00:00:00 2001 From: chrisr3d Date: Thu, 15 Apr 2021 16:39:28 +0200 Subject: [PATCH] fix: [tests] Fixed variable names that have been changed with the latest commit --- tests/test_expansions.py | 14 ++++++++------ 1 file changed, 8 insertions(+), 6 deletions(-) diff --git a/tests/test_expansions.py b/tests/test_expansions.py index 5a9a990..88f974d 100644 --- a/tests/test_expansions.py +++ b/tests/test_expansions.py @@ -554,9 +554,10 @@ class TestExpansions(unittest.TestCase): except Exception: self.assertEqual(self.get_errors(response), "VirusTotal request rate limit exceeded.") else: - query = {"module": module_name, - "attribute": {"type": query_types[0], - "value": query_values[0]}} + query = { + "module": module_name, + "attribute": attributes[0] + } response = self.misp_modules_post(query) self.assertEqual(self.get_errors(response), "A VirusTotal api key is required for this module.") @@ -596,9 +597,10 @@ class TestExpansions(unittest.TestCase): except Exception: self.assertEqual(self.get_errors(response), "VirusTotal request rate limit exceeded.") else: - query = {"module": module_name, - "attribute": {"type": query_types[0], - "value": query_values[0]}} + query = { + "module": module_name, + "attribute": attributes[0] + } response = self.misp_modules_post(query) self.assertEqual(self.get_errors(response), "A VirusTotal api key is required for this module.")