# pylint: disable=unused-argument

from db_schema.schemas.apps import ContentStatus as AppsContentStatus

from delphi_slz_admin.modelviews.apps_content_status import FilterUnitOfWorkId


def test_filter_unit_of_work_id__apply_included(db, clean_db, apps_content_status):
    filter_ = FilterUnitOfWorkId()
    initial_query = db.query(AppsContentStatus)

    filtered_query = filter_.apply(initial_query, value=[apps_content_status.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, apps_content_status):
    filter_ = FilterUnitOfWorkId()
    initial_query = db.query(AppsContentStatus)

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

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