IterableResult must be indexable

Author: explorigin

pydanticrud/main.py

@@ -26,6 +26,9 @@ def __len__(self):
     def __iter__(self):
         return self
 
+    def __getitem__(self, indices):
+        return self.records.__getitem__(indices)
+
     def __next__(self):
         try:
             member = self.records[self._current_index]

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "pydanticrud"
-version = "0.1.1"
+version = "0.2.0"
 description = "Supercharge your Pydantic models with CRUD methods and a pluggable backend"
 authors = ["Timothy Farrell <[email protected]>"]
 license = "MIT"

tests/test_model.py

@@ -11,6 +11,10 @@ def __init__(self, cfg):
     def get(cls, id):
         pass
 
+    @classmethod
+    def query(cls, id):
+        pass
+
 
 class Model(BaseModel):
     id: int
@@ -38,5 +42,19 @@ def test_model_backend_get():
         assert m.total == 3.0
 
 
+def test_model_backend_query():
+    with patch.object(
+        FalseBackend, "query", return_value=[dict(id=1, name="two", total=3.0)]
+    ) as mock_query:
+        m = Model.query(2)
+
+        mock_query.assert_called_with(2)
+        assert m.count is None  # In this case the backend did not provide a total count.
+        assert len(m[:]) == 1  # .. but we can cast to a list and get that length.
+        assert m[0].id == 1
+        assert m[0].name == "two"
+        assert m[0].total == 3.0
+
+
 def test_model_table_name_from_title():
     assert Model.get_table_name() == Model.Config.title.lower()