from feed_ingestion.flows.apple_music_streams import exec # noqa:A004 def test_prepare_parser_args(): parser = exec.FlowExec().prepare_parser() expected_args = { 'licensor', 'context_date', 'days', 'skip', 'context', 'reload', 'check_status', } args = {action.dest: action for action in parser._actions} assert expected_args.issubset(args), expected_args.difference(args) assert set(args['licensor'].choices) == { 'ALL', 'theorchard', 'sme', 'awal'}