def test_probe_building(mocker):
ep_query = mocker.patch.object(queries, 'get_service_monitoring_endpoints')
ep_query.return_value = get_endpoint_fixtures("test_probe_building.yaml")
endpoints = get_endpoints()
assert len(endpoints) == 1
provider = list(endpoints.keys())[0]
probe_resource = build_probe(provider, endpoints.get(provider))
assert probe_resource is not None
# verify prober url decomposition
spec = probe_resource.body.get("spec")
assert spec.get("prober") == {
"url": "exporterhost:9115",
"scheme": "http",
"path": "/probe"
}
# verify labels
labels = spec["targets"]["staticConfig"]["labels"]
assert labels.get("environment") == "staging"
# verify timeout and interval
assert spec["scrapeTimeout"] == provider.timeout
assert spec["interval"] == provider.checkInterval
# verify targets
assert "https://test1.url" in spec["targets"]["staticConfig"]["static"]
assert "https://test2.url" in spec["targets"]["staticConfig"]["static"]
def test_endpoint_loading(mocker):
ep_query = mocker.patch.object(queries, 'get_service_monitoring_endpoints')
ep_query.return_value = get_endpoint_fixtures("test_endpoint.yaml")
endpoints = get_endpoints()
assert len(endpoints) == 1
provider = list(endpoints.keys())[0]
assert provider.provider == "blackbox-exporter"
ep = endpoints.get(provider)[0]
assert len(ep.monitoring) == 1
def test_loading_multiple_providers_per_endpoint(mocker):
ep_query = mocker.patch.object(queries, 'get_service_monitoring_endpoints')
ep_query.return_value = \
get_endpoint_fixtures("test_multiple_providers_per_endpoint.yaml")
endpoints = get_endpoints()
assert len(endpoints) == 2
for provider, eps in endpoints.items():
assert provider.provider == "blackbox-exporter"
assert len(eps) == 2
def test_filling_desired_state(mocker):
ep_query = mocker.patch.object(queries, 'get_service_monitoring_endpoints')
ep_query.return_value = get_endpoint_fixtures("test_endpoint.yaml")
add_desired_mock = mocker.patch.object(ResourceInventory, 'add_desired')
endpoints = get_endpoints()
provider = list(endpoints.keys())[0]
fill_desired_state(provider, endpoints[provider], ResourceInventory())
assert add_desired_mock.call_count == 1
add_desired_mock.assert_called_with(
cluster="app-sre-stage-01",
namespace="openshift-customer-monitoring",
resource_type="Probe",
name="blackbox-exporter-http-2xx",
value=ANY)
def test_invalid_endpoints(mocker):
query = mocker.patch.object(queries, 'get_service_monitoring_endpoints')
query.return_value = get_endpoint_fixtures("test_invalid_endpoints.yaml")
endpoints = get_endpoints()
assert len(endpoints) == 0