import pytest
import os

TEST_CASES = [
    ("alertname=SomeAlert", "some-alert-receiver"),

    ("owner=platform severity=warning", "slack-platform-alerts"),
    ("owner=platform severity=critical", "pagerduty-platform-team"),

    ("severity=info", "slack-monitoring"),
    ("severity=warning", "slack-monitoring,pd-platform-warning"),
    ("severity=critical", "slack-monitoring,pd-platform-critical"),
]


@pytest.mark.parametrize("labels,receivers", TEST_CASES)
def test_routing(labels, receivers):
    cmd = f"amtool config routes test --config.file=./am-config.yaml --verify.receivers={receivers} {labels}"
    exit_code = os.system(cmd)
    assert exit_code == 0