Testing if it is possible to iterate over queryset

saeschdivara · Sep 26, 2014 · e8d0a29 · e8d0a29
1 parent abfb84e
commit e8d0a29
Showing 1 changed file with 21 additions and 0 deletions.
diff --git a/arangodb/tests.py b/arangodb/tests.py
@@ -519,6 +519,27 @@ class TestModel(CollectionModel):
 
         TestModel.destroy()
 
+    def test_iterate_over_queryset(self):
+
+        class TestModel(CollectionModel):
+            pass
+
+        TestModel.init()
+
+        model1 = TestModel()
+        model1.save()
+
+        model2 = TestModel()
+        model2.save()
+
+        all_models = TestModel.objects.all()
+
+        for model in all_models:
+            self.assertTrue(isinstance(model, TestModel))
+
+        TestModel.destroy()
+
+
 
 class CollectionModelForeignKeyFieldTestCase(unittest.TestCase):
     def setUp(self):