mirror of
https://github.com/jazzband/django-defender.git
synced 2026-03-16 22:10:32 +00:00
24 lines
650 B
Python
24 lines
650 B
Python
from django.test.testcases import TestCase, TransactionTestCase
|
|
|
|
from .connection import get_redis_connection
|
|
|
|
|
|
class DefenderTestCaseMixin:
|
|
"""Mixin used to provide a common tearDown method"""
|
|
|
|
def tearDown(self):
|
|
"""cleanup django-defender cache after each test"""
|
|
super().tearDown()
|
|
get_redis_connection().flushdb()
|
|
|
|
|
|
class DefenderTransactionTestCase(DefenderTestCaseMixin, TransactionTestCase):
|
|
"""Helper TransactionTestCase that cleans the cache after each test"""
|
|
|
|
pass
|
|
|
|
|
|
class DefenderTestCase(DefenderTestCaseMixin, TestCase):
|
|
"""Helper TestCase that cleans the cache after each test"""
|
|
|
|
pass
|