test_jhu.py

from unittest import mock

import pytest

from app import location
from app.services.location import jhu
from tests.conftest import mocked_strptime_isoformat

DATETIME_STRING = "2020-03-17T10:23:22.505550"


@pytest.mark.asyncio
async def test_get_locations(mock_client_session):
    with mock.patch("app.services.location.jhu.datetime") as mock_datetime:
        mock_datetime.utcnow.return_value.isoformat.return_value = DATETIME_STRING
        mock_datetime.strptime.side_effect = mocked_strptime_isoformat
        output = await jhu.get_locations()

    assert isinstance(output, list)
    assert isinstance(output[0], location.Location)

    # `jhu.get_locations()` creates id based on confirmed list
    location_confirmed = await jhu.get_category("confirmed")
    assert len(output) == len(location_confirmed["locations"])

    # `jhu.get_locations()` creates id based on deaths list
    location_deaths = await jhu.get_category("deaths")
    assert len(output) == len(location_deaths["locations"])

    # `jhu.get_locations()` creates id based on recovered list
    location_recovered = await jhu.get_category("recovered")
    assert len(output) == len(location_recovered["locations"])


@pytest.mark.parametrize(
    "key, locations, index, expected",
    [
        (("Thailand", "TH"), [{"country": "Thailand", "province": "TH", "history": {"test": "yes"}}], 0, {"test": "yes"}), # Success
        (("Deutschland", "DE"), [{"country": "Deutschland", "province": "DE", "history": {"test": "no"}}], 1, {}), # IndexError
        (("US", "NJ"), [{"country": "Deutschland", "province": "DE", "history": {"test": "no"}}], 0, {}), # Invaid Key Merge
    ],
)
def test_parse_history(key, locations, index, expected):
    """
    Test validating and extracting history content from
    locations data based on index.
    """
    assert jhu.parse_history(key, locations, index) == expected