diff --git a/tos/apps.py b/tos/apps.py index ac86781..789a0bd 100644 --- a/tos/apps.py +++ b/tos/apps.py @@ -19,4 +19,4 @@ class TOSConfig(AppConfig): dispatch_uid='invalidate_cached_agreements') # Create the TOS key version immediately - invalidate_cached_agreements(TermsOfService, None) + invalidate_cached_agreements(TermsOfService) diff --git a/tos/signal_handlers.py b/tos/signal_handlers.py index c29398c..800a0a5 100644 --- a/tos/signal_handlers.py +++ b/tos/signal_handlers.py @@ -6,7 +6,7 @@ from django.core.cache import caches cache = caches[getattr(settings, 'TOS_CACHE_NAME', 'default')] -def invalidate_cached_agreements(TermsOfService, instance, **kwargs): +def invalidate_cached_agreements(sender, **kwargs): if kwargs.get('raw', False): return diff --git a/tos/tests/test_middleware.py b/tos/tests/test_middleware.py index bacbd2e..4d25fde 100644 --- a/tos/tests/test_middleware.py +++ b/tos/tests/test_middleware.py @@ -167,14 +167,14 @@ class BumpCoverage(TestCase): def test_invalidate_cached_agreements(self): cache = caches[getattr(settings, 'TOS_CACHE_NAME', 'default')] - invalidate_cached_agreements(TermsOfService, {}) + invalidate_cached_agreements(TermsOfService) key_version = cache.get('django:tos:key_version') - invalidate_cached_agreements(TermsOfService, {}) + invalidate_cached_agreements(TermsOfService) self.assertEqual(cache.get('django:tos:key_version'), key_version+1) - invalidate_cached_agreements(TermsOfService, {}, raw=True) + invalidate_cached_agreements(TermsOfService, raw=True) self.assertEqual(cache.get('django:tos:key_version'), key_version+1)