From 1bd619914b0c25a92699c7adab9a21d978046246 Mon Sep 17 00:00:00 2001 From: Jakub Onderka Date: Sat, 1 Jul 2023 14:46:28 +0200 Subject: [PATCH] chg: [internal] Resolve deprecation warning in dbl_spamhaus --- misp_modules/modules/expansion/dbl_spamhaus.py | 2 +- tests/test_expansions.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) 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 5f4d326..0f3f69b 100644 --- a/tests/test_expansions.py +++ b/tests/test_expansions.py @@ -198,7 +198,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')