# pylint: disable=unused-argument

from db_schema.schemas.slz import Snapshot

from delphi_slz_admin.modelviews.snapshot import FilterContentStatusId


def test_filter_content_status_id__apply_included(db, clean_db, snapshot):
    filter_ = FilterContentStatusId()
    initial_query = db.query(Snapshot)

    filtered_query = filter_.apply(initial_query, value=[snapshot.content_status_id])

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


def test_filter_content_status_id__apply_excluded(db, clean_db, snapshot):
    filter_ = FilterContentStatusId()
    initial_query = db.query(Snapshot)

    filtered_query = filter_.apply(initial_query, value=[snapshot.content_status_id + 1])

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