diff --git a/tests/sentry/integrations/github_enterprise/test_client.py b/tests/sentry/integrations/github_enterprise/test_client.py index 7d15d6409d8e20..2a57d33fe14e46 100644 --- a/tests/sentry/integrations/github_enterprise/test_client.py +++ b/tests/sentry/integrations/github_enterprise/test_client.py @@ -4,9 +4,10 @@ import pytest import responses +from sentry.integrations.mixins.repositories import RepositoryMixin from sentry.models import Repository from sentry.shared_integrations.exceptions import ApiError -from sentry.testutils import TestCase +from sentry.testutils.cases import TestCase GITHUB_CODEOWNERS = { "filepath": "CODEOWNERS", @@ -87,6 +88,7 @@ def test_check_file(self): json={"text": 200}, ) + assert hasattr(self.client, "check_file") resp = self.client.check_file(self.repo, path, version) assert resp.status_code == 200 @@ -99,6 +101,7 @@ def test_check_no_file(self): responses.add(method=responses.HEAD, url=url, status=404) with pytest.raises(ApiError): + assert hasattr(self.client, "check_file") self.client.check_file(self.repo, path, version) assert responses.calls[1].response.status_code == 404 @@ -114,6 +117,7 @@ def test_get_stacktrace_link(self): json={"text": 200}, ) + assert isinstance(self.install, RepositoryMixin) source_url = self.install.get_stacktrace_link(self.repo, path, "master", version) assert ( source_url @@ -142,6 +146,7 @@ def test_get_codeowner_file(self, mock_check_file): url=url, json={"content": base64.b64encode(GITHUB_CODEOWNERS["raw"].encode()).decode("ascii")}, ) + assert isinstance(self.install, RepositoryMixin) result = self.install.get_codeowner_file( self.config.repository, ref=self.config.default_branch )