def test_autopatching_twice(self): patch(django=True) # Call django.setup() twice and ensure we don't add a duplicate tracer import django django.setup() django.setup() from django.conf import settings found_app = 0 for app in settings.INSTALLED_APPS: if app == 'ddtrace.contrib.django': found_app += 1 eq_(found_app, 1) eq_(settings.MIDDLEWARE_CLASSES[0], 'ddtrace.contrib.django.TraceMiddleware') found_mw = 0 for mw in settings.MIDDLEWARE_CLASSES: if mw == 'ddtrace.contrib.django.TraceMiddleware': found_mw += 1 eq_(found_mw, 1)
def test_patch_raise_exception_manual_patch(self): # Manual patching should not be affected by the environment variable override. with self.assertRaises(monkey.ModuleNotFoundException) as me: monkey.patch(module_dne=True) assert ( "integration module ddtrace.contrib.module_dne does not exist, module will not have tracing available" in str(me.exception)) assert "module_dne" not in monkey._PATCHED_MODULES
def test_autopatching(self): patch(django=True) import django ok_(django._datadog_patch) django.setup() from django.conf import settings ok_('ddtrace.contrib.django' in settings.INSTALLED_APPS) eq_(settings.MIDDLEWARE_CLASSES[0], 'ddtrace.contrib.django.TraceMiddleware')
def test_autopatching_empty_middleware(self): with self.settings(MIDDLEWARE=[]): patch(django=True) django.setup() assert django._datadog_patch assert 'ddtrace.contrib.django' in settings.INSTALLED_APPS assert settings.MIDDLEWARE[ 0] == 'ddtrace.contrib.django.TraceMiddleware' # MIDDLEWARE_CLASSES gets created internally in django 1.10 & 1.11 but doesn't # exist at all in 2.0. assert not getattr(settings, 'MIDDLEWARE_CLASSES', None) or \ 'ddtrace.contrib.django.TraceMiddleware' \ not in settings.MIDDLEWARE_CLASSES assert settings.MIDDLEWARE[ -1] == 'ddtrace.contrib.django.TraceExceptionMiddleware' assert not getattr(settings, 'MIDDLEWARE_CLASSES', None) or \ 'ddtrace.contrib.django.TraceExceptionMiddleware' \ not in settings.MIDDLEWARE_CLASSES
def setUp(self): super(DjangoAutopatchTest, self).setUp() patch(django=True) django.setup()
def setUp(self): super(DjangoAutopatchTest, self).setUp() patch(django=True) django.setup()
def test_patch_all_env_override_manual_patch(self): # Manual patching should not be affected by the environment variable override. monkey.patch(sqlite3=True) assert "sqlite3" in monkey._PATCHED_MODULES