diff --git a/misp_modules/modules/expansion/dbl_spamhaus.py b/misp_modules/modules/expansion/dbl_spamhaus.py index 4ecdfb9..2bccd00 100644 --- a/misp_modules/modules/expansion/dbl_spamhaus.py +++ b/misp_modules/modules/expansion/dbl_spamhaus.py @@ -50,7 +50,7 @@ def handler(q=False): return misperrors query = "{}.{}".format(requested_value, dbl) try: - query_result = resolver.query(query, 'A')[0] + query_result = resolver.resolve(query, 'A')[0] result = "{} - {}".format(requested_value, dbl_mapping[str(query_result)]) except dns.resolver.NXDOMAIN as e: result = e.msg diff --git a/tests/test_expansions.py b/tests/test_expansions.py index 8099a62..9601724 100644 --- a/tests/test_expansions.py +++ b/tests/test_expansions.py @@ -215,7 +215,7 @@ class TestExpansions(unittest.TestCase): self.assertEqual(self.get_values(response), 'totalmateria.net - spam test domain') except Exception: try: - self.assertTrue(self.get_values(response).startswith('None of DNS query names exist:')) + self.assertTrue(self.get_values(response).startswith('The DNS query name does not exist:')) except Exception: self.assertEqual(self.get_errors(response), 'Not able to reach dbl.spamhaus.org or something went wrong')