sentinel/tests/sentry/test_killswitches.py

from __future__ import annotations

import pytest

from sentry.killswitches import _value_matches, normalize_value


def test_normalize_value():
    assert normalize_value("store.load-shed-group-creation-projects", [1, 2, 3]) == [
        {"project_id": "1", "platform": None},
        {"project_id": "2", "platform": None},
        {"project_id": "3", "platform": None},
    ]

    assert normalize_value("store.load-shed-group-creation-projects", [{"project_id": 123}]) == [
        {"project_id": "123", "platform": None},  # match any platform
    ]


@pytest.mark.parametrize(
    ("cfg", "value"),
    (
        (
            [
                {"project_id": "1"},
                {"project_id": "2"},
                {"project_id": "3"},
            ],
            {"project_id": 2},
        ),
        (
            [
                {"project_id": 1},
                {"project_id": 2},
                {"project_id": 3},
            ],
            {"project_id": 2},
        ),
        (
            [{}],  # [{}] corresponds to any([all([])]), which is True
            {"project_id": 3},
        ),
        (
            [{"project_id": None}],
            {"project_id": 3},
        ),
        (
            [{"project_id": None, "platform": None}],
            {"project_id": 3},
        ),
        (
            [{"project_id": 3, "platform": None, "event_type": None}],
            {"project_id": 3},
        ),
        (
            [{"project_id": 3, "platform": None}],
            {"project_id": 3},
        ),
        (
            [{"event_type": "transaction"}],
            {"project_id": 3, "event_type": "transaction"},
        ),
    ),
)
def test_value_matches_positive(cfg, value):
    assert _value_matches("store.load-shed-group-creation-projects", cfg, value)


@pytest.mark.parametrize(
    ("cfg", "value"),
    (
        (
            [
                {"project_id": "1"},
                {"project_id": "2"},
                {"project_id": "3"},
            ],
            {"project_id": 4},
        ),
        (
            [],
            {"project_id": 4},
        ),
        (
            [{"project_id": 2, "platform": None}],
            {"project_id": 3},
        ),
        (
            [
                {"project_id": "1"},
                {"project_id": "2"},
                {"project_id": "3"},
            ],
            {},
        ),
    ),
)
def test_value_matches_negative(cfg, value):
    assert not _value_matches("store.load-shed-group-creation-projects", cfg, value)