"""Queries for test."""

from flows.theatrical import config

assert_queries = {
    'logs_count': {
        'title': 'logs count check',
        'sql': 'select workflow_run_id from {log_table};'.format(
            log_table=config.THEATRICAL_REVENUE_ETL_LOG),
        'check_value': 1,
        'error_message': 'Test was interrupted due explicit database changes'
    },
    'raw_data': {
        'title': 'raw data record count check',
        'sql': 'select id from {revenue_raw_table};'.format(
            revenue_raw_table=config.THEATRICAL_REVENUE_RAW),
        'check_value': 105,
        'error_message': 'Incorrect raw data records count'
    },
    'revenue_data': {
        'title': 'revenue data record count check',
        'sql': 'select id from {revenue_raw_table};'.format(
            revenue_raw_table=config.THEATRICAL_REVENUE),
        'check_value': 21,
        'error_message': 'Incorrect theatrical_revenue data records count'
    }
}