File size: 1,271 Bytes
8b96826
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
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
"""SQLite TTL cache behavior."""

import time

from app import cache


def test_set_and_get_roundtrip():
    cache.set("key1", {"a": 1, "b": [1, 2]}, ttl_seconds=60)
    assert cache.get("key1") == {"a": 1, "b": [1, 2]}


def test_missing_key_returns_none():
    assert cache.get("never-set") is None


def test_expired_entry_returns_none():
    cache.set("short", "value", ttl_seconds=-1)  # already expired
    assert cache.get("short") is None


def test_overwrite():
    cache.set("key", "old", ttl_seconds=60)
    cache.set("key", "new", ttl_seconds=60)
    assert cache.get("key") == "new"


def test_cached_decorator_calls_function_once():
    calls = {"count": 0}

    @cache.cached(ttl_seconds=60)
    def expensive(x):
        calls["count"] += 1
        return x * 2

    assert expensive(21) == 42
    assert expensive(21) == 42
    assert calls["count"] == 1
    # Different args -> separate cache entry
    assert expensive(10) == 20
    assert calls["count"] == 2


def test_cached_decorator_skips_none_results():
    calls = {"count": 0}

    @cache.cached(ttl_seconds=60)
    def flaky():
        calls["count"] += 1
        return None

    assert flaky() is None
    assert flaky() is None
    assert calls["count"] == 2  # None results are not cached