fix tests

cowprotocol · Oct 10, 2024 · d76b4b9 · d76b4b9
1 parent 52392c8
commit d76b4b9
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/token_decimals.py b/src/token_decimals.py
@@ -25,7 +25,7 @@ def update_token_decimals(engine, web3) -> None:
 
 
 if __name__ == "__main__":
-    engine = create_engine(f"postgresql+psycopg2://{getenv('SOLVER_SLIPPAGE_DB_URL')}")
+    engine = create_engine(f"postgresql+psycopg://{getenv('SOLVER_SLIPPAGE_DB_URL')}")
     web3 = Web3(Web3.HTTPProvider(getenv("NODE_URL")))
 
     update_token_decimals(engine, web3)
diff --git a/tests/unit/test_database.py b/tests/unit/test_database.py
@@ -8,7 +8,7 @@
 
 def tests_write_transaction_timestamp():
     engine = create_engine(
-        f"postgresql+psycopg2://postgres:postgres@localhost:5432/mainnet"
+        f"postgresql+psycopg://postgres:postgres@localhost:5432/mainnet"
     )
     db = Database(engine, "mainnet")
     # truncate table
@@ -38,7 +38,7 @@ def tests_write_transaction_tokens():
     # import has to happen after patching environment variable
 
     engine = create_engine(
-        f"postgresql+psycopg2://postgres:postgres@localhost:5432/mainnet"
+        f"postgresql+psycopg://postgres:postgres@localhost:5432/mainnet"
     )
     db = Database(engine, "mainnet")
     transaction_tokens = [
@@ -69,7 +69,7 @@ def tests_write_transaction_tokens():
 
 def tests_write_prices():
     engine = create_engine(
-        f"postgresql+psycopg2://postgres:postgres@localhost:5432/mainnet"
+        f"postgresql+psycopg://postgres:postgres@localhost:5432/mainnet"
     )
     db = Database(engine, "mainnet")
     token_prices = [
@@ -109,7 +109,7 @@ def test_get_latest_transaction():
     from src.helpers.database import Database
 
     engine = create_engine(
-        f"postgresql+psycopg2://postgres:postgres@localhost:5432/mainnet"
+        f"postgresql+psycopg://postgres:postgres@localhost:5432/mainnet"
     )
     db = Database(engine, "mainnet")
     # truncate table