updates

coned_rtu/tests/store_test.py

@@ -0,0 +1,52 @@
+from datetime import datetime, timedelta, timezone
+import unittest
+
+from coned_rtu import Reading, BucketList, OverlappingReadingError
+
+
+class TestStore(unittest.TestCase):
+    def setUp(self):
+        self.sometime = datetime(2010, 12, 25, 14, 30, 0, tzinfo=timezone.utc)
+        self.later = datetime(2010, 12, 26, 14, 30, 0, tzinfo=timezone.utc)
+        self.hour = timedelta(hours=1)
+        self.day = timedelta(hours=24)
+
+    def test(self):
+        bl = BucketList()
+        t1 = self.sometime
+        t2 = t1 + self.hour
+        t3 = t2 + self.hour
+        t4 = t3 + self.day
+
+        b = Reading(t2, t3, "wh", 1)
+        bl.insert(b)
+
+        a = Reading(t1, t2, "wh", 1)
+        bl.insert(a)
+
+        # inserting an overlapping reading is an error
+        bad = Reading(t2, t2 + timedelta(minutes=30), "wh", 1)
+        with self.assertRaises(OverlappingReadingError):
+            bl.insert(bad)
+
+        # inserting an identical reading results in no change
+        bl.insert(a)
+
+        # inserted readings are sorted
+        got = bl.get_bucket(a.hash_bucket())
+        want = [a, b]
+        self.assertEqual(got, want)
+
+        # to_list returns a list in sorted order
+        c = Reading(t3, t4, "wh", 3)
+        bl.insert(c)
+        got = bl.to_list()
+        want = [a, b, c]
+        self.assertEqual(got, want)
+
+    def test_get_bucket_makes_bucket_if_not_exist(self):
+        bl = BucketList()
+        day = "2021-08-30"
+        _ = bl.get_bucket(day)
+        self.assertEqual(len(bl._dict), 1)
+        self.assertEqual(list(bl._dict.keys())[0], day)