# pylint: disable=unused-argument,protected-access,too-many-arguments
import pytest
from db_schema.schemas import slz


@pytest.mark.integration
def test_get_statuses_content_files(db, unit_of_work, content_status_repository):
    status_context_dict = {
        'us': 'ACTIVE',
        'global': 'COMPLETE_VOLATILE',
        'au': 'COMPLETE_VOLATILE',
        'fr': 'MISSING'
    }

    for context, status in status_context_dict.items():
        content_status = slz.ContentStatus(
            **{
                'unit_of_work_id': unit_of_work.unit_of_work_id,
                'context': context,
                'content_status': status,
                'content_name': f'{context}_{status}.parquet',
                'failure_count': 0,
                'latest_job_id': 'test_job_id__charts_daily_regional',
                'created_at': '2021-08-01 12:30:17',
                'last_checked_at': '2021-08-02 12:30:17',
                'metadata_process_status': 'NOT_QUEUED',
            }
        )
        db.add(content_status)

    file_names = content_status_repository.get_statuses_content_files()
    assert file_names == ['global_COMPLETE_VOLATILE.parquet', 'au_COMPLETE_VOLATILE.parquet']