def get_tracer_config(service_name):
if config.TRACING_CONFIG_FILENAME is not None:
tracing_config = load_yaml_config(config.TRACING_CONFIG_FILENAME)
else:
tracing_config = DEFAULT_TRACER_CONFIG
return jaeger_client.Config(config=tracing_config,
service_name=service_name)
def _check_valid_yaml(self, yamldata: str):
with temp_file_containing(yamldata.encode()) as fp:
filename = fp.name
loaded = load_yaml_config(filename)
assert 'api' in loaded
assert 'number' in loaded['api']
assert loaded['api']['number'] == 42
assert loaded['api']['ingress']['enabled']
return loaded
def setup_logging(default_path='default_logging.yaml'):
"""
Setup logging configuration
"""
if config.LOG_CONFIG_FILENAME is not None:
path = Path(config.LOG_CONFIG_FILENAME)
else:
path = Path(__file__).parent / default_path
loggingconfig = load_yaml_config(path)
logging.config.dictConfig(loggingconfig)
# Configure Structlog wrapper for client use
setup_structlog()
logging.debug(f"Loaded logging config from file: {path}")
def test_random_bytes(self):
with temp_file_containing(generate_bytes(128)) as fp:
filename = fp.name
with pytest.raises(InvalidConfiguration):
load_yaml_config(filename)
def test_missing_file(self):
filename = 'unlikely a valid file'
with pytest.raises(InvalidConfiguration):
load_yaml_config(filename)
def test_list(self):
with temp_file_containing(b'[1,2,3]') as fp:
filename = fp.name
assert [1, 2, 3] == load_yaml_config(filename)
def test_empty(self):
with temp_file_containing(b'') as fp:
filename = fp.name
assert None == load_yaml_config(filename)