Add tests for Django's get_or_create.

Related to issue #97.
rbarrois · Jul 7, 2020 · a7742f0 · a7742f0
1 parent 2351383
commit a7742f0
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tests/test_django.py b/tests/test_django.py
@@ -261,3 +261,17 @@ def test_db_interaction(self):
 
         obj2 = models.VersionModel.objects.get(pk=o2.pk)
         self.assertEqual(o2.version, obj2.version)
+
+    def test_get_or_create(self):
+        o1, created = models.VersionModel.objects.get_or_create(version=Version('0.1.1'), spec=SimpleSpec('==0.4.3'))
+        self.assertTrue(created)
+        self.assertIsNotNone(o1.pk)
+        self.assertEqual(Version('0.1.1'), o1.version)
+        self.assertEqual(SimpleSpec('==0.4.3'), o1.spec)
+
+        o2, created = models.VersionModel.objects.get_or_create(version=Version('0.1.1'), spec=SimpleSpec('==0.4.3'))
+        self.assertFalse(created)
+        self.assertEqual(o1, o2)
+        self.assertEqual(o1.pk, o2.pk)
+        self.assertEqual(Version('0.1.1'), o2.version)
+        self.assertEqual(SimpleSpec('==0.4.3'), o2.spec)