fix: tests

fetchai · Oct 4, 2023 · 83418ca · 83418ca
1 parent 96df1d9
commit 83418ca
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/tests/test_agent_registration.py b/python/tests/test_agent_registration.py
@@ -35,7 +35,7 @@ def test_name_service_failed_ownership(self):
 
         domain = "agent"
 
-        name_service_contract = get_name_service_contract()
+        name_service_contract = get_name_service_contract(test=True)
 
         is_owner = name_service_contract.is_owner(
             agent.name, domain, str(agent.wallet.address())
@@ -58,7 +58,7 @@ async def test_name_service_registration(self):
             "Almanac registration failed",
         )
 
-        name_service_contract = get_name_service_contract()
+        name_service_contract = get_name_service_contract(test=True)
 
         is_name_available = name_service_contract.is_name_available(agent.name, domain)
         self.assertEqual(is_name_available, True, "Agent name should be available")
@@ -80,7 +80,7 @@ async def test_name_service_registration(self):
         )
         self.assertEqual(is_owner, True, "Domain ownership failed")
 
-        query_address = get_agent_address(agent.name + "." + domain)
+        query_address = get_agent_address(agent.name + "." + domain, True)
 
         self.assertEqual(
             query_address == agent.address, True, "Service contract registration failed"