get_or_create function added to querySetJob

cacheback/queryset.py

@@ -38,6 +38,16 @@ def fetch(self, *args, **kwargs):
         return self.model.objects.get(**kwargs)
 
 
+class QuerySetGetOrCreateJob(QuerySetJob):
+    """
+    for ORM reads that use ``get_or_create`` method.
+    """
+
+    def fetch(self, *args, **kwargs):
+        model_object, created = self.model.objects.get_or_create(**kwargs)
+        return model_object
+
+
 class QuerySetFilterJob(QuerySetJob):
     """
     For ORM reads that use the ``filter`` method.

tests/queryset_tests.py

@@ -2,18 +2,17 @@
 from django.test import TestCase
 
 from cacheback.base import Job
-from cacheback.queryset import QuerySetFilterJob, QuerySetGetJob
+from cacheback.queryset import (
+    QuerySetFilterJob, QuerySetGetJob, QuerySetGetOrCreateJob)
 from tests.dummyapp import models
 
 
 class ManualQuerySetJob(Job):
-
     def fetch(self, name):
         return models.DummyModel.objects.filter(name=name)
 
 
 class TestManualQuerySetJob(TestCase):
-
     def setUp(self):
         self.job = ManualQuerySetJob()
         models.DummyModel.objects.create(name="Alan")
@@ -34,7 +33,6 @@ def test_makes_only_one_database_query(self):
 
 
 class TestFilterQuerySetJob(TestCase):
-
     def setUp(self):
         self.job = QuerySetFilterJob(models.DummyModel)
         models.DummyModel.objects.create(name="Alan")
@@ -50,7 +48,6 @@ def test_returns_result_on_first_call(self):
 
 
 class TestGetQuerySetJob(TestCase):
-
     def setUp(self):
         self.job = QuerySetGetJob(models.DummyModel)
         models.DummyModel.objects.create(name="Alan")
@@ -65,13 +62,26 @@ def test_returns_result_on_first_call(self):
         self.assertEqual('Alan', result.name)
 
 
+class TestGetOrCreateQuerySetJob(TestCase):
+    def setUp(self):
+        self.job = QuerySetGetOrCreateJob(models.DummyModel)
+        models.DummyModel.objects.create(name="Barry")
+
+    def tearDown(self):
+        models.DummyModel.objects.all().delete()
+        cache.clear()
+
+    def test_returns_result_on_first_call(self):
+        result = self.job.get(name='Alan')
+        self.assertEqual('Alan', result.name)
+
+
 class EchoJob(Job):
     def fetch(self, *args, **kwargs):
         return (args, kwargs)
 
 
 class TestEdgeCases(TestCase):
-
     def setUp(self):
         self.job = EchoJob()