from cicd_debug_env.memory.failure_bank import FailureMemoryBank
from cicd_debug_env.models import Action

def test_memory():
    bank = FailureMemoryBank(store="dict")
    act = Action("fail_act", {}, 0.1, "bad")
    bank.store("Error 404", act, "Fail", 0.0)
    
    assert len(bank.memory) == 1
    
    # We used dict fallback implicitly in query logic when passing top matching queries
    res = bank.query("Error 404", 1)
    
    assert bank.should_avoid("Error 404", "fail_act")
    assert not bank.should_avoid("Error 404", "good_act")