from unittest.mock import AsyncMock, patch, ANY
from pytest import fixture, mark
from scouter.workflow.fake_data import FakeData
from scouter.activities.faker import Faker


@fixture
def fake_data():
    return FakeData()


@mark.asyncio
@patch('scouter.workflow.fake_data.workflow', new_callable=AsyncMock)
async def test_fake_data_workflow(mock_workflow, fake_data):
    mock_workflow.execute_activity_method.return_value = None
    await fake_data.run(
        {
            'topic': 'test_topic'
        }
    )

    mock_workflow.execute_activity_method.assert_called_once_with(
        Faker.generate_and_send_data,
        {
            'topic': 'test_topic'
        },
        retry_policy=ANY,
        start_to_close_timeout=ANY
    )