• Search by Module
• Search by Words
• Search Projects

• Java
• Python
• JavaScript
• TypeScript
• C++
• Scala
• Blog

• django-cachalot-master
  • .github
    • PULL_REQUEST_TEMPLATE.md
    • ISSUE_TEMPLATE
      • feature.md
      • bug.md
      • question.md
      • improvement.md
    • workflows
      • publish.yml
  • cachalot
    • apps.py
    • cache.py
    • panels.py
    • templates
      • cachalot
        • panel.html
    • jinja2ext.py
    • templatetags
      • __init__.py
      • cachalot.py
    • models.py
    • transaction.py
    • api.py
    • __init__.py
    • utils.py
    • tests
      • db_router.py
      • postgres.py
      • test_utils.py
      • write.py
      • loaddata_fixture.json
      • multi_db.py
      • models.py
      • read.py
      • transaction.py
      • api.py
      • thread_safety.py
      • __init__.py
      • migrations
        • 0001_initial.py
        • __init__.py
      • debug_toolbar.py
      • settings.py
      • signals.py
    • settings.py
    • signals.py
    • monkey_patch.py
    • management
      • commands
        • invalidate_cachalot.py
        • __init__.py
      • __init__.py
  • LICENSE
  • travis-matrix.py
  • runtests_requirements.txt
  • README.rst
  • setup.py
  • CHANGELOG.rst
  • runtests_urls.py
  • runtests.py
  • .travis.yml
  • CONTRIBUTING.rst
  • settings.py
  • requirements.txt
  • benchmark_requirements.txt
  • benchmark.py
  • .gitignore
  • docs
    • limits.rst
    • todo.rst
    • Makefile
    • introduction.rst
    • api.rst
    • benchmark.rst
    • how.rst
    • reporting.rst
    • legacy.rst
    • quickstart.rst
    • requirements.txt
    • index.rst
    • conf.py
    • changelog.rst
  • MANIFEST.in
  • benchmark
    • docs
      • 2018-08-09
        • cache_memcached.svg
        • cache_locmem.svg
        • cache_pylibmc.svg
        • db_sqlite.svg
        • cache.svg
        • db_results.rst
        • cache_results.rst
        • conditions.rst
        • db_postgresql.svg
        • db.svg
        • cache_redis.svg
        • cache_filebased.svg
        • db_mysql.svg
  • tox.ini
  • .coveralls.yml

from unittest import skipIf

from django.conf import settings
from django.contrib.auth.models import User
from django.db import DEFAULT_DB_ALIAS, transaction
from django.test import TransactionTestCase

from ..api import invalidate
from ..signals import post_invalidation

from .models import Test


class SignalsTestCase(TransactionTestCase):
    databases = set(settings.DATABASES.keys())

    def test_table_invalidated(self):
        l = []

        def receiver(sender, **kwargs):
            db_alias = kwargs['db_alias']
            l.append((sender, db_alias))

        post_invalidation.connect(receiver)
        self.assertListEqual(l, [])
        list(Test.objects.all())
        self.assertListEqual(l, [])
        Test.objects.create(name='test1')
        self.assertListEqual(l, [('cachalot_test', DEFAULT_DB_ALIAS)])
        post_invalidation.disconnect(receiver)

        del l[:]  # Empties the list
        post_invalidation.connect(receiver, sender=User._meta.db_table)
        Test.objects.create(name='test2')
        self.assertListEqual(l, [])
        User.objects.create_user('user')
        self.assertListEqual(l, [('auth_user', DEFAULT_DB_ALIAS)])
        post_invalidation.disconnect(receiver, sender=User._meta.db_table)

    def test_table_invalidated_in_transaction(self):
        """
        Checks that the ``post_invalidation`` signal is triggered only after
        the end of a transaction.
        """
        l = []

        def receiver(sender, **kwargs):
            db_alias = kwargs['db_alias']
            l.append((sender, db_alias))

        post_invalidation.connect(receiver)

        self.assertListEqual(l, [])
        with transaction.atomic():
            Test.objects.create(name='test1')
            self.assertListEqual(l, [])
        self.assertListEqual(l, [('cachalot_test', DEFAULT_DB_ALIAS)])

        del l[:]  # Empties the list
        self.assertListEqual(l, [])
        with transaction.atomic():
            Test.objects.create(name='test2')
            with transaction.atomic():
                Test.objects.create(name='test3')
                self.assertListEqual(l, [])
            self.assertListEqual(l, [])
        self.assertListEqual(l, [('cachalot_test', DEFAULT_DB_ALIAS)])

        post_invalidation.disconnect(receiver)

    def test_table_invalidated_once_per_transaction_or_invalidate(self):
        """
        Checks that the ``post_invalidation`` signal is triggered only after
        the end of a transaction.
        """
        l = []

        def receiver(sender, **kwargs):
            db_alias = kwargs['db_alias']
            l.append((sender, db_alias))

        post_invalidation.connect(receiver)

        self.assertListEqual(l, [])
        with transaction.atomic():
            Test.objects.create(name='test1')
            self.assertListEqual(l, [])
            Test.objects.create(name='test2')
            self.assertListEqual(l, [])
        self.assertListEqual(l, [('cachalot_test', DEFAULT_DB_ALIAS)])

        del l[:]  # Empties the list
        self.assertListEqual(l, [])
        invalidate(Test, db_alias=DEFAULT_DB_ALIAS)
        self.assertListEqual(l, [('cachalot_test', DEFAULT_DB_ALIAS)])

        del l[:]  # Empties the list
        self.assertListEqual(l, [])
        with transaction.atomic():
            invalidate(Test, db_alias=DEFAULT_DB_ALIAS)
            self.assertListEqual(l, [])
        self.assertListEqual(l, [('cachalot_test', DEFAULT_DB_ALIAS)])

        post_invalidation.disconnect(receiver)

    @skipIf(len(settings.DATABASES) == 1,
            'We can’t change the DB used since there’s only one configured')
    def test_table_invalidated_multi_db(self):
        db_alias2 = next(alias for alias in settings.DATABASES
                         if alias != DEFAULT_DB_ALIAS)
        l = []

        def receiver(sender, **kwargs):
            db_alias = kwargs['db_alias']
            l.append((sender, db_alias))

        post_invalidation.connect(receiver)
        self.assertListEqual(l, [])
        Test.objects.using(DEFAULT_DB_ALIAS).create(name='test')
        self.assertListEqual(l, [('cachalot_test', DEFAULT_DB_ALIAS)])
        Test.objects.using(db_alias2).create(name='test')
        self.assertListEqual(l, [
            ('cachalot_test', DEFAULT_DB_ALIAS),
            ('cachalot_test', db_alias2)])
        post_invalidation.disconnect(receiver)