Add tests

tests/test_api.py

@@ -8,7 +8,7 @@
 from app import crud
 from app.api import app, get_db
 from app.db import Base
-from app.schemas import LocationCreate, PriceCreate, UserBase
+from app.schemas import LocationCreate, PriceCreate, ProductCreate, UserBase
 
 # database setup
 # ------------------------------------------------------------------------------
@@ -40,6 +40,7 @@ def override_get_db():
 client = TestClient(app)
 
 USER = UserBase(user_id="user1", token="user1__Utoken")
+PRODUCT = ProductCreate(code="8001505005707")
 LOCATION = LocationCreate(osm_id=3344841823, osm_type="NODE")
 PRICE_1 = PriceCreate(
     product_code="1111111111111",
@@ -57,6 +58,12 @@ def user(db=override_get_db()):
     return db_user
 
 
+@pytest.fixture(scope="module")
+def product(db=override_get_db()):
+    db_product = crud.create_product(next(db), PRODUCT)
+    return db_product
+
+
 @pytest.fixture(scope="module")
 def location(db=override_get_db()):
     db_location = crud.create_location(next(db), LOCATION)
@@ -148,6 +155,15 @@ def test_get_proofs(user):
     assert response.status_code == 200
 
 
+def test_get_product(product):
+    # product exists
+    response = client.get(f"/products/{product.id}")
+    assert response.status_code == 200
+    # product does not exist
+    response = client.get(f"/products/{product.id+1}")
+    assert response.status_code == 404
+
+
 def test_get_location(location):
     # location exists
     response = client.get(f"/locations/{location.id}")