import tempfile
import unittest
from pathlib import Path

import app


class StoreTests(unittest.TestCase):
    def setUp(self):
        self.temp = tempfile.TemporaryDirectory()
        self.store = app.Store(Path(self.temp.name) / "state.sqlite3")
        self.store.initialize()

    def tearDown(self):
        self.temp.cleanup()

    def test_empty_store(self):
        self.assertEqual(self.store.snapshot()["checks"], {})
        self.assertEqual(self.store.snapshot()["revision"], 0)

    def test_shared_check_round_trip(self):
        result = self.store.set_check("crop-parsnip", True, "fred")
        self.assertEqual(result["revision"], 1)
        snapshot = self.store.snapshot()
        self.assertTrue(snapshot["checks"]["crop-parsnip"])
        self.assertEqual(snapshot["updatedBy"], "fred")

        self.store.set_check("crop-parsnip", False, "kayla")
        snapshot = self.store.snapshot()
        self.assertFalse(snapshot["checks"]["crop-parsnip"])
        self.assertEqual(snapshot["revision"], 2)


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