"""Test dynamo_sync flow class."""

from unittest import mock

from dim_refresh_etl.flows.dynamo_sync.flow import Flow


def test_decider(mocker):
    """Test decider."""
    schedule = mocker.MagicMock()
    schedule_result_object = mocker.MagicMock()
    schedule_result_object.result = {'bootstrap.full_refresh': True}
    schedule.return_value = schedule_result_object

    flow = Flow()
    flow.decider(schedule, mocker.MagicMock())

    schedule.assert_has_calls([
        mock.call('bootstrap', mock.ANY),
        mock.call('unload_data_to_s3', mock.ANY, requires=[mock.ANY]),
        # mock.call('increase_table_capacity', mock.ANY, requires=[mock.ANY]),
        mock.call('upload_to_dynamodb', mock.ANY, requires=[mock.ANY]),
        # mock.call('decrease_table_capacity', mock.ANY, requires=[mock.ANY]),
        mock.call('health_check', mock.ANY, requires=[mock.ANY]),
    ])