sentinel/tests/sentry/integrations/github_enterprise/test_client.py

import datetime
from unittest import mock

import pytest
import responses

from sentry.integrations.github_enterprise.integration import GitHubEnterpriseIntegration
from sentry.models.repository import Repository
from sentry.shared_integrations.exceptions import ApiError
from sentry.testutils.cases import TestCase
from sentry.testutils.silo import region_silo_test

GITHUB_CODEOWNERS = {
    "filepath": "CODEOWNERS",
    "html_url": "https://github.example.org/Test-Organization/foo/blob/master/CODEOWNERS",
    "raw": "docs/*    @jianyuan   @getsentry/ecosystem\n* @jianyuan\n",
}


@region_silo_test
class GitHubAppsClientTest(TestCase):
    base_url = "https://github.example.org/api/v3"

    def setUp(self):
        super().setUp()

        patcher_1 = mock.patch(
            "sentry.integrations.github_enterprise.client.get_jwt", return_value="jwt_token_1"
        )
        patcher_1.start()
        self.addCleanup(patcher_1.stop)

        patcher_2 = mock.patch(
            "sentry.integrations.github_enterprise.integration.get_jwt", return_value="jwt_token_1"
        )
        patcher_2.start()
        self.addCleanup(patcher_2.stop)

        ten_days = datetime.datetime.utcnow() + datetime.timedelta(days=10)
        integration = self.create_integration(
            organization=self.organization,
            provider="github_enterprise",
            name="Github Test Org",
            external_id="1",
            metadata={
                "access_token": "12345token",
                "expires_at": ten_days.strftime("%Y-%m-%dT%H:%M:%S"),
                "icon": "https://github.example.org/avatar.png",
                "domain_name": "github.example.org/Test-Organization",
                "account_type": "Organization",
                "installation_id": "install_id_1",
                "installation": {
                    "client_id": "client_id",
                    "client_secret": "client_secret",
                    "id": "2",
                    "name": "test-app",
                    "private_key": "private_key",
                    "url": "github.example.org",
                    "webhook_secret": "webhook_secret",
                    "verify_ssl": True,
                },
            },
        )
        self.repo = Repository.objects.create(
            organization_id=self.organization.id,
            name="Test-Organization/foo",
            url="https://github.example.org/Test-Organization/foo",
            provider="integrations:github_enterprise",
            external_id=123,
            integration_id=integration.id,
        )
        install = integration.get_installation(organization_id=self.organization.id)
        assert isinstance(install, GitHubEnterpriseIntegration)
        self.install = install
        self.gh_client = self.install.get_client()

    @responses.activate
    def test_check_file(self):
        path = "src/sentry/integrations/github/client.py"
        version = "master"
        url = f"{self.base_url}/repos/{self.repo.name}/contents/{path}?ref={version}"

        responses.add(
            method=responses.HEAD,
            url=url,
            json={"text": 200},
        )

        resp = self.gh_client.check_file(self.repo, path, version)
        assert resp.status_code == 200

    @responses.activate
    def test_check_no_file(self):
        path = "src/santry/integrations/github/client.py"
        version = "master"
        url = f"{self.base_url}/repos/{self.repo.name}/contents/{path}?ref={version}"

        responses.add(method=responses.HEAD, url=url, status=404)

        with pytest.raises(ApiError):
            self.gh_client.check_file(self.repo, path, version)

        assert responses.calls[0].response.status_code == 404

    @responses.activate
    def test_get_stacktrace_link(self):
        path = "/src/sentry/integrations/github/client.py"
        version = "master"
        url = f"{self.base_url}/repos/{self.repo.name}/contents/{path.lstrip('/')}?ref={version}"

        responses.add(
            method=responses.HEAD,
            url=url,
            json={"text": 200},
        )

        source_url = self.install.get_stacktrace_link(self.repo, path, "master", version)
        assert (
            source_url
            == "https://github.example.org/Test-Organization/foo/blob/master/src/sentry/integrations/github/client.py"
        )

    @responses.activate
    def test_get_codeowner_file(self):
        self.config = self.create_code_mapping(
            repo=self.repo,
            project=self.project,
        )
        url = f"{self.base_url}/repos/{self.repo.name}/contents/CODEOWNERS?ref=master"

        responses.add(
            method=responses.HEAD,
            url=url,
            json={"text": 200},
        )
        responses.add(
            method=responses.GET,
            url=url,
            body="docs/*    @jianyuan   @getsentry/ecosystem\n* @jianyuan\n",
        )
        result = self.install.get_codeowner_file(
            self.config.repository, ref=self.config.default_branch
        )
        assert (
            responses.calls[1].request.headers["Content-Type"] == "application/raw; charset=utf-8"
        )
        assert result == GITHUB_CODEOWNERS