test: add unit tests

splunk · Dec 19, 2024 · 945235f · 945235f
1 parent b3642db
commit 945235f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/test/snmp/test_tasks.py b/test/snmp/test_tasks.py
@@ -303,6 +303,7 @@ def test_format_ipv4_address(
         # m_poller.return_value = None
         from splunk_connect_for_snmp.snmp.tasks import format_ipv4_address
 
+        m_mongo_client.return_value = MagicMock()
         m_time.return_value = 1640692955.365186
 
         ip_address = "::ffff:172.31.20.76"
@@ -322,6 +323,7 @@ def test_format_ipv4_address_disabled(
         # m_poller.return_value = None
         from splunk_connect_for_snmp.snmp.tasks import format_ipv4_address
 
+        m_mongo_client.return_value = MagicMock()
         # m_time.return_value = 1640692955.365186
 
         ip_address = "::ffff:172.31.20.76"
@@ -341,6 +343,7 @@ def test_format_ipv4_address_ipv6(
         # m_poller.return_value = None
         from splunk_connect_for_snmp.snmp.tasks import format_ipv4_address
 
+        m_mongo_client.return_value = MagicMock()
         m_time.return_value = 1640692955.365186
 
         ip_address = "fd02::b24a:409e:a35e:b580"