Spaces:
Running
Running
File size: 2,250 Bytes
d5a6f3e | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 | """
Unit tests for user_state.py — pure in-memory logic, no DB/network.
"""
import pytest
from collections import deque
# We test the UserState dataclass directly
from app.user_state import UserState, MAX_POSITIVES, MAX_NEGATIVES
def test_add_positive_basic():
s = UserState()
s.add_positive("1706.03762")
assert "1706.03762" in s.positives
assert "1706.03762" not in s.negatives
def test_add_negative_basic():
s = UserState()
s.add_negative("1706.03762")
assert "1706.03762" in s.negatives
assert "1706.03762" not in s.positives
def test_save_moves_from_negatives_to_positives():
s = UserState()
s.add_negative("abc")
s.add_positive("abc")
assert "abc" in s.positives
assert "abc" not in s.negatives
def test_dismiss_moves_from_positives_to_negatives():
s = UserState()
s.add_positive("abc")
s.add_negative("abc")
assert "abc" in s.negatives
assert "abc" not in s.positives
def test_no_duplicates_in_positives():
s = UserState()
s.add_positive("abc")
s.add_positive("abc")
assert list(s.positives).count("abc") == 1
def test_positive_list_most_recent_first():
s = UserState()
for i in range(3):
s.add_positive(f"paper{i}")
# appendleft → most recent is first
assert s.positive_list[0] == "paper2"
assert s.positive_list[-1] == "paper0"
def test_maxlen_positives_evicts_oldest():
s = UserState()
for i in range(MAX_POSITIVES + 5):
s.add_positive(f"paper{i}")
assert len(s.positives) == MAX_POSITIVES
# Newest are kept (appendleft evicts right end)
assert "paper4" not in s.positives # oldest evicted
def test_has_enough_for_recs_false_when_empty():
s = UserState()
assert not s.has_enough_for_recs()
def test_has_enough_for_recs_true_after_one_save():
s = UserState()
s.add_positive("1706.03762")
assert s.has_enough_for_recs()
def test_all_seen_union():
from app.user_state import _cache, record_positive, record_negative, all_seen
uid = "test-seen-user"
_cache.pop(uid, None)
record_positive(uid, "pos1")
record_negative(uid, "neg1")
seen = all_seen(uid)
assert "pos1" in seen
assert "neg1" in seen
_cache.pop(uid, None)
|