"""Tests for SME Latam flow.""" from unittest import mock from unittest.mock import MagicMock from feed_ingestion.flows.insta_reels.flow import Flow def test_flow_decider(): """Test normal decider execution.""" insta_reels_flow = Flow() schedule = MagicMock() schedule_result_object = MagicMock() schedule_result_object.result = {} schedule.return_value = schedule_result_object insta_reels_flow.decider(schedule) schedule.assert_has_calls([ mock.call('bootstrap', mock.ANY), mock.call('fetch_from_athena', mock.ANY, requires=[mock.ANY]), mock.call( 'load_staging_raw_table', mock.ANY, requires=[mock.ANY]), ])