20260130114916

Author: jeanschmidt

aws/lambda/pytorch-auto-revert/pytorch_auto_revert/tests/test_autorevert_circuit_breaker.py

@@ -35,7 +35,9 @@ def test_circuit_breaker_disabled_when_no_issues(self, mock_factory):
     @patch("pytorch_auto_revert.autorevert_circuit_breaker.GHClientFactory")
     def test_circuit_breaker_enabled_when_issue_exists(self, mock_factory):
         """Circuit breaker should be active when an open issue with label exists."""
-        mock_issue = create_mock_issue(number=12345, user_login="test-user", is_pr=False)
+        mock_issue = create_mock_issue(
+            number=12345, user_login="test-user", is_pr=False
+        )
         mock_repo = MagicMock()
         mock_repo.get_issues.return_value = [mock_issue]
         mock_factory.return_value.client.get_repo.return_value = mock_repo
@@ -64,7 +66,9 @@ def test_circuit_breaker_ignores_pull_requests(self, mock_factory):
     def test_circuit_breaker_with_mixed_issues_and_prs(self, mock_factory):
         """Circuit breaker should activate on issue but skip PR when both exist."""
         mock_pr = create_mock_issue(number=100, user_login="pr-author", is_pr=True)
-        mock_issue = create_mock_issue(number=200, user_login="issue-author", is_pr=False)
+        mock_issue = create_mock_issue(
+            number=200, user_login="issue-author", is_pr=False
+        )
         mock_repo = MagicMock()
         mock_repo.get_issues.return_value = [mock_pr, mock_issue]
         mock_factory.return_value.client.get_repo.return_value = mock_repo
@@ -105,7 +109,9 @@ def test_circuit_breaker_uses_custom_repo(self, mock_factory):
     @patch("pytorch_auto_revert.autorevert_circuit_breaker.GHClientFactory")
     def test_circuit_breaker_allows_approved_user(self, mock_factory):
         """Circuit breaker should activate when issue is created by approved user."""
-        mock_issue = create_mock_issue(number=12345, user_login="approved-user", is_pr=False)
+        mock_issue = create_mock_issue(
+            number=12345, user_login="approved-user", is_pr=False
+        )
         mock_repo = MagicMock()
         mock_repo.get_issues.return_value = [mock_issue]
         mock_factory.return_value.client.get_repo.return_value = mock_repo
@@ -118,7 +124,9 @@ def test_circuit_breaker_allows_approved_user(self, mock_factory):
     @patch("pytorch_auto_revert.autorevert_circuit_breaker.GHClientFactory")
     def test_circuit_breaker_blocks_unapproved_user(self, mock_factory):
         """Circuit breaker should not activate when issue is created by unapproved user."""
-        mock_issue = create_mock_issue(number=12345, user_login="unauthorized-user", is_pr=False)
+        mock_issue = create_mock_issue(
+            number=12345, user_login="unauthorized-user", is_pr=False
+        )
         mock_repo = MagicMock()
         mock_repo.get_issues.return_value = [mock_issue]
         mock_factory.return_value.client.get_repo.return_value = mock_repo
@@ -131,7 +139,9 @@ def test_circuit_breaker_blocks_unapproved_user(self, mock_factory):
         self.assertFalse(result)
         log_output = "\n".join(log_ctx.output)
         self.assertIn("Ignoring issue #12345", log_output)
-        self.assertIn("User 'unauthorized-user' is not in the approved users list", log_output)
+        self.assertIn(
+            "User 'unauthorized-user' is not in the approved users list", log_output
+        )
 
     @patch("pytorch_auto_revert.autorevert_circuit_breaker.GHClientFactory")
     def test_circuit_breaker_no_approval_check_when_empty_set(self, mock_factory):
@@ -162,8 +172,12 @@ def test_circuit_breaker_no_approval_check_when_none(self, mock_factory):
     @patch("pytorch_auto_revert.autorevert_circuit_breaker.GHClientFactory")
     def test_circuit_breaker_with_mixed_approved_and_unapproved(self, mock_factory):
         """Circuit breaker should activate if any issue is from approved user."""
-        mock_unapproved = create_mock_issue(number=100, user_login="unauthorized", is_pr=False)
-        mock_approved = create_mock_issue(number=200, user_login="approved-user", is_pr=False)
+        mock_unapproved = create_mock_issue(
+            number=100, user_login="unauthorized", is_pr=False
+        )
+        mock_approved = create_mock_issue(
+            number=200, user_login="approved-user", is_pr=False
+        )
         mock_repo = MagicMock()
         mock_repo.get_issues.return_value = [mock_unapproved, mock_approved]
         mock_factory.return_value.client.get_repo.return_value = mock_repo

torchci/lib/bot/autoLabelBot.ts

@@ -388,11 +388,14 @@ async function handleDisableAutorevertLabel(
   labelName: string
 ): Promise<void> {
   // Check if we already posted a warning
-  const comments = await context.octokit.issues.listComments(
-    context.repo({
-      issue_number: prNumber,
-    })
-  );
+  const owner = context.payload.repository.owner.login;
+  const repo = context.payload.repository.name;
+  
+  const comments = await context.octokit.issues.listComments({
+    owner,
+    repo,
+    issue_number: prNumber,
+  });
 
   const hasExistingWarning = comments.data.some((comment) =>
     comment.body?.includes(DISABLE_AUTOREVERT_WARNING_MARKER)
@@ -402,31 +405,31 @@ async function handleDisableAutorevertLabel(
   context.log(
     `Removing incorrect label "${labelName}" from PR ${prNumber}`
   );
-  await context.octokit.issues.removeLabel(
-    context.repo({
-      issue_number: prNumber,
-      name: labelName,
-    })
-  );
+  await context.octokit.issues.removeLabel({
+    owner,
+    repo,
+    issue_number: prNumber,
+    name: labelName,
+  });
 
   // Only post a comment if we haven't already warned about this
   if (!hasExistingWarning) {
     context.log(
       `Posting warning comment about incorrect label "${labelName}" on PR ${prNumber}`
     );
-    await context.octokit.issues.createComment(
-      context.repo({
-        issue_number: prNumber,
-        body: `${DISABLE_AUTOREVERT_WARNING_MARKER}
+    await context.octokit.issues.createComment({
+      owner,
+      repo,
+      issue_number: prNumber,
+      body: `${DISABLE_AUTOREVERT_WARNING_MARKER}
 The label \`${labelName}\` is incorrect for pull requests.
 
 This label is used to disable the **entire autorevert system** and should only be added to **issues**, not PRs.
 
 If you want to prevent autorevert from reverting **this specific PR**, please use the label \`autorevert: disable\` instead.
 
 The incorrect label has been removed.`,
-      })
-    );
+    });
   } else {
     context.log(
       `Skipping duplicate warning comment for "${labelName}" on PR ${prNumber} - warning already exists`