core/tests/components/coronavirus/conftest.py

18 lines
463 B
Python
Raw Normal View History

"""Test helpers."""
from asynctest import Mock, patch
import pytest
@pytest.fixture(autouse=True)
def mock_cases():
"""Mock coronavirus cases."""
with patch(
"coronavirus.get_cases",
return_value=[
Mock(country="Netherlands", confirmed=10, recovered=8, deaths=1, current=1),
Mock(country="Germany", confirmed=1, recovered=0, deaths=0, current=0),
],
) as mock_get_cases:
yield mock_get_cases