add basic smoke tests for the data layer

velodrome-finance · Oct 30, 2023 · ed672b3 · ed672b3
1 parent 25c1c87
commit ed672b3
Show file tree

Hide file tree

Showing 4 changed files with 40 additions and 6 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -15,6 +15,7 @@ python-dotenv = "1.0.0"
 
 [tool.poetry.group.dev.dependencies]
 pytest = "^7.4.3"
+pytest-asyncio = "^0.21.1"
 
 [build-system]
 requires = ["poetry-core"]

diff --git a/tests/test_common_sense.py b/tests/test_common_sense.py
diff --git a/tests/test_data.py b/tests/test_data.py
@@ -0,0 +1,20 @@
+import pytest
+
+from dotenv import load_dotenv
+load_dotenv(".env.example")
+
+from bots.settings import TOKEN_ADDRESS, STABLE_TOKEN_ADDRESS
+from bots.data import Token, Price
+
+@pytest.mark.asyncio
+async def test_get_by_token_address():
+    t = await Token.get_by_token_address(TOKEN_ADDRESS)
+    assert t.token_address == TOKEN_ADDRESS
+    assert t.listed == True
+
+@pytest.mark.asyncio
+async def test_get_prices():
+    prices = await Price.get_prices([await Token.get_by_token_address(TOKEN_ADDRESS)])
+    assert len(prices) == 1
+    [price] = prices
+    assert price.pretty_price != 0