Add object relation store impl and test case

test/test_object.py

@@ -17,25 +17,56 @@ handler.setFormatter(formatter)
 root.addHandler(handler)
 
 
-from knowledge import ObjectID, HashValue, EmailObjectBuilder
+from knowledge import (
+    ObjectID,
+    HashValue,
+    EmailObjectBuilder,
+    ObjectRelationStore,
+    KnowledgeStore,
+)
 import asyncio
 import unittest
 
 
 class TestVectorSTorage(unittest.TestCase):
     def test_object(self):
-        data = HashValue.hash_data("1233".encode("utf-8"));
+        data = HashValue.hash_data("1233".encode("utf-8"))
         print(data.to_base58())
         print(data.to_base36())
-        
+
         data2 = HashValue.from_base58(data.to_base58())
         self.assertEqual(data.to_base36(), data2.to_base36())
-        
+
         data2 = HashValue.from_base36(data.to_base36())
         self.assertEqual(data.to_base58(), data2.to_base58())
 
         email_folder = "F:\\system\\Downloads\\8081ffdb80925f5bff9c6ab9c4756c7d"
         email_object = EmailObjectBuilder({}, email_folder).build()
-        
+
+    def test_relation(self):
+        obj1 = ObjectID.hash_data("12345".encode("utf-8"))
+        obj2 = ObjectID.hash_data("67890".encode("utf-8"))
+        obj3 = ObjectID.hash_data("abcde".encode("utf-8"))
+        obj4 = ObjectID.hash_data("fghij".encode("utf-8"))
+        print(obj1.to_base58(), obj2.to_base58(), obj3.to_base58())
+        relation_store = KnowledgeStore().get_relation_store()
+        relation_store.add_relation(obj1, obj2)
+        relation_store.add_relation(obj1, obj2)
+        relation_store.add_relation(obj2, obj3)
+
+        relation_store.add_relation(obj1, obj3)
+        relation_store.add_relation(obj1, obj4)
+
+        objs = relation_store.get_related_objects(obj2)
+        self.assertEqual(len(objs), 1)
+        self.assertEqual(objs[0], obj3)
+
+        objs = relation_store.get_related_root_objects(obj1)
+        self.assertEqual(len(objs), 2)
+        self.assertEqual(obj3 in objs, True)
+        self.assertEqual(obj4 in objs, True)
+        # self.assertCountEqual(objs, [obj3, obj4])
+
+
 if __name__ == "__main__":
     unittest.main()