gridfs tests

Author: averagepythonfan

tests/conftest.py

@@ -9,3 +9,11 @@
 def mongomv_client():
     cl = MongoMVClient(uri=TEST_MONGO_URI)
     yield cl
+
+
+@pytest.fixture(scope="module")
+def model():
+    client = MongoMVClient(uri=TEST_MONGO_URI)
+    md = client.create_model(name="test_model", tags=["testing", "pytest"])
+    yield md
+    md.delete()

tests/test_gridfs.py

@@ -0,0 +1,30 @@
+import os
+from pathlib import Path
+from mongomv.schemas import ModelEntity
+import pytest
+
+
+@pytest.mark.usefixtures("model")
+class TestGridFS:
+
+    def test_dump_model(self, model: ModelEntity):
+        path = Path(os.getcwd()).joinpath("text.txt")
+        with open(path, "w") as file:
+            file.write("This is test case for GridFS")
+        result = model.dump_model(model_path=path, filename="text.txt")
+        assert type(result) == str
+        assert model.serialized_model is not None
+        os.remove(path=path)
+
+
+    def test_load_model(self, model: ModelEntity):
+        path = Path(os.getcwd()).joinpath("text.txt")
+        result = model.load_model(model_path=path)
+        assert type(result) == str
+        assert path.exists()
+        os.remove(path=path)
+
+
+    def test_delete_serialized_model(self, model: ModelEntity):
+        result = model.delete_model()
+        assert type(result) == str