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