def test_get_labels_of_service(monkeypatch): automation = automations.AutomationGetLabelsOf() ts = Scenario().add_host("test-host") ts.set_ruleset( "service_label_rules", [ ({ "label1": "val1" }, [], config.ALL_HOSTS, ["CPU load$"], {}), ({ "label2": "val2" }, [], config.ALL_HOSTS, ["CPU load$"], {}), ], ) ts.apply(monkeypatch) assert automation.execute(["service", "test-host", "CPU load"]) == GetLabelsOfResult({ "labels": { "label1": "val1", "label2": "val2" }, "label_sources": { "label1": "ruleset", "label2": "ruleset" }, })
def test_get_labels_of_host(monkeypatch): automation = automations.AutomationGetLabelsOf() ts = Scenario().add_host("test-host") ts.set_option("host_labels", { "test-host": { "explicit": "ding", }, }) ts.apply(monkeypatch) assert automation.execute(["host", "test-host"]) == { "labels": { "explicit": "ding" }, "label_sources": { "explicit": "explicit" }, }