qontract-reconcile 0.10.1rc1202__py3-none-any.whl → 0.10.2.dev2__py3-none-any.whl

tools/test/test_saas_promotion_state.py

@@ -1,187 +0,0 @@
-from collections.abc import (
-    Callable,
-    Iterable,
-    Mapping,
-)
-from unittest.mock import (
-    create_autospec,
-)
-
-from pytest import raises
-
-from reconcile.typed_queries.saas_files import SaasFile
-from reconcile.utils.promotion_state import PromotionData, PromotionState
-from tools.saas_promotion_state.saas_promotion_state import (
-    SaasPromotionState,
-    SaasPromotionStateException,
-    SaasPromotionStateMissingException,
-)
-
-
-def test_get_saas_promotion_state(
-    saas_files_builder: Callable[[Iterable[Mapping]], list[SaasFile]],
-) -> None:
-    saas_files = saas_files_builder([
-        {
-            "path": "/saas1.yml",
-            "name": "saas_1",
-            "resourceTemplates": [
-                {
-                    "name": "template_1",
-                    "url": "repo1/url",
-                    "targets": [
-                        {
-                            "ref": "main",
-                            "namespace": {"path": "/namespace1.yml"},
-                            "promotion": {
-                                "publish": ["channel-a"],
-                            },
-                        }
-                    ],
-                }
-            ],
-        },
-        {
-            "path": "/saas2.yml",
-            "name": "saas_2",
-            "resourceTemplates": [
-                {
-                    "name": "template_2",
-                    "url": "repo2/url",
-                    "targets": [
-                        {
-                            "ref": "main",
-                            "namespace": {"path": "/namespace2.yml"},
-                            "promotion": {
-                                "publish": ["channel-b"],
-                                "subscribe": ["channel-a"],
-                            },
-                        },
-                        {
-                            "ref": "main",
-                            "namespace": {"path": "/namespace3.yml"},
-                        },
-                    ],
-                }
-            ],
-        },
-    ])
-
-    expected = PromotionData(
-        check_in="test1",
-        saas_file="test2",
-        success=True,
-        target_config_hash="test3",
-    )
-    promotion_state = create_autospec(spec=PromotionState)
-    promotion_state.get_promotion_data.return_value = expected
-    saas_promotion_state = SaasPromotionState.create(
-        promotion_state=promotion_state, saas_files=saas_files
-    )
-    result = saas_promotion_state.get(channel="channel-a", sha="main")
-
-    assert result == {"616af45d7fad7f4eea8d52b8b5e8a058cef82ab0": expected}
-    promotion_state.get_promotion_data.assert_called_once_with(
-        sha="main",
-        channel="channel-a",
-        use_cache=False,
-        target_uid="616af45d7fad7f4eea8d52b8b5e8a058cef82ab0",
-        pre_check_sha_exists=False,
-    )
-
-
-def test_set_saas_promotion_state_success(
-    saas_files_builder: Callable[[Iterable[Mapping]], list[SaasFile]],
-) -> None:
-    saas_files = saas_files_builder([{"resourceTemplates": []}])
-
-    current_data = PromotionData(
-        check_in="test1",
-        saas_file="test2",
-        success=False,
-        target_config_hash="test3",
-    )
-    promotion_state = create_autospec(spec=PromotionState)
-    promotion_state.get_promotion_data.return_value = current_data
-    saas_promotion_state = SaasPromotionState.create(
-        promotion_state=promotion_state, saas_files=saas_files
-    )
-    saas_promotion_state.set_successful(
-        channel="test-channel", sha="test-sha", publisher_uid="test-uid"
-    )
-
-    promotion_state.get_promotion_data.assert_called_once_with(
-        sha="test-sha",
-        channel="test-channel",
-        use_cache=False,
-        target_uid="test-uid",
-        pre_check_sha_exists=False,
-    )
-    promotion_state.publish_promotion_data.assert_called_once_with(
-        data=PromotionData(
-            check_in="test1",
-            saas_file="test2",
-            success=True,
-            target_config_hash="test3",
-        ),
-        channel="test-channel",
-        sha="test-sha",
-        target_uid="test-uid",
-    )
-
-
-def test_set_saas_promotion_state_missing(
-    saas_files_builder: Callable[[Iterable[Mapping]], list[SaasFile]],
-) -> None:
-    saas_files = saas_files_builder([{"resourceTemplates": []}])
-    promotion_state = create_autospec(spec=PromotionState)
-    promotion_state.get_promotion_data.return_value = None
-    saas_promotion_state = SaasPromotionState.create(
-        promotion_state=promotion_state, saas_files=saas_files
-    )
-
-    with raises(SaasPromotionStateMissingException):
-        saas_promotion_state.set_successful(
-            channel="test-channel", sha="test-sha", publisher_uid="test-uid"
-        )
-
-    promotion_state.get_promotion_data.assert_called_once_with(
-        sha="test-sha",
-        channel="test-channel",
-        use_cache=False,
-        target_uid="test-uid",
-        pre_check_sha_exists=False,
-    )
-    promotion_state.publish_promotion_data.assert_not_called()
-
-
-def test_set_saas_promotion_state_already_successful(
-    saas_files_builder: Callable[[Iterable[Mapping]], list[SaasFile]],
-) -> None:
-    saas_files = saas_files_builder([{"resourceTemplates": []}])
-
-    current_data = PromotionData(
-        check_in="test1",
-        saas_file="test2",
-        success=True,
-        target_config_hash="test3",
-    )
-    promotion_state = create_autospec(spec=PromotionState)
-    promotion_state.get_promotion_data.return_value = current_data
-    saas_promotion_state = SaasPromotionState.create(
-        promotion_state=promotion_state, saas_files=saas_files
-    )
-
-    with raises(SaasPromotionStateException):
-        saas_promotion_state.set_successful(
-            channel="test-channel", sha="test-sha", publisher_uid="test-uid"
-        )
-
-    promotion_state.get_promotion_data.assert_called_once_with(
-        sha="test-sha",
-        channel="test-channel",
-        use_cache=False,
-        target_uid="test-uid",
-        pre_check_sha_exists=False,
-    )
-    promotion_state.publish_promotion_data.assert_not_called()

tools/test/test_sd_app_sre_alert_report.py

@@ -1,74 +0,0 @@
-from statistics import median
-
-from reconcile.test.fixtures import Fixtures
-from tools.sd_app_sre_alert_report import (
-    gen_alert_stats,
-    group_alerts,
-)
-
-messages = Fixtures("slack_api").get_anymarkup("conversations_history_messages.yaml")
-
-
-def test_group_alerts():
-    alerts = group_alerts(messages)
-
-    ptf = alerts["PrometheusTargetFlapping"]
-    assert ptf
-    assert len(ptf) == 6
-
-    sma = alerts["SLOMetricAbsent"]
-    assert sma
-    assert len(sma) == 4
-
-    paed = alerts["PatchmanAlertEvalDelay"]
-    assert paed
-    assert len(paed) == 2
-
-    csopc = alerts["ContainerSecurityOperatorPodCount"]
-    assert csopc
-    assert len(csopc) == 1
-
-    art = alerts["AlertmanagerReceiverTest"]
-    assert art
-    assert len(art) == 2
-
-    # This means one of the list elements has been ignored as it didn't have alertname
-    # as the total alerts we have from the above tests is 13 and the total of messages
-    # from the fixture is 14.
-    assert set(alerts.keys()) == {
-        "PrometheusTargetFlapping",
-        "SLOMetricAbsent",
-        "PatchmanAlertEvalDelay",
-        "ContainerSecurityOperatorPodCount",
-        "AlertmanagerReceiverTest",
-    }
-    assert len(messages) == 14
-
-
-def test_alert_stats():
-    alert_stats = gen_alert_stats(group_alerts(messages))
-
-    ptf = alert_stats["PrometheusTargetFlapping"]
-    assert ptf.triggered_alerts == 3
-    assert ptf.resolved_alerts == 3
-    assert median(ptf.elapsed_times) == 600
-
-    sma = alert_stats["SLOMetricAbsent"]
-    assert sma.triggered_alerts == 2
-    assert sma.resolved_alerts == 2
-    assert median(sma.elapsed_times) == 3600
-
-    paed = alert_stats["PatchmanAlertEvalDelay"]
-    assert paed.triggered_alerts == 1
-    assert paed.resolved_alerts == 1
-    assert not paed.elapsed_times
-
-    csopc = alert_stats["ContainerSecurityOperatorPodCount"]
-    assert csopc.triggered_alerts == 0
-    assert csopc.resolved_alerts == 1
-    assert not csopc.elapsed_times
-
-    art = alert_stats["AlertmanagerReceiverTest"]
-    assert art.triggered_alerts == 1
-    assert art.resolved_alerts == 1
-    assert median(art.elapsed_times) == 300

tools/test/test_sre_checkpoints.py

@@ -1,79 +0,0 @@
-from unittest.mock import patch
-
-from tools.qontract_cli import sre_checkpoints
-
-# from reconcile import queries
-from tools.sre_checkpoints import (
-    full_name,
-    get_latest_sre_checkpoints,
-)
-
-
-class TestFullName:
-    @staticmethod
-    def test_with_parent():
-        app = {
-            "parentApp": {"name": "app1"},
-            "name": "app2",
-        }
-        assert full_name(app) == "app1/app2"
-
-    @staticmethod
-    def test_without_parent():
-        app = {
-            "name": "app2",
-        }
-        assert full_name(app) == "app2"
-
-
-class TestLatestSRECheckpoints:
-    @staticmethod
-    @patch("reconcile.queries.get_sre_checkpoints")
-    def test_latest(get_sre_checkpoints):
-        get_sre_checkpoints.return_value = [
-            {"date": "2020-01-01", "app": {"name": "app_single"}},
-            {"date": "2021-01-01", "app": {"name": "app_single"}},
-            {"date": "2020-03-01", "app": {"name": "app_single"}},
-            {
-                "date": "2020-01-02",
-                "app": {"parentApp": {"name": "app1"}, "name": "app2"},
-            },
-        ]
-
-        latest_sre_checkpoints = get_latest_sre_checkpoints()
-        assert latest_sre_checkpoints["app_single"] == "2021-01-01"
-        assert latest_sre_checkpoints["app1/app2"] == "2020-01-02"
-
-
-class TestGetSRECheckpoints:
-    @staticmethod
-    @patch("tools.qontract_cli.print_output")
-    @patch("reconcile.queries.get_sre_checkpoints")
-    @patch("reconcile.queries.get_apps")
-    def test_sre_checkpoints(get_apps, get_sre_checkpoints, print_output):
-        get_apps.return_value = [
-            {"name": "app1", "path": "/app1", "onboardingStatus": "OnBoarded"},
-            {"name": "app2", "path": "/app2", "onboardingStatus": "OnBoarded"},
-            {"name": "app3", "path": "/app3", "onboardingStatus": "OnBoarded"},
-            {"name": "app4", "path": "/app4", "onboardingStatus": "InProgress"},
-        ]
-
-        get_sre_checkpoints.return_value = [
-            {"app": {"name": "app1"}, "date": "2021-01-02"},
-            {"app": {"name": "app1"}, "date": "2021-01-01"},
-        ]
-
-        expected_data = [
-            {"name": "app1", "latest": "2021-01-02"},
-            {"name": "app2", "latest": ""},
-            {"name": "app3", "latest": ""},
-        ]
-
-        with sre_checkpoints.make_context(info_name="info", args=[]) as ctx:
-            ctx.obj = {"options": {"output": "json"}}
-            sre_checkpoints.invoke(ctx)
-
-            cols = ["name", "latest"]
-            print_output.assert_called_once_with(
-                {"output": "json"}, expected_data, cols
-            )