adds new fields

nf1s · May 30, 2021 · 32a8170 · 32a8170
1 parent b5f9faa
commit 32a8170
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/covid/worldometers/models.py b/covid/worldometers/models.py
@@ -11,14 +11,15 @@ class CovidModel(BaseModel):
     """Dataclass acts as a Model for Covid data"""
 
     country: str = Field(..., alias="Country,Other")
+    total_cases: int = Field(0, alias="TotalCases")
     confirmed: int = Field(0, alias="TotalCases")
     new_cases: int = Field(0, alias="NewCases")
     deaths: int = Field(0, alias="TotalDeaths")
     new_deaths: int = Field(0, alias="NewDeaths")
     recovered: int = Field(0, alias="TotalRecovered")
     active: int = Field(0, alias="ActiveCases")
+    active_cases: int = Field(0, alias="ActiveCases")
     critical: int = Field(0, alias="Serious,Critical")
-    new_deaths: int = Field(0, alias="NewDeaths")
     total_tests: int = Field(0, alias="TotalTests")
     total_tests_per_million: Decimal = Field(Decimal(0), alias="Tests/1M pop")
     total_cases_per_million: Decimal = Field(

diff --git a/tests/test_worldometer.py b/tests/test_worldometer.py
@@ -15,7 +15,9 @@ def test_all_data():
     element = data[0]
     assert "country" in element
     assert "confirmed" in element
+    assert "total_cases" in element
     assert "active" in element
+    assert "active_cases" in element
     assert "deaths" in element
     assert "recovered" in element
     assert "new_cases" in element
@@ -34,7 +36,9 @@ def test_get_by_country_name():
     assert type(element) is dict
     assert "country" in element
     assert "confirmed" in element
+    assert "total_cases" in element
     assert "active" in element
+    assert "active_cases" in element
     assert "deaths" in element
     assert "recovered" in element
     assert "new_cases" in element