qontract-reconcile 0.9.1rc298__py3-none-any.whl → 0.10.1.dev1203__py3-none-any.whl

reconcile/test/test_github_org.py

@@ -1,154 +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:
-            with 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,123 +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=set(
-            [
-                "https://github.com/org1/project1",
-                "https://github.com/org2/project1",
-                "https://github.com/org2/project2",
-            ]
-        ),
-        known_orgs=set(
-            [
-                "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=set([f"{expected_org}/project1"]),
-        known_orgs=set([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 == set([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=set([f"{expected_org}/project1"]),
-        known_orgs=set([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 == set([expected_org])

reconcile/test/test_gitlab_housekeeping.py

@@ -1,88 +0,0 @@
-from datetime import (
-    datetime,
-    timedelta,
-)
-from unittest.mock import patch
-
-from gitlab import Gitlab
-
-import reconcile.gitlab_housekeeping as gl_h
-from reconcile.test.fixtures import Fixtures
-from reconcile.utils.secret_reader import SecretReader
-
-DATE_FORMAT = "%Y-%m-%dT%H:%M:%S.%fZ"
-
-fixture = Fixtures("gitlab_housekeeping").get_anymarkup("api.yml")
-
-
-def get_mock(path, query_data=None, streamed=False, raw=False, **kwargs):
-    path = path[1:]
-    data = fixture.get("gitlab").get(path)
-    return data
-
-
-class TestGitLabHousekeeping:
-    @staticmethod
-    @patch.object(SecretReader, "read")
-    @patch.object(Gitlab, "http_get")
-    @patch.object(Gitlab, "http_post")
-    def test_clean_pipelines_happy_path(http_post, http_get, secret_read):
-        http_get.side_effect = get_mock
-        now = datetime.utcnow()
-
-        ten_minutes_ago = now - timedelta(minutes=10)
-        two_hours_ago = now - timedelta(minutes=120)
-
-        pipelines = [
-            {
-                "id": 46,
-                "iid": 11,
-                "project_id": 1,
-                "status": "canceled",
-                "ref": "new-pipeline",
-                "sha": "dddd9c1e5c9015edee04321e423429d2f8924609",
-                "web_url": "https://example.com/foo/bar/pipelines/46",
-                "created_at": two_hours_ago.strftime(DATE_FORMAT),
-                "updated_at": two_hours_ago.strftime(DATE_FORMAT),
-            },
-            {
-                "id": 47,
-                "iid": 12,
-                "project_id": 1,
-                "status": "pending",
-                "ref": "new-pipeline",
-                "sha": "a91957a858320c0e17f3a0eca7cfacbff50ea29a",
-                "web_url": "https://example.com/foo/bar/pipelines/47",
-                "created_at": two_hours_ago.strftime(DATE_FORMAT),
-                "updated_at": two_hours_ago.strftime(DATE_FORMAT),
-            },
-            {
-                "id": 48,
-                "iid": 13,
-                "project_id": 1,
-                "status": "running",
-                "ref": "new-pipeline",
-                "sha": "eb94b618fb5865b26e80fdd8ae531b7a63ad851a",
-                "web_url": "https://example.com/foo/bar/pipelines/48",
-                "created_at": ten_minutes_ago.strftime(DATE_FORMAT),
-                "updated_at": ten_minutes_ago.strftime(DATE_FORMAT),
-            },
-        ]
-        instance = {"url": "http://localhost", "sslVerify": False, "token": "token"}
-
-        dry_run = False
-        timeout = 60
-
-        timeout_pipelines = gl_h.get_timed_out_pipelines(pipelines, timeout)
-        gl_h.clean_pipelines(dry_run, instance, "1", "", timeout_pipelines)
-
-        # Test if mock have this exact calls
-        http_post.assert_called_once_with("/projects/1/pipelines/47/cancel")
-
-
-def test_calculate_time_since_approval():
-    one_hour_ago = (datetime.utcnow() - timedelta(minutes=60)).strftime(DATE_FORMAT)
-
-    time_since_merge = gl_h._calculate_time_since_approval(one_hour_ago)
-
-    assert round(time_since_merge) == 60

reconcile/test/test_gitlab_labeler.py

@@ -1,129 +0,0 @@
-from typing import (
-    Any,
-    Optional,
-)
-from unittest.mock import (
-    create_autospec,
-    patch,
-)
-
-import reconcile.gitlab_labeler as gl
-from reconcile.queries import APPS_QUERY
-from reconcile.utils import gql
-
-from .fixtures import Fixtures
-
-
-class StubData:
-    """Class to add data to tests in setUp. It will be used by mocks"""
-
-    def __init__(self):
-        self._apps = []
-
-    @property
-    def apps(self) -> list[dict]:
-        return self._apps
-
-    @apps.setter
-    def apps(self, apps: list[dict]) -> None:
-        if not isinstance(apps, list):
-            raise TypeError(f"Expecting list, have {type(apps)}")
-        self._apps = apps
-
-
-# This was originally written in unittest, hence the use of xunit-style
-# setup/teardown methods instead of pytest fixtures.
-class TestOnboardingGuesser:
-    def setup_method(self) -> None:
-        self.test_data = StubData()
-
-        self.fxt = Fixtures("apps")
-        self.app1 = self.fxt.get_json("app1.json")
-        self.app2 = self.fxt.get_json("app2.json")
-        self.app3 = self.fxt.get_json("childapp.json")
-        self.app4 = self.fxt.get_json("parentapp.json")
-
-        # Patcher for GqlApi methods
-        self.gql_patcher = patch.object(gql, "get_api", autospec=True)
-        self.gql = self.gql_patcher.start()
-        gqlapi_mock = create_autospec(gql.GqlApi)
-        self.gql.return_value = gqlapi_mock
-        gqlapi_mock.query.side_effect = self.mock_gql_query
-
-    def mock_gql_query(self, query: str) -> Optional[dict[str, Any]]:
-        """Mock for GqlApi.query using test_data set in setUp"""
-        if query == APPS_QUERY:
-            return {"apps": self.test_data.apps}
-        return None
-
-    def teardown_method(self) -> None:
-        """cleanup patches created in self.setUp"""
-        self.gql_patcher.stop()
-
-    def test_get_app_list(self):
-        self.test_data.apps = [self.app1]
-        t = gl.get_app_list()
-        assert t == {"normalapp": {"onboardingStatus": "BestEffort", "parentApp": None}}
-
-    def test_get_app_list_2(self):
-        self.test_data.apps = [self.app1, self.app2]
-        t = gl.get_app_list()
-        assert t, {
-            "normalapp": {"onboardingStatus": "BestEffort", "parentApp": None},
-            "normalapp2": {"onboardingStatus": "BestEffort", "parentApp": None},
-        }
-
-    def test_get_parents_list(self):
-        self.test_data.apps = [self.app1, self.app2, self.app3]
-        t = gl.get_parents_list()
-        assert t == {"parentapp"}
-
-    def test_get_parents_list_empty(self):
-        self.test_data.apps = [self.app1, self.app2]
-        t = gl.get_parents_list()
-        assert t == set()
-
-    def test_guess_onboarding_status_child(self):
-        self.test_data.apps = [self.app1, self.app2, self.app3]
-        parents = gl.get_parents_list()
-        apps = gl.get_app_list()
-        changed_paths = ["data/services/parentapp/childapp"]
-
-        t = gl.guess_onboarding_status(changed_paths, apps, parents)
-        assert t == "BestEffort"
-
-    def test_guess_onboarding_status_parent(self):
-        self.test_data.apps = [self.app1, self.app2, self.app3, self.app4]
-        parents = gl.get_parents_list()
-        apps = gl.get_app_list()
-        changed_paths = ["data/services/parentapp/test"]
-
-        t = gl.guess_onboarding_status(changed_paths, apps, parents)
-        assert t == "OnBoarded"
-
-    def test_guess_onboarding_status_normal(self):
-        self.test_data.apps = [self.app1, self.app2, self.app3]
-        parents = gl.get_parents_list()
-        apps = gl.get_app_list()
-        changed_paths = ["data/services/normalapp2"]
-
-        t = gl.guess_onboarding_status(changed_paths, apps, parents)
-        assert t == "BestEffort"
-
-    def test_guess_onboarding_status_no_app(self):
-        self.test_data.apps = [self.app1, self.app2, self.app3]
-        parents = gl.get_parents_list()
-        apps = gl.get_app_list()
-        changed_paths = ["data/test/test"]
-
-        t = gl.guess_onboarding_status(changed_paths, apps, parents)
-        assert t is None
-
-    def test_guess_onboarding_status_key_error(self):
-        self.test_data.apps = [self.app1, self.app2, self.app3]
-        parents = gl.get_parents_list()
-        apps = gl.get_app_list()
-        changed_paths = ["data/services/normalapp3"]
-
-        t = gl.guess_onboarding_status(changed_paths, apps, parents)
-        assert t is None