fix mypy errors

getsentry · Oct 15, 2023 · db1b805 · db1b805
1 parent 99631fd
commit db1b805
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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
         )