"""spotify_artificial_streams flow tests constants."""

__all__ = [
    '_GARCON_METHOD_PREFIX',
    '_TASK_STATUS_PREFIX',
    'garcon_methods',
    'garcon_related_methods',
    's3_related_methods',
    'snowflake_executor_methods',
]


_GARCON_METHOD_PREFIX = 'garcon_contrib.dynamo_feed_status.garcon_feed_status'
_S3_TASKS = 'feed_ingestion.tasks.s3_tasks'
_SF_EXECUTOR_PREFIX = (
    'feed_ingestion.flows.spotify_artificial_streams'
    '.snowflake_executor.SpotifyArtificialStreamsSnowflakeSQLExecutor'
)
_TASK_STATUS_PREFIX = 'feed_ingestion.util.task_status'
garcon_methods = {
    'delete_status': f'{_GARCON_METHOD_PREFIX}.delete_status',
    'get_overall_status': f'{_GARCON_METHOD_PREFIX}.get_overall_status',
    'set_overall_status': f'{_GARCON_METHOD_PREFIX}.set_overall_status',
}
garcon_related_methods = {
    'is_completed_task': f'{_TASK_STATUS_PREFIX}.is_completed_task',
    'mark_completed_task': f'{_TASK_STATUS_PREFIX}.mark_completed_task',
}
s3_related_methods = {
    'copy_files': f'{_S3_TASKS}.copy_files',
    'remove_files_from_path': f'{_S3_TASKS}.remove_files_from_path',
    'source_files': f'{_S3_TASKS}.source_files',
}
snowflake_executor_methods = {
    'execute_query': f'{_SF_EXECUTOR_PREFIX}.execute_query',
    'fetchall_query': f'{_SF_EXECUTOR_PREFIX}.fetchall_query',
    'get_connection': f'{_SF_EXECUTOR_PREFIX}.get_connection',
    'clean_staging_raw_on_date': (
        f'{_SF_EXECUTOR_PREFIX}.clean_staging_raw_on_date'
    ),
}