def test_credentials_token_noverify():
expected_headers = {'Authorization': 'Bearer mytoken'}
creds = KubeletCredentials({
"verify_tls": "false",
"ca_cert": "ca_cert",
"client_crt": "ignore_me",
"token": "mytoken"
})
assert creds.verify() is False
assert creds.cert_pair() is None
assert creds.headers("https://dummy") == expected_headers
# Make sure we don't leak the token over http
assert creds.headers("http://dummy") is None
scraper = OpenMetricsBaseCheck('prometheus', {}, {})
scraper_config = scraper.create_scraper_configuration()
scraper_config['prometheus_url'] = 'https://dummy'
creds.configure_scraper(scraper_config)
assert scraper_config['ssl_ca_cert'] is False
assert scraper_config['ssl_cert'] is None
assert scraper_config['ssl_private_key'] is None
assert scraper_config['extra_headers'] == expected_headers
# Make sure we don't leak the token over http
scraper_config['prometheus_url'] = "http://dummy"
creds.configure_scraper(scraper_config)
assert scraper_config['ssl_ca_cert'] is False
assert scraper_config['ssl_cert'] is None
assert scraper_config['ssl_private_key'] is None
assert scraper_config['extra_headers'] == {}
def test_bearer_token_auto_http():
endpoint = "http://localhost:12345/metrics"
instance = {
'prometheus_url': endpoint,
'namespace': 'default_namespace',
'bearer_token_auth': 'tls_only'
}
with patch.object(OpenMetricsBaseCheck, 'KUBERNETES_TOKEN_PATH',
os.path.join(FIXTURE_PATH, 'default_token')):
check = OpenMetricsBaseCheck('prometheus_check', {}, {}, [instance])
assert check.get_scraper_config(instance)['_bearer_token'] is None
def test_get_default_kubernetes_bearer_token():
endpoint = "none"
instance = {
'prometheus_url': endpoint,
'namespace': 'default_namespace',
'bearer_token_auth': True
}
with patch.object(OpenMetricsBaseCheck, 'KUBERNETES_TOKEN_PATH',
os.path.join(FIXTURE_PATH, 'default_token')):
check = OpenMetricsBaseCheck('prometheus_check', {}, {}, [instance])
assert check.get_scraper_config(
instance)['_bearer_token'] == 'my default token'
def test_bearer_token_disabled():
endpoint = "none"
custom_path = os.path.join(FIXTURE_PATH, 'custom_token')
instance = {
'prometheus_url': endpoint,
'namespace': 'default_namespace',
'bearer_token_auth': False,
'bearer_token_path': custom_path,
}
with patch.object(OpenMetricsBaseCheck, 'KUBERNETES_TOKEN_PATH',
os.path.join(FIXTURE_PATH, 'default_token')):
check = OpenMetricsBaseCheck('prometheus_check', {}, {}, [instance])
assert check.get_scraper_config(instance)['_bearer_token'] is None
def test_label_to_hostname_override():
endpoint = "none"
default_instance = {
'default_namespace': {
'prometheus_url': endpoint,
'metrics': [{"test_rate": "test.rate"}],
'label_to_hostname': 'node',
}
}
instance = {'prometheus_url': endpoint, 'namespace': 'default_namespace'}
check = OpenMetricsBaseCheck('prometheus_check', {}, {}, [instance], default_instance, default_namespace="foo")
assert check.get_scraper_config(instance)['label_to_hostname'] == 'node'
def test_credentials_empty():
creds = KubeletCredentials({})
assert creds.verify() is None
assert creds.cert_pair() is None
assert creds.headers("https://dummy") is None
instance = {'prometheus_url': 'https://dummy', 'namespace': 'foo'}
scraper = OpenMetricsBaseCheck('prometheus', {}, [instance])
scraper_config = scraper.create_scraper_configuration(instance)
creds.configure_scraper(scraper_config)
assert scraper_config['ssl_ca_cert'] is None
assert scraper_config['ssl_cert'] is None
assert scraper_config['ssl_private_key'] is None
assert scraper_config['extra_headers'] == {}
def test_credentials_empty():
creds = KubeletCredentials({})
assert creds.verify() is None
assert creds.cert_pair() is None
assert creds.headers("https://dummy") is None
scraper = OpenMetricsBaseCheck('prometheus', {}, {})
scraper_config = scraper.create_scraper_configuration()
scraper_config['prometheus_url'] = "https://dummy"
creds.configure_scraper(scraper_config)
assert scraper_config['ssl_ca_cert'] is None
assert scraper_config['ssl_cert'] is None
assert scraper_config['ssl_private_key'] is None
assert scraper_config['extra_headers'] == {}
def test_credentials_certificates():
creds = KubeletCredentials(
{"verify_tls": "true", "ca_cert": "ca_cert", "client_crt": "crt", "client_key": "key", "token": "ignore_me"}
)
assert creds.verify() == "ca_cert"
assert creds.cert_pair() == ("crt", "key")
assert creds.headers("https://dummy") is None
instance = {'prometheus_url': 'https://dummy', 'namespace': 'foo'}
scraper = OpenMetricsBaseCheck('prometheus', {}, [instance])
scraper_config = scraper.create_scraper_configuration(instance)
creds.configure_scraper(scraper_config)
assert scraper_config['ssl_ca_cert'] == "ca_cert"
assert scraper_config['ssl_cert'] == "crt"
assert scraper_config['ssl_private_key'] == "key"
assert scraper_config['extra_headers'] == {}
def test_args_legacy(self):
instance = {'prometheus_url': 'endpoint'}
check = OpenMetricsBaseCheck('openmetrics_check', {}, {}, [instance],
None, 'openmetrics')
assert check.default_instances == {}
assert check.default_namespace == 'openmetrics'
assert 'endpoint' in check.config_map
def test_default_legacy_basic(self):
check = OpenMetricsBaseCheck('openmetrics_check', self.INIT_CONFIG,
self.AGENT_CONFIG)
assert check.default_instances == {}
assert check.default_namespace is None
assert check.agentConfig == {'my_agent_config': 'foo bar agent config'}
assert check.name == 'openmetrics_check'
assert check.init_config == {'my_init_config': 'foo bar init config'}
def test_bearer_token_not_found():
endpoint = "none"
inexistent_file = os.path.join(FIXTURE_PATH, 'inexistent_file')
instance = {
'prometheus_url': endpoint,
'namespace': 'default_namespace',
'bearer_token_auth': True,
'bearer_token_path': inexistent_file,
}
with pytest.raises(IOError):
OpenMetricsBaseCheck('prometheus_check', {}, {}, [instance])
def test_default_instances_legacy_kwarg(self):
instance1 = {'prometheus_url': 'endpoint1'}
instance2 = {'prometheus_url': 'endpoint2'}
check = OpenMetricsBaseCheck('openmetrics_check', {}, {},
instances=[instance1, instance2],
default_namespace='openmetrics')
assert check.default_instances == {}
assert check.default_namespace == 'openmetrics'
assert 'endpoint1' in check.config_map
assert 'endpoint2' in check.config_map
def test_timeout_override():
endpoint = "none"
default_instance = {
'default_namespace': {
'prometheus_url': endpoint,
'metrics': [{
"test_rate": "test.rate"
}],
'prometheus_timeout': 30,
}
}
instance = {'prometheus_url': endpoint, 'namespace': 'default_namespace'}
check = OpenMetricsBaseCheck('prometheus_check', {}, {}, [instance],
default_instance,
default_namespace="foo")
assert check.get_scraper_config(instance)['prometheus_timeout'] == 30
instance = {
'prometheus_url': endpoint,
'namespace': 'default_namespace',
'prometheus_timeout': 5
}
check = OpenMetricsBaseCheck('prometheus_check', {}, {}, [instance],
default_instance,
default_namespace="foo")
assert check.get_scraper_config(instance)['prometheus_timeout'] == 5
def test_rate_override():
endpoint = "none"
instance = {
'prometheus_url': endpoint,
'metrics': [{"test_rate": "test.rate"}],
'type_overrides': {"test_rate": "rate"}
}
expected_type_overrides = {"test_rate": "rate"}
check = OpenMetricsBaseCheck('prometheus_check', {}, {}, [instance], default_namespace="foo")
processed_type_overrides = check.config_map[endpoint]['type_overrides']
assert expected_type_overrides == processed_type_overrides
def test_default_instances_with_tuple_instances(self):
instance = {'prometheus_url': 'endpoint'}
check = OpenMetricsBaseCheck('openmetrics_check',
self.INIT_CONFIG, (instance, ),
default_namespace='openmetrics')
assert check.default_instances == {}
assert check.default_namespace == 'openmetrics'
assert 'endpoint' in check.config_map
assert check.instance == instance
assert check.agentConfig == {}
assert check.name == 'openmetrics_check'
assert check.init_config == {'my_init_config': 'foo bar init config'}
def test_args_legacy(self):
instance = {'prometheus_url': 'endpoint'}
check = OpenMetricsBaseCheck(
'openmetrics_check',
self.INIT_CONFIG,
self.AGENT_CONFIG,
[instance],
{'my_inst': {
'foo': 'bar'
}},
'openmetrics',
)
assert check.default_instances == {'my_inst': {'foo': 'bar'}}
assert check.default_namespace == 'openmetrics'
assert 'endpoint' in check.config_map
assert check.instance == instance
assert check.agentConfig == {'my_agent_config': 'foo bar agent config'}
assert check.name == 'openmetrics_check'
assert check.init_config == {'my_init_config': 'foo bar init config'}
def test_default_instances_legacy_kwarg(self):
instance1 = {'prometheus_url': 'endpoint1'}
instance2 = {'prometheus_url': 'endpoint2'}
check = OpenMetricsBaseCheck(
'openmetrics_check',
self.INIT_CONFIG,
self.AGENT_CONFIG,
instances=[instance1, instance2],
default_instances={'my_inst': {
'foo': 'bar'
}},
default_namespace='openmetrics',
)
assert check.default_instances == {'my_inst': {'foo': 'bar'}}
assert check.default_namespace == 'openmetrics'
assert 'endpoint1' in check.config_map
assert 'endpoint2' in check.config_map
assert check.instance == instance1
assert check.agentConfig == {'my_agent_config': 'foo bar agent config'}
assert check.name == 'openmetrics_check'
assert check.init_config == {'my_init_config': 'foo bar init config'}
def test_default_legacy_basic(self):
check = OpenMetricsBaseCheck('openmetrics_check', {}, {})
assert check.default_instances == {}
assert check.default_namespace is None
