tests/test_llm.py


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

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parent.parent))

import unittest

import deckit.modules.llm as llm


class Down(unittest.TestCase):
    def test_offline_when_dashboard_down(self):
        def boom(url, timeout=2):
            raise OSError("refused")
        st = llm.make({"llm": {"url": "http://x"}, "fetch": boom}).state()
        self.assertTrue(st["offline"])


class Up(unittest.TestCase):
    def test_merges_endpoints(self):
        data = {"/api/claude_limits": [{"context_pct": 12}],
                "/api/claude_session": {"context_used": 1200},
                "/api/summary": {"total_cost": 1.5}}
        st = llm.make({"llm": {"url": "http://x"},
                       "fetch": lambda u, timeout=2: data[u]}).state()
        self.assertFalse(st["offline"])
        self.assertEqual(st["claude"]["context_pct"], 12)
        self.assertEqual(st["opencode"]["total_cost"], 1.5)


if __name__ == "__main__":
    unittest.main()