# pylint: disable=unused-argument from db_schema.schemas.apps import FailureLog as AppsFailureLog from delphi_slz_admin.modelviews.apps_failure_log import FilterUnitOfWorkId def test_filter_unit_of_work_id__apply_included(db, clean_db, apps_failure_log): filter_ = FilterUnitOfWorkId() initial_query = db.query(AppsFailureLog) filtered_query = filter_.apply(initial_query, value=[apps_failure_log.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_failure_log): filter_ = FilterUnitOfWorkId() initial_query = db.query(AppsFailureLog) filtered_query = filter_.apply(initial_query, value=[apps_failure_log.unit_of_work_id + 1]) assert initial_query.count() == 1 assert filtered_query.count() == 0