def test_django_db_connection_is_closed(self): from django.db import connection connection._was_closed = False old_connection_close = connection.close def monkeypatched_connection_close(*args, **kwargs): connection._was_closed = True return old_connection_close(*args, **kwargs) connection.close = monkeypatched_connection_close try: jail(gen_unique_id(), get_db_connection.name, [2], {}) self.assertTrue(connection._was_closed) finally: connection.close = old_connection_close
def test_django_cache_connection_is_closed(self): from django.core import cache old_cache_close = getattr(cache.cache, "close", None) cache._was_closed = False old_cache_parse_backend = getattr(cache, "parse_backend_uri", None) if old_cache_parse_backend: # checks to make sure attr exists delattr(cache, 'parse_backend_uri') def monkeypatched_cache_close(*args, **kwargs): cache._was_closed = True cache.cache.close = monkeypatched_cache_close jail(gen_unique_id(), mytask.name, [4], {}) self.assertTrue(cache._was_closed) cache.cache.close = old_cache_close if old_cache_parse_backend: cache.parse_backend_uri = old_cache_parse_backend
def test_django_cache_connection_is_closed_django_1_1(self): from django.core import cache old_cache_close = getattr(cache.cache, "close", None) cache._was_closed = False old_cache_parse_backend = getattr(cache, "parse_backend_uri", None) cache.parse_backend_uri = lambda uri: ["libmemcached", "1", "2"] def monkeypatched_cache_close(*args, **kwargs): cache._was_closed = True cache.cache.close = monkeypatched_cache_close jail(gen_unique_id(), mytask.name, [4], {}) self.assertTrue(cache._was_closed) cache.cache.close = old_cache_close if old_cache_parse_backend: cache.parse_backend_uri = old_cache_parse_backend else: del(cache.parse_backend_uri)
def test_django_cache_connection_is_closed(self): old_cache_close = getattr(cache.cache, "close", None) old_backend = cache.settings.CACHE_BACKEND cache.settings.CACHE_BACKEND = "libmemcached" cache._was_closed = False old_cache_parse_backend = getattr(cache, "parse_backend_uri", None) if old_cache_parse_backend: # checks to make sure attr exists delattr(cache, 'parse_backend_uri') def monkeypatched_cache_close(*args, **kwargs): cache._was_closed = True cache.cache.close = monkeypatched_cache_close jail(gen_unique_id(), mytask.name, [4], {}) self.assertTrue(cache._was_closed) cache.cache.close = old_cache_close cache.settings.CACHE_BACKEND = old_backend if old_cache_parse_backend: cache.parse_backend_uri = old_cache_parse_backend
def test_django_cache_connection_is_closed_django_1_1(self): old_cache_close = getattr(cache.cache, "close", None) old_backend = cache.settings.CACHE_BACKEND cache.settings.CACHE_BACKEND = "libmemcached" cache._was_closed = False old_cache_parse_backend = getattr(cache, "parse_backend_uri", None) cache.parse_backend_uri = lambda uri: ["libmemcached", "1", "2"] def monkeypatched_cache_close(*args, **kwargs): cache._was_closed = True cache.cache.close = monkeypatched_cache_close jail(gen_unique_id(), mytask.name, [4], {}) self.assertTrue(cache._was_closed) cache.cache.close = old_cache_close cache.settings.CACHE_BACKEND = old_backend if old_cache_parse_backend: cache.parse_backend_uri = old_cache_parse_backend else: del (cache.parse_backend_uri)