|
1 | 1 | from datetime import datetime, timezone
|
2 |
| -from unittest.mock import patch |
| 2 | +from unittest.mock import MagicMock, patch |
3 | 3 |
|
4 | 4 | from sentry.integrations.models.external_actor import ExternalActor
|
5 | 5 | from sentry.models.commit import Commit
|
|
18 | 18 |
|
19 | 19 |
|
20 | 20 | class CodeOwnersTest(TestCase):
|
21 |
| - def setUp(self): |
| 21 | + def setUp(self) -> None: |
22 | 22 | self.login_as(user=self.user)
|
23 | 23 |
|
24 | 24 | self.team = self.create_team(
|
@@ -49,7 +49,7 @@ def setUp(self):
|
49 | 49 | self.project, self.code_mapping, raw=self.data["raw"]
|
50 | 50 | )
|
51 | 51 |
|
52 |
| - def test_simple(self): |
| 52 | + def test_simple(self) -> None: |
53 | 53 | with self.tasks() and self.feature({"organizations:integrations-codeowners": True}):
|
54 | 54 | # new external team mapping
|
55 | 55 | self.external_team = self.create_external_team(integration=self.integration)
|
@@ -87,7 +87,9 @@ def test_simple(self):
|
87 | 87 | "sentry.integrations.github.integration.GitHubIntegration.get_codeowner_file",
|
88 | 88 | return_value=LATEST_GITHUB_CODEOWNERS,
|
89 | 89 | )
|
90 |
| - def test_codeowners_auto_sync_successful(self, mock_get_codeowner_file, mock_timezone_now): |
| 90 | + def test_codeowners_auto_sync_successful( |
| 91 | + self, mock_get_codeowner_file: MagicMock, mock_timezone_now: MagicMock |
| 92 | + ) -> None: |
91 | 93 | with self.tasks() and self.feature({"organizations:integrations-codeowners": True}):
|
92 | 94 | self.create_external_team()
|
93 | 95 | self.create_external_user(external_name="@NisanthanNanthakumar")
|
@@ -134,10 +136,9 @@ def test_codeowners_auto_sync_successful(self, mock_get_codeowner_file, mock_tim
|
134 | 136 | @patch("sentry.notifications.notifications.codeowners_auto_sync.AutoSyncNotification.send")
|
135 | 137 | def test_codeowners_auto_sync_failed_to_fetch_file(
|
136 | 138 | self,
|
137 |
| - mock_send_email, |
138 |
| - mock_get_codeowner_file, |
139 |
| - ): |
140 |
| - |
| 139 | + mock_send_email: MagicMock, |
| 140 | + mock_get_codeowner_file: MagicMock, |
| 141 | + ) -> None: |
141 | 142 | with self.tasks() and self.feature({"organizations:integrations-codeowners": True}):
|
142 | 143 | commit = Commit.objects.create(
|
143 | 144 | repository_id=self.repo.id,
|
|
0 commit comments