Update Document model and fix tests

Author: nnethercott

meilisearch/models/document.py

@@ -2,17 +2,14 @@
 
 
 class Document:
-    __doc: Dict
-
     def __init__(self, doc: Dict[str, Any]) -> None:
-        self.__doc = doc
-        for key in doc:
-            setattr(self, key, doc[key])
+        self.__dict__.update(**doc)
 
     def __getattr__(self, attr: str) -> str:
-        if attr in self.__doc.keys():
-            return attr
-        raise AttributeError(f"{self.__class__.__name__} object has no attribute {attr}")
+        try:
+            return self.__dict__[attr]
+        except Exception as _:
+            raise AttributeError(f"{self.__class__.__name__} object has no attribute {attr}")
 
     def __iter__(self) -> Iterator:
         return iter(self.__dict__.items())

pyproject.toml

@@ -114,6 +114,7 @@ disable=[
   "too-few-public-methods",
   "line-too-long",
   "too-many-positional-arguments",
+  "raise-missing-from",
 ]
 enable=[
   "use-symbolic-message-instead",

tests/models/test_document.py

@@ -6,9 +6,15 @@
 from meilisearch.models.document import Document
 
 
+def test_doc_init():
+    d = {"field1": "test 1", "fiels2": "test 2"}
+    document = Document(d)
+    assert dict(document) == d
+
+
 def test_getattr():
     document = Document({"field1": "test 1", "fiels2": "test 2"})
-    assert document.__getattr__("field1") == "field1"
+    assert document.__getattr__("field1") == "test 1"
 
 
 def test_getattr_not_found():
@@ -18,11 +24,5 @@ def test_getattr_not_found():
 
 
 def test_iter():
-    # I wrote a test what what this does, but I have a feeling this isn't actually what it was
-    # expected to do when written as it doesn't really act like I would expect an iterator to act.
     document = Document({"field1": "test 1", "fiels2": "test 2"})
-
-    assert next(document.__iter__()) == (
-        "_Document__doc",
-        {"field1": "test 1", "fiels2": "test 2"},
-    )
+    assert list(iter(document)) == [("field1", "test 1"), ("fiels2", "test 2")]