Test for multiple objects for class-based as_of

jazzband · Aug 1, 2014 · 7bfb406 · 7bfb406
1 parent 887d44d
commit 7bfb406
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/simple_history/tests/tests/test_manager.py b/simple_history/tests/tests/test_manager.py
@@ -79,3 +79,9 @@ def test_create_and_delete(self):
             doc_change.save()
         docs_as_of_tmw = models.Document.history.as_of(now + timedelta(days=1))
         self.assertFalse(list(docs_as_of_tmw))
+
+    def test_multiple(self):
+        document1 = models.Document.objects.create()
+        document2 = models.Document.objects.create()
+        historical = models.Document.history.as_of(datetime.now() + timedelta(days=1))
+        self.assertEqual(list(historical), [document1, document2])