chore: more tests

tests/parser/test_schema.py

@@ -0,0 +1,58 @@
+import pytest
+
+from application.parser.schema.schema import BaseDocument
+
+
+class ConcreteDoc(BaseDocument):
+    @classmethod
+    def get_type(cls) -> str:
+        return "test"
+
+
+@pytest.mark.unit
+class TestBaseDocument:
+
+    def test_get_text(self):
+        doc = ConcreteDoc(text="hello")
+        assert doc.get_text() == "hello"
+
+    def test_get_text_raises_when_none(self):
+        doc = ConcreteDoc()
+        with pytest.raises(ValueError, match="text field not set"):
+            doc.get_text()
+
+    def test_get_doc_id(self):
+        doc = ConcreteDoc(text="x", doc_id="doc1")
+        assert doc.get_doc_id() == "doc1"
+
+    def test_get_doc_id_raises_when_none(self):
+        doc = ConcreteDoc(text="x")
+        with pytest.raises(ValueError, match="doc_id not set"):
+            doc.get_doc_id()
+
+    def test_is_doc_id_none(self):
+        doc = ConcreteDoc(text="x")
+        assert doc.is_doc_id_none is True
+
+    def test_is_doc_id_not_none(self):
+        doc = ConcreteDoc(text="x", doc_id="y")
+        assert doc.is_doc_id_none is False
+
+    def test_get_embedding(self):
+        doc = ConcreteDoc(text="x", embedding=[1.0, 2.0])
+        assert doc.get_embedding() == [1.0, 2.0]
+
+    def test_get_embedding_raises_when_none(self):
+        doc = ConcreteDoc(text="x")
+        with pytest.raises(ValueError, match="embedding not set"):
+            doc.get_embedding()
+
+    def test_extra_info_str(self):
+        doc = ConcreteDoc(text="x", extra_info={"key": "value", "num": 42})
+        result = doc.extra_info_str
+        assert "key: value" in result
+        assert "num: 42" in result
+
+    def test_extra_info_str_none(self):
+        doc = ConcreteDoc(text="x")
+        assert doc.extra_info_str is None