Fix tests

safe-global · Dec 19, 2024 · a4b4315 · a4b4315
1 parent 2433738
commit a4b4315
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 4 deletions.
diff --git a/app/tests/db/test_model.py b/app/tests/db/test_model.py
@@ -10,7 +10,6 @@ class TestModel(DbAsyncConn):
     async def test_contract(self, session: AsyncSession):
         contract = Contract(address=b"a", name="A test contract", chain_id=1)
         await contract.create(session)
-        await contract.create(session)
         result = await contract.get_all(session)
         self.assertEqual(result[0], contract)
 

diff --git a/app/tests/routers/test_contracts.py b/app/tests/routers/test_contracts.py
@@ -17,12 +17,18 @@ def setUpClass(cls):
 
     @database_session
     async def test_view_contracts(self, session: AsyncSession):
-        contract = Contract(address=b"a", name="A Test Contracts", chain_id=1)
+        contract = Contract(
+            address=b"0xe94B2EC38FA88bDc8cA9110b24deB5341ECeF251",
+            name="A Test Contracts",
+            chain_id=1,
+        )
         expected_response = {
             "name": "A Test Contracts",
             "description": None,
-            "address": "a",
+            "address": "0xe94B2EC38FA88bDc8cA9110b24deB5341ECeF251",
         }
         await contract.create(session)
-        response = self.client.get("/api/v1/contracts")
+        response = self.client.get(
+            "/api/v1/contracts/0xe94B2EC38FA88bDc8cA9110b24deB5341ECeF251"
+        )
         self.assertEqual(response.status_code, 200)