def test_convert_to_import_error(self):
from opencensus.trace.ext.django import config
module_name = 'test_module'
class_name = 'test_class'
path = '{}.{}'.format(module_name, class_name)
with self.assertRaises(ImportError):
config.convert_to_import(path)
def __init__(self, get_response=None):
# One-time configuration and initialization.
self.get_response = get_response
self._sampler = settings.SAMPLER
self._exporter = settings.EXPORTER
self._propagator = settings.PROPAGATOR
self._blacklist_paths = settings.params.get(BLACKLIST_PATHS)
# Initialize the sampler
if self._sampler.__name__ == 'ProbabilitySampler':
_rate = settings.params.get(SAMPLING_RATE,
probability.DEFAULT_SAMPLING_RATE)
self.sampler = self._sampler(_rate)
else:
self.sampler = self._sampler()
# Initialize the exporter
transport = convert_to_import(settings.params.get(TRANSPORT))
if self._exporter.__name__ == 'GoogleCloudExporter':
_project_id = settings.params.get(GCP_EXPORTER_PROJECT, None)
self.exporter = self._exporter(project_id=_project_id,
transport=transport)
elif self._exporter.__name__ == 'ZipkinExporter':
_service_name = self._get_service_name(settings.params)
_zipkin_host_name = settings.params.get(ZIPKIN_EXPORTER_HOST_NAME,
'localhost')
_zipkin_port = settings.params.get(ZIPKIN_EXPORTER_PORT, 9411)
_zipkin_protocol = settings.params.get(ZIPKIN_EXPORTER_PROTOCOL,
'http')
self.exporter = self._exporter(service_name=_service_name,
host_name=_zipkin_host_name,
port=_zipkin_port,
protocol=_zipkin_protocol,
transport=transport)
elif self._exporter.__name__ == 'TraceExporter':
_service_name = self._get_service_name(settings.params)
_endpoint = settings.params.get(OCAGENT_TRACE_EXPORTER_ENDPOINT,
None)
self.exporter = self._exporter(service_name=_service_name,
endpoint=_endpoint,
transport=transport)
elif self._exporter.__name__ == 'JaegerExporter':
_service_name = self._get_service_name(settings.params)
self.exporter = self._exporter(service_name=_service_name,
transport=transport)
else:
self.exporter = self._exporter(transport=transport)
self.blacklist_hostnames = settings.params.get(BLACKLIST_HOSTNAMES,
None)
# Initialize the propagator
self.propagator = self._propagator()
def __init__(self, get_response=None):
# One-time configuration and initialization.
self.get_response = get_response
self._sampler = settings.SAMPLER
self._exporter = settings.EXPORTER
self._propagator = settings.PROPAGATOR
self._blacklist_paths = settings.params.get(BLACKLIST_PATHS)
# Initialize the sampler
if self._sampler.__name__ == 'ProbabilitySampler':
_rate = settings.params.get(
SAMPLING_RATE, probability.DEFAULT_SAMPLING_RATE)
self.sampler = self._sampler(_rate)
else:
self.sampler = self._sampler()
# Initialize the exporter
transport = convert_to_import(settings.params.get(TRANSPORT))
if self._exporter.__name__ == 'GoogleCloudExporter':
_project_id = settings.params.get(GCP_EXPORTER_PROJECT, None)
self.exporter = self._exporter(
project_id=_project_id,
transport=transport)
elif self._exporter.__name__ == 'ZipkinExporter':
_zipkin_service_name = settings.params.get(
ZIPKIN_EXPORTER_SERVICE_NAME, 'my_service')
_zipkin_host_name = settings.params.get(
ZIPKIN_EXPORTER_HOST_NAME, 'localhost')
_zipkin_port = settings.params.get(
ZIPKIN_EXPORTER_PORT, 9411)
self.exporter = self._exporter(
service_name=_zipkin_service_name,
host_name=_zipkin_host_name,
port=_zipkin_port,
transport=transport)
else:
self.exporter = self._exporter(transport=transport)
# Initialize the propagator
self.propagator = self._propagator()
def test_convert_to_import(self):
from opencensus.trace.ext.django import config
module_name = 'module'
class_name = 'class'
path = '{}.{}'.format(module_name, class_name)
mock_importlib = mock.Mock()
mock_import_module = mock.Mock()
mock_module = mock.Mock()
mock_class = mock.Mock()
setattr(mock_module, class_name, mock_class)
mock_import_module.return_value = mock_module
mock_importlib.import_module = mock_import_module
patch = mock.patch('opencensus.trace.ext.django.config.importlib',
mock_importlib)
with patch:
result = config.convert_to_import(path)
self.assertEqual(result, mock_class)
