Add tests to dry URL

rframe/data_accessor.py

@@ -32,6 +32,7 @@ def __init__(self, schema, datasource, initdb=False):
         self.schema = schema
         self.storage = datasource
 
+
         for name in dir(self):
 
             if name.startswith("__"):
@@ -50,7 +51,7 @@ def __init__(self, schema, datasource, initdb=False):
             )
             method = makefun.create_function(signature, impl, func_name=name)
             setattr(self, name[1:], method)
-            
+
         if initdb:
             self.initdb()
 
@@ -108,7 +109,7 @@ def _min(self, fields=None, **labels) -> Any:
             fields = list(self.schema.__fields__)
         elif isinstance(fields, str):
             fields = [fields]
-        
+
         labels = {k: v for k, v in labels.items() if v is not None}
         query = self.schema.compile_query(self.storage, **labels)
 
@@ -162,10 +163,10 @@ def _count(self, **labels):
     def insert(self, docs, raise_on_error=True, dry=False):
         if not self.initialized:
             self.initdb()
-            
+
         if not isinstance(docs, (list, tuple)):
             docs = [docs]
-        
+
         res = {
             "success": [],
             "failed": [],

tests/test_schemas.py

@@ -78,7 +78,7 @@ def insert_data(
 
         PERMISSIONS["insert"] = True
         db.insert(docs)
- 
+
         PERMISSIONS["update"] = False
         with tester.assertRaises(UpdateError):
             db.insert(docs)
@@ -90,6 +90,14 @@ def delete_data(
         db.delete(docs)
         tester.assertEqual(db.count(), 0)
 
+    @classmethod
+    def dry_insert_data(
+        cls, tester: unittest.TestCase, db, docs: List["BaseTestSchema"]
+    ):
+        before_insert = db.count()
+        db.insert(docs, dry=True)
+        tester.assertEqual(db.count(), before_insert)
+
     @classmethod
     def basic_tests(
         cls, tester: unittest.TestCase, db, docs: List["BaseTestSchema"]
@@ -145,6 +153,7 @@ def test(cls, tester: unittest.TestCase, db, docs: List["BaseTestSchema"]):
         cls.basic_tests(tester, db, docs)
         cls.frame_test(tester, db, docs)
         cls.delete_data(tester, db, docs)
+        cls.dry_insert_data(tester, db, docs)
 
 
 class SimpleSchema(BaseTestSchema):
@@ -219,6 +228,7 @@ def basic_tests(cls, tester, db, docs: List["InterpolatingSchema"]):
             ratio = doc.value / value
             tester.assertAlmostEqual(ratio, 1, delta=1e-2)
 
+
     @classmethod
     def test(cls, tester, db, docs: List["BaseTestSchema"]):
         cls.insert_data(tester, db, docs)