qontract-reconcile 0.10.1rc1201__py3-none-any.whl → 0.10.2.dev1__py3-none-any.whl

reconcile/test/test_deadmanssnitch.py

@@ -1,290 +0,0 @@
-from unittest.mock import MagicMock, create_autospec
-
-import pytest
-from pytest_mock import MockerFixture
-
-from reconcile.deadmanssnitch import (
-    DeadMansSnitchIntegration,
-)
-from reconcile.gql_definitions.common.app_interface_dms_settings import (
-    DeadMansSnitchSettingsV1,
-    VaultSecretV1,
-)
-from reconcile.gql_definitions.common.clusters_with_dms import (
-    ClusterV1,
-)
-from reconcile.utils.deadmanssnitch_api import (
-    DeadMansSnitchApi,
-    Snitch,
-)
-
-
-@pytest.fixture
-def deadmanssnitch_api() -> MagicMock:
-    return create_autospec(DeadMansSnitchApi)
-
-
-@pytest.fixture
-def deadmanssnitch_settings() -> DeadMansSnitchSettingsV1:
-    settings = DeadMansSnitchSettingsV1(
-        alertMailAddresses=["test_email"],
-        notesLink="test_link",
-        snitchesPath="test_snitches_path",
-        tokenCreds=VaultSecretV1(path="test_path", field="test_field"),
-        tags=["test-tags"],
-        alertType="Heartbeat",
-        interval="15_minute",
-    )
-    return settings
-
-
-@pytest.fixture
-def vault_mock(mocker: MockerFixture) -> MockerFixture:
-    return mocker.patch("reconcile.utils.vault._VaultClient")
-
-
-@pytest.fixture
-def secret_reader(mocker: MockerFixture) -> MockerFixture:
-    mock_secretreader = mocker.patch(
-        "reconcile.utils.secret_reader.SecretReader", autospec=True
-    )
-    mock_secretreader.read.return_value = "secret"
-    mock_secretreader.read_secret.return_value = "secret"
-    return mock_secretreader
-
-
-def test_get_current_state(
-    secret_reader: MagicMock,
-    deadmanssnitch_api: MagicMock,
-    mocker: MockerFixture,
-    deadmanssnitch_settings: DeadMansSnitchSettingsV1,
-) -> None:
-    deadmanssnitch_api.get_snitches.return_value = [
-        Snitch(
-            name="prometheus.test_cluster_1.net",
-            token="test",
-            href="testc",
-            status="healthy",
-            alert_type="basic",
-            alert_email=["test_mail"],
-            interval="15_minute",
-            check_in_url="test_url",
-            tags=["app-sre"],
-            notes="test_notes",
-        )
-    ]
-    clusters = [
-        ClusterV1(
-            name="test_cluster_1",
-            serverUrl="testurl",
-            consoleUrl="test_c_url",
-            alertmanagerUrl="test_alert_manager",
-            managedClusterRoles=True,
-            prometheusUrl="https://prometheus.test_cluster_1.net",
-            enableDeadMansSnitch=True,
-        ),
-        ClusterV1(
-            name="test_cluster_2",
-            serverUrl="testurl",
-            consoleUrl="test_c_url",
-            alertmanagerUrl="test_alert_manager",
-            managedClusterRoles=True,
-            prometheusUrl="https://prometheus.test_cluster_2.net",
-            enableDeadMansSnitch=True,
-        ),
-    ]
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchIntegration.__init__"
-    ).return_value = None
-    dms_integration = DeadMansSnitchIntegration()
-    dms_integration._secret_reader = secret_reader
-    dms_integration.settings = deadmanssnitch_settings
-    current_state = dms_integration.get_current_state(
-        deadmanssnitch_api=deadmanssnitch_api,
-        clusters=clusters,
-        vault_snitch_map={"deadmanssnitch-test_cluster_1-url": "secret"},
-    )
-    assert current_state["test_cluster_1"].vault_data == "secret"
-
-
-def test_integration_for_create(
-    vault_mock: MagicMock,
-    deadmanssnitch_settings: DeadMansSnitchSettingsV1,
-    mocker: MockerFixture,
-    secret_reader: MagicMock,
-) -> None:
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchIntegration.__init__"
-    ).return_value = None
-    dms_integration = DeadMansSnitchIntegration()
-    secret_reader.read_all.return_value = {}
-    dms_integration._secret_reader = secret_reader
-    dms_integration.settings = deadmanssnitch_settings
-    dms_integration.vault_client = vault_mock
-    mocker.patch("reconcile.deadmanssnitch.get_clusters_with_dms").return_value = [
-        ClusterV1(
-            name="create_cluster",
-            prometheusUrl="https://prometheus.create_cluster.devshift.net",
-            enableDeadMansSnitch=True,
-            alertmanagerUrl="alertmanager.create_cluster.devshift.net",
-            managedClusterRoles=True,
-            serverUrl="testurl",
-            consoleUrl="test_console",
-        )
-    ]
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchApi.get_snitches"
-    ).return_value = []
-    mock_create_snitch = mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchIntegration.create_snitch"
-    )
-    mock_create_snitch.return_value = Snitch(
-        name="prometheus.create_cluster.devshift.net",
-        token="test",
-        href="testc",
-        status="healthy",
-        alert_type="basic",
-        alert_email=["test_mail"],
-        interval="15_minute",
-        check_in_url="test_url",
-        tags=["app-sre"],
-        notes="test_notes",
-    )
-    dms_integration.run(dry_run=False)
-    mock_create_snitch.assert_called_once()
-
-
-def test_integration_for_delete(
-    deadmanssnitch_settings: DeadMansSnitchSettingsV1,
-    vault_mock: MagicMock,
-    mocker: MockerFixture,
-    secret_reader: MagicMock,
-) -> None:
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchIntegration.__init__"
-    ).return_value = None
-    dms_integration = DeadMansSnitchIntegration()
-    secret_reader.read_all.return_value = {"deadmanssnitch-create_cluster-url": "test"}
-    dms_integration._secret_reader = secret_reader
-    dms_integration.settings = deadmanssnitch_settings
-    dms_integration.vault_client = vault_mock
-    mocker.patch("reconcile.deadmanssnitch.get_clusters_with_dms").return_value = [
-        ClusterV1(
-            name="create_cluster",
-            prometheusUrl="https://prometheus.create_cluster.devshift.net",
-            enableDeadMansSnitch=False,
-            alertmanagerUrl="alertmanager.create_cluster.devshift.net",
-            managedClusterRoles=True,
-            serverUrl="testurl",
-            consoleUrl="test_console",
-        )
-    ]
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchApi.get_snitches"
-    ).return_value = [
-        Snitch(
-            name="prometheus.create_cluster.devshift.net",
-            token="test",
-            href="testc",
-            status="healthy",
-            alert_type="basic",
-            alert_email=["test_mail"],
-            interval="15_minute",
-            check_in_url="test_url",
-            tags=["app-sre"],
-            notes="test_notes",
-        )
-    ]
-    mock_delete_snitch = mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchApi.delete_snitch"
-    )
-    mock_delete_snitch.return_value = None
-    dms_integration.run(dry_run=False)
-    mock_delete_snitch.assert_called_once_with("test")
-
-
-def test_integration_for_update_vault(
-    deadmanssnitch_settings: DeadMansSnitchSettingsV1,
-    vault_mock: MagicMock,
-    mocker: MockerFixture,
-    secret_reader: MagicMock,
-) -> None:
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchIntegration.__init__"
-    ).return_value = None
-    dms_integration = DeadMansSnitchIntegration()
-    secret_reader.read_all.return_value = {"deadmanssnitch-test_cluster-url": "test"}
-    dms_integration._secret_reader = secret_reader
-    dms_integration.settings = deadmanssnitch_settings
-    dms_integration.vault_client = vault_mock
-    mocker.patch("reconcile.deadmanssnitch.get_clusters_with_dms").return_value = [
-        ClusterV1(
-            name="test_cluster",
-            prometheusUrl="https://prometheus.create_cluster.devshift.net",
-            enableDeadMansSnitch=True,
-            alertmanagerUrl="alertmanager.create_cluster.devshift.net",
-            managedClusterRoles=True,
-            serverUrl="testurl",
-            consoleUrl="test_console",
-        )
-    ]
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchApi.get_snitches"
-    ).return_value = [
-        Snitch(
-            name="prometheus.create_cluster.devshift.net",
-            token="test",
-            href="testc",
-            status="healthy",
-            alert_type="basic",
-            alert_email=["test_mail"],
-            interval="15_minute",
-            check_in_url="test_url",
-            tags=["app-sre"],
-            notes="test_notes",
-        )
-    ]
-    dms_integration.run(dry_run=False)
-    data = {"deadmanssnitch-test_cluster-url": "test_url"}
-    vault_mock.write.assert_called_once_with(
-        {
-            "path": deadmanssnitch_settings.snitches_path,
-            "data": data,
-        },
-        decode_base64=False,
-    )
-
-
-def test_integration_while_failed(
-    deadmanssnitch_settings: DeadMansSnitchSettingsV1,
-    vault_mock: MagicMock,
-    mocker: MockerFixture,
-    secret_reader: MagicMock,
-) -> None:
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchIntegration.__init__"
-    ).return_value = None
-    dms_integration = DeadMansSnitchIntegration()
-    dms_integration._secret_reader = secret_reader
-    dms_integration.settings = deadmanssnitch_settings
-    dms_integration.vault_client = vault_mock
-    mocker.patch("reconcile.deadmanssnitch.get_clusters_with_dms").return_value = [
-        ClusterV1(
-            name="create_cluster",
-            prometheusUrl="https://prometheus.create_cluster.devshift.net",
-            enableDeadMansSnitch=True,
-            alertmanagerUrl="alertmanager.create_cluster.devshift.net",
-            managedClusterRoles=True,
-            serverUrl="testurl",
-            consoleUrl="test_console",
-        )
-    ]
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchApi.get_snitches"
-    ).return_value = []
-    mocker.patch(
-        "reconcile.deadmanssnitch.DeadMansSnitchIntegration.create_snitch",
-        side_effect=Exception("mock vault"),
-    )
-    with pytest.raises(ExceptionGroup):
-        dms_integration.run(dry_run=False)

reconcile/test/test_gabi_authorized_users.py

@@ -1,72 +0,0 @@
-from datetime import (
-    date,
-    timedelta,
-)
-
-import pytest
-from pytest_mock import MockerFixture
-
-import reconcile.gabi_authorized_users as gabi_u
-import reconcile.openshift_base as ob
-from reconcile.test.fixtures import Fixtures
-from reconcile.utils.aggregated_list import RunnerException
-from reconcile.utils.openshift_resource import OpenshiftResource as OR
-from reconcile.utils.openshift_resource import ResourceInventory
-
-fixture = Fixtures("gabi_authorized_users").get_anymarkup("api.yml")
-
-
-def test_gabi_authorized_users_apply(mocker: MockerFixture) -> None:
-    ri = ResourceInventory()
-    ri.initialize_resource_type("server", "gabi-db", "ConfigMap")
-    gabi_instance = fixture["gql_response"][0]
-    gabi_u.fetch_desired_state([gabi_instance], ri)
-    mock_apply = mocker.patch.object(ob, "apply", autospec=True)
-    oc_map = mocker.patch("reconcile.utils.oc_map.OCMap", autospec=True)
-    ob.realize_data(True, oc_map, ri, 1)
-    expected = OR(
-        fixture["desired"],
-        gabi_u.QONTRACT_INTEGRATION,
-        gabi_u.QONTRACT_INTEGRATION_VERSION,
-    )
-    _, kwargs = mock_apply.call_args
-    assert kwargs["resource"] == expected
-
-
-def test_gabi_authorized_users_exist(mocker: MockerFixture) -> None:
-    ri = ResourceInventory()
-    ri.initialize_resource_type("server", "gabi-db", "ConfigMap")
-    current = OR(
-        fixture["current"],
-        gabi_u.QONTRACT_INTEGRATION,
-        gabi_u.QONTRACT_INTEGRATION_VERSION,
-    )
-    ri.add_current(
-        "server",
-        "gabi-db",
-        "ConfigMap",
-        current.name,
-        current,
-    )
-    gabi_instance = fixture["gql_response"][0]
-    gabi_u.fetch_desired_state([gabi_instance], ri)
-    mock_apply = mocker.patch.object(ob, "apply", autospec=True)
-    oc_map = mocker.patch("reconcile.utils.oc_map.OCMap", autospec=True)
-    ob.realize_data(True, oc_map, ri, 1)
-    mock_apply.assert_not_called()
-
-
-def test_gabi_authorized_users_exceed() -> None:
-    expiration_date = date.today() + timedelta(days=(gabi_u.EXPIRATION_DAYS_MAX + 1))
-    gabi_instance = fixture["gql_response"][0]
-    gabi_instance["expirationDate"] = str(expiration_date)
-    ri = ResourceInventory()
-    with pytest.raises(RunnerException):
-        gabi_u.fetch_desired_state([gabi_instance], ri)
-
-
-def test_gabi_authorized_users_rds_not_found() -> None:
-    gabi_instance = fixture["gql_response"][1]
-    ri = ResourceInventory()
-    with pytest.raises(RunnerException):
-        gabi_u.fetch_desired_state([gabi_instance], ri)

reconcile/test/test_gcr_mirror.py

@@ -1,14 +0,0 @@
-from pytest_mock import MockerFixture
-
-from reconcile.gcr_mirror import QuayMirror
-
-
-def test_gcr_mirror_session(mocker: MockerFixture) -> None:
-    mocker.patch("reconcile.gcr_mirror.gql")
-    mocker.patch("reconcile.gcr_mirror.queries")
-    mocked_request = mocker.patch("reconcile.gcr_mirror.requests")
-
-    with QuayMirror() as gcr_mirror:
-        assert gcr_mirror.session == mocked_request.Session.return_value
-
-    mocked_request.Session.return_value.close.assert_called_once_with()

reconcile/test/test_github_org.py

@@ -1,156 +0,0 @@
-from unittest.mock import patch
-
-from reconcile import github_org
-from reconcile.utils import (
-    config,
-    gql,
-)
-from reconcile.utils.aggregated_list import AggregatedList
-
-from .fixtures import Fixtures
-
-fxt = Fixtures("github_org")
-
-
-class RawGithubApiMock:
-    @staticmethod
-    def org_invitations(org_name):
-        return []
-
-    @staticmethod
-    def team_invitations(org_id, team_id):
-        return []
-
-
-class AttrDict(dict):
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        self.__dict__ = self
-
-
-class GithubMock:
-    def __init__(self, spec):
-        self.spec = spec
-
-    class GithubOrgMock:
-        def __init__(self, spec_org):
-            self.spec_org = spec_org
-
-        @property
-        def id(self):
-            return self.spec_org["id"]
-
-        class GithubTeamMock:
-            def __init__(self, spec_team):
-                self.spec_team = spec_team
-
-            def get_members(self):
-                return map(AttrDict, self.spec_team["members"])
-
-            @property
-            def id(self):
-                return self.spec_team["name"]
-
-            @property
-            def name(self):
-                return self.spec_team["name"]
-
-        def get_members(self):
-            return map(AttrDict, self.spec_org["members"])
-
-        def get_teams(self):
-            return map(self.GithubTeamMock, self.spec_org["teams"])
-
-    def get_organization(self, org_name):
-        return self.GithubOrgMock(self.spec[org_name])
-
-
-def get_items_by_params(state, params):
-    h = AggregatedList.hash_params(params)
-    for group in state:
-        this_h = AggregatedList.hash_params(group["params"])
-
-        if h == this_h:
-            return sorted(group["items"])
-    return False
-
-
-class TestGithubOrg:
-    @staticmethod
-    def setup_method(method):
-        config.init_from_toml(fxt.path("config.toml"))
-        gql.init_from_config(autodetect_sha=False)
-
-    @staticmethod
-    def do_current_state_test(path):
-        fixture = fxt.get_anymarkup(path)
-
-        with (
-            patch("reconcile.github_org.RawGithubApi") as m_rga,
-            patch("reconcile.github_org.Github") as m_gh,
-        ):
-            m_gh.return_value = GithubMock(fixture["gh_api"])
-            m_rga.return_value = RawGithubApiMock()
-
-            gh_api_store = github_org.GHApiStore(config.get_config())
-            current_state = github_org.fetch_current_state(gh_api_store)
-            current_state = current_state.dump()
-
-            expected_current_state = fixture["state"]
-
-            assert len(current_state) == len(expected_current_state)
-            for group in current_state:
-                params = group["params"]
-                items = sorted(group["items"])
-                assert items == get_items_by_params(expected_current_state, params)
-
-    @staticmethod
-    def do_desired_state_test(path):
-        fixture = fxt.get_anymarkup(path)
-
-        with patch("reconcile.utils.gql.GqlApi.query") as m_gql:
-            m_gql.return_value = fixture["gql_response"]
-
-            desired_state = github_org.fetch_desired_state(infer_clusters=False).dump()
-
-            expected_desired_state = fixture["state"]
-
-            assert len(desired_state) == len(expected_desired_state)
-            for group in desired_state:
-                params = group["params"]
-                items = sorted(group["items"])
-                assert items == get_items_by_params(expected_desired_state, params)
-
-    def test_current_state_simple(self):
-        self.do_current_state_test("current_state_simple.yml")
-
-    def test_desired_state_simple(self):
-        self.do_desired_state_test("desired_state_simple.yml")
-
-    def test_get_members(self):
-        class SimpleMemberMock:
-            def __init__(self, login):
-                self.login = login
-
-        class SimpleOrgMock:
-            @staticmethod
-            def get_members():
-                return [SimpleMemberMock("a"), SimpleMemberMock("b")]
-
-        org = SimpleOrgMock()
-        assert github_org.get_members(org) == ["a", "b"]
-
-    def test_get_org_teams(self):
-        class SimpleOrgMock:
-            @staticmethod
-            def get_teams():
-                return ["teams"]
-
-        class SimpleGithubMock:
-            @staticmethod
-            def get_organization(org_name):
-                return SimpleOrgMock()
-
-        g = SimpleGithubMock()
-        _, teams = github_org.get_org_and_teams(g, "org")
-        assert teams == ["teams"]

reconcile/test/test_github_repo_invites.py

@@ -1,119 +0,0 @@
-from collections.abc import (
-    Iterable,
-    Mapping,
-)
-from typing import Any
-from unittest.mock import MagicMock
-
-import pytest
-
-from reconcile import github_repo_invites
-from reconcile.utils.raw_github_api import RawGithubApi
-
-
-def test_parse_null_code_components():
-    raw_code_components = None
-    expected = github_repo_invites.CodeComponents(
-        urls=set(),
-        known_orgs=set(),
-    )
-    assert github_repo_invites._parse_code_components(raw_code_components) == expected
-
-
-def test_parse_valid_code_components():
-    raw_code_components: Iterable[Mapping[str, Any]] = [
-        {
-            "codeComponents": [
-                {
-                    "url": "https://github.com/org1/project1",
-                    "resource": "upstream",
-                },
-                {
-                    "url": "https://github.com/org2/project1",
-                    "resource": "upstream",
-                },
-            ],
-        },
-        {
-            "codeComponents": [],
-        },
-        {
-            "codeComponents": [
-                {
-                    "url": "https://github.com/org2/project2",
-                    "resource": "upstream",
-                }
-            ],
-        },
-    ]
-    expected = github_repo_invites.CodeComponents(
-        urls={
-            "https://github.com/org1/project1",
-            "https://github.com/org2/project1",
-            "https://github.com/org2/project2",
-        },
-        known_orgs={
-            "https://github.com/org1",
-            "https://github.com/org2",
-        },
-    )
-    assert github_repo_invites._parse_code_components(raw_code_components) == expected
-
-
-@pytest.fixture
-def github():
-    mock = MagicMock(spec=RawGithubApi)
-    mock.repo_invitations = MagicMock()
-    mock.accept_repo_invitation = MagicMock()
-    return mock
-
-
-def test_accept_invitations_no_dry_run(github):
-    expected_id = "123"
-    expected_org = "https://github.com/org1"
-    github.repo_invitations.side_effect = [
-        [
-            {
-                "id": expected_id,
-                "html_url": f"{expected_org}/project1",
-            },
-            {
-                "id": "456",
-                "html_url": "https://github.com/org3/project1",
-            },
-        ]
-    ]
-    code_components = github_repo_invites.CodeComponents(
-        urls={f"{expected_org}/project1"},
-        known_orgs={expected_org},
-    )
-    dry_run = False
-    accepted_invitations = github_repo_invites._accept_invitations(
-        github, code_components, dry_run
-    )
-
-    github.accept_repo_invitation.assert_called_once_with(expected_id)
-    assert accepted_invitations == {expected_org}
-
-
-def test_accept_invitations_dry_run(github):
-    expected_org = "https://github.com/org1"
-    github.repo_invitations.side_effect = [
-        [
-            {
-                "id": "123",
-                "html_url": f"{expected_org}/project1",
-            },
-        ],
-    ]
-    code_components = github_repo_invites.CodeComponents(
-        urls={f"{expected_org}/project1"},
-        known_orgs={expected_org},
-    )
-    dry_run = True
-    accepted_invitations = github_repo_invites._accept_invitations(
-        github, code_components, dry_run
-    )
-
-    github.accept_repo_invitation.assert_not_called()
-    assert accepted_invitations == {expected_org}