Test data_from_year field

City-of-Turku · Sep 13, 2023 · 7cbc2b4 · 7cbc2b4
1 parent 5da1524
commit 7cbc2b4
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/eco_counter/tests/test_api.py b/eco_counter/tests/test_api.py
@@ -3,6 +3,7 @@
 import pytest
 from rest_framework.reverse import reverse
 
+from .conftest import TEST_TIMESTAMP
 from .constants import TEST_EC_STATION_NAME
 
 
@@ -271,6 +272,7 @@ def test__station(api_client, stations, year_datas):
  assert response.status_code == 200
  assert response.json()["results"][0]["name"] == TEST_EC_STATION_NAME
  assert response.json()["results"][0]["sensor_types"] == ["at"]
+ assert response.json()["results"][0]["data_from_year"] == TEST_TIMESTAMP.year
  # Test retrieving station by data type
  url = reverse("eco_counter:stations-list") + "?data_type=a"
  response = api_client.get(url)