From 65e4902f633a006333d4d50b92f1f1ee6a9a85eb Mon Sep 17 00:00:00 2001
From: Lee Miller <lee.miller@tutanota.com>
Date: Sun, 29 Sep 2024 17:04:34 +0300
Subject: [PATCH] Started a test case for the objects

---
 minode/tests/test_objects.py | 67 ++++++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)
 create mode 100644 minode/tests/test_objects.py

diff --git a/minode/tests/test_objects.py b/minode/tests/test_objects.py
new file mode 100644
index 0000000..c2c1cad
--- /dev/null
+++ b/minode/tests/test_objects.py
@@ -0,0 +1,67 @@
+"""Tests for the Inventory implementation"""
+import os
+import random
+import tempfile
+import time
+import unittest
+
+from minode import sql, shared, structure
+
+
+# + __bool__
+# + __contains__
+# + __getitem__
+# + __setitem__
+# cleanup
+# + get
+# filter
+# select
+# + vectors_to_send
+
+
+class TestObjectsSQL(unittest.TestCase):
+    """A test case for the sqlite inventory"""
+
+    @classmethod
+    def setUpClass(cls):
+        shared.data_directory = tempfile.gettempdir()
+        cls.objects = sql.Inventory()
+
+    def test_set_get(self):
+        """Put some objects and check presence and getting"""
+        obj = structure.Object(
+            int(time.time()), 42, 1, 1, object_payload=b'HELLO')
+        self.assertFalse(obj.vector in self.objects)
+        with self.assertRaises(KeyError):
+            self.objects[obj.vector]
+        self.assertIsNone(self.objects.get(obj.vector))
+        self.objects[obj.vector] = obj
+        self.assertTrue(self.objects)
+        self.assertTrue(obj.vector in self.objects)
+        obj1 = self.objects[obj.vector]
+        self.assertEqual(obj.vector, obj1.vector)
+        self.assertEqual(obj.data, obj1.data)
+
+    def test_vectors_to_send(self):
+        """Check vectors_to_send method"""
+        needed = set()
+        for _ in range(10):
+            # wrong stream
+            obj = structure.Object(
+                int(time.time() + 10), 42, 1, random.randint(1, 3),
+                object_payload=os.urandom(32))
+            self.objects[obj.vector] = obj
+            # expired
+            obj = structure.Object(
+                int(time.time() - 10), 42, 1, 4,
+                object_payload=os.urandom(32))
+            self.objects[obj.vector] = obj
+            # interesting
+            obj = structure.Object(
+                int(time.time() + 10), 42, 1, 4,
+                object_payload=os.urandom(32))
+            self.objects[obj.vector] = obj
+            needed.add(obj.vector)
+
+        self.assertEqual(set(self.objects.vectors_to_send(4)), needed)
+        self.assertTrue(set(self.objects.vectors_to_send()).difference(needed))