# pylint: disable=unused-argument

from db_schema.schemas.apps import DatabricksExecution, DBXJobStatusEnum

from delphi_slz_admin.modelviews.apps_dbx_execution import FilterStatus, FilterUnitOfWorkId


def test_filter_unit_of_work_id__apply_included(db, clean_db, dbx_execution):
    filter_ = FilterUnitOfWorkId()
    initial_query = db.query(DatabricksExecution)

    filtered_query = filter_.apply(initial_query, value=[dbx_execution.unit_of_work_id])

    assert initial_query.count() == 1
    assert filtered_query.count() == 1


def test_filter_unit_of_work_id__apply_excluded(db, clean_db, dbx_execution):
    filter_ = FilterUnitOfWorkId()
    initial_query = db.query(DatabricksExecution)

    filtered_query = filter_.apply(initial_query, value=[dbx_execution.unit_of_work_id + 1])

    assert initial_query.count() == 1
    assert filtered_query.count() == 0


def test_filter_status__apply_included(db, clean_db, dbx_execution):
    dbx_execution.status = DBXJobStatusEnum.IN_PROGRESS
    filter_ = FilterStatus()
    initial_query = db.query(DatabricksExecution)

    filtered_query = filter_.apply(initial_query, value=DBXJobStatusEnum.IN_PROGRESS.value)

    assert initial_query.count() == 1
    assert filtered_query.count() == 1


def test_filter_status__apply_excluded(db, clean_db, dbx_execution):
    dbx_execution.status = DBXJobStatusEnum.COMPLETE
    filter_ = FilterStatus()
    initial_query = db.query(DatabricksExecution)

    filtered_query = filter_.apply(initial_query, value=DBXJobStatusEnum.IN_PROGRESS.value)

    assert initial_query.count() == 1
    assert filtered_query.count() == 0