queryset_tests.py

from django.test import TestCase
from django.core.cache import cache

from cacheback.base import Job
from cacheback.queryset import QuerySetFilterJob, QuerySetGetJob
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")
        models.DummyModel.objects.create(name="Barry")

    def tearDown(self):
        models.DummyModel.objects.all().delete()
        cache.clear()

    def test_returns_result_on_first_call(self):
        results = self.job.get('Alan')
        self.assertEqual(1, len(results))

    def test_makes_only_one_database_query(self):
        with self.assertNumQueries(1):
            for _ in xrange(10):
                self.job.get('Alan')


class TestFilterQuerySetJob(TestCase):

    def setUp(self):
        self.job = QuerySetFilterJob(models.DummyModel)
        models.DummyModel.objects.create(name="Alan")
        models.DummyModel.objects.create(name="Barry")

    def tearDown(self):
        models.DummyModel.objects.all().delete()
        cache.clear()

    def test_returns_result_on_first_call(self):
        results = self.job.get(name='Alan')
        self.assertEqual(1, len(results))


class TestGetQuerySetJob(TestCase):

    def setUp(self):
        self.job = QuerySetGetJob(models.DummyModel)
        models.DummyModel.objects.create(name="Alan")
        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()

    def tearDown(self):
        cache.clear()

    def test_unhashable_arg_raises_exception(self):
        with self.assertRaises(RuntimeError):
            self.job.get({})

    def test_unhashable_kwarg_raises_exception(self):
        with self.assertRaises(RuntimeError):
            self.job.get(name={})