"""Unit: InMemoryStateStore."""

from __future__ import annotations

import unittest

from bot_bottle.orchestrator.model import RunRecord
from bot_bottle.orchestrator.store import InMemoryStateStore


def _rec(issue: int, owner: str = "o") -> RunRecord:
    return RunRecord(owner=owner, repo="r", issue_number=issue, slug=f"s{issue}",
                     agent_name="a")


class InMemoryStoreTest(unittest.TestCase):
    def setUp(self):
        self.store = InMemoryStateStore()

    def test_upsert_get(self):
        self.store.upsert(_rec(1))
        got = self.store.get("o", "r", 1)
        assert got is not None
        self.assertEqual("s1", got.slug)

    def test_get_missing(self):
        self.assertIsNone(self.store.get("o", "r", 99))

    def test_upsert_replaces(self):
        self.store.upsert(_rec(1))
        r = _rec(1)
        r.slug = "changed"
        self.store.upsert(r)
        self.assertEqual("changed", self.store.get("o", "r", 1).slug)  # type: ignore[union-attr]
        self.assertEqual(1, len(self.store.all()))

    def test_delete(self):
        self.store.upsert(_rec(1))
        self.store.delete("o", "r", 1)
        self.assertIsNone(self.store.get("o", "r", 1))

    def test_all_sorted(self):
        self.store.upsert(_rec(2, owner="b"))
        self.store.upsert(_rec(1, owner="a"))
        self.assertEqual([("a", 1), ("b", 2)],
                         [(r.owner, r.issue_number) for r in self.store.all()])


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