fix test

fetchai · Jan 14, 2025 · 95e0e8f · 95e0e8f
1 parent c29372a
commit 95e0e8f
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/tests/test_registration.py b/python/tests/test_registration.py
@@ -87,7 +87,7 @@ class TestContextSendMethods(unittest.IsolatedAsyncioTestCase):
     def setUp(self):
         self.identity = Identity.generate()
         self.policy = AlmanacApiRegistrationPolicy(
-            self.identity, almanac_api=self.MOCKED_ALMANAC_API, max_retries=1
+            almanac_api=self.MOCKED_ALMANAC_API, max_retries=1
         )
 
     @aioresponses()
@@ -97,6 +97,7 @@ async def test_registration_success(self, mocked_responses):
 
         await self.policy.register(
             agent_identifier=self.identity.address,
+            identity=self.identity,
             protocols=TEST_PROTOCOLS,
             endpoints=TEST_ENDPOINTS,
         )
@@ -109,6 +110,7 @@ async def test_registration_failure(self, mocked_responses):
         with pytest.raises(ClientResponseError):
             await self.policy.register(
                 agent_identifier=self.identity.address,
+                identity=self.identity,
                 protocols=TEST_PROTOCOLS,
                 endpoints=TEST_ENDPOINTS,
             )
@@ -121,6 +123,7 @@ async def test_registration_server_failure(self, mocked_responses):
         with pytest.raises(ClientResponseError):
             await self.policy.register(
                 agent_identifier=self.identity.address,
+                identity=self.identity,
                 protocols=TEST_PROTOCOLS,
                 endpoints=TEST_ENDPOINTS,
             )