"""Queries for test."""

from flows.projections import config as flow_config


assert_queries = {
    'logs_count': {
        'title': 'logs count check',
        'sql': 'select workflow_run_id from {log_table};'.format(
            log_table=flow_config.PROJECTIONS_REVENUE_ETL_LOG),
        'check_value': 1,
        'error_message': 'Test was interrupted due explicit database changes'
    },
    'revenue_data': {
        'title': 'revenue data record count check',
        'sql': 'select upc from {revenue_table};'.format(
            revenue_table=flow_config.PROJECTIONS_REVENUE),
        'check_value': 50,
        'error_message': 'Incorrect projections_revenue data records count'
    }
}

insert_revenue = """
    INSERT test_theatrical_revenue(
        upc, display_upc, amount, date, transaction_type_id, country_id,
        units)
    VALUES (%s, %s, %s, %s, %s, %s, %s)
"""