from jb.decorators import HM
from jb.flow.tasks import refill_hits, check_stale_hits, check_expired_hits


def test_refill_hits(
    set_hit_types_in_db_min_active_0, hit_type_in_db, expire_all_hits, amt_manager
):

    assert HM.get_active_count(hit_type_in_db.id) == 0
    assert hit_type_in_db.min_active > 0
    refill_hits()
    assert HM.get_active_count(hit_type_in_db.id) == hit_type_in_db.min_active

    amt_hit_ids = HM.filter_active_ids(hit_type_id=hit_type_in_db.id)
    amt_hit_id = list(amt_hit_ids)[0]
    hit, _ = amt_manager.get_hit_if_exists(amt_hit_id=amt_hit_id)
    assert hit


def test_check_stale_hits():
    # todo: I'd have to create some purposely stale hits.
    # just make sure it runs for now
    check_stale_hits()


def test_check_expired_hits():
    # todo: I'd have to create some purposely expired hits.
    # just make sure it runs for now
    check_expired_hits()