Fix create_mute_test_issue (#20765)

Author: naspirato

.github/scripts/tests/create_new_muted_ya.py

@@ -313,8 +313,9 @@ def create_mute_issues(all_tests, file_path, close_issues=True):
 
     # Check and close issues if needed
     closed_issues = []
+    partially_unmuted_issues = []
     if close_issues:
-        closed_issues = close_unmuted_issues(muted_tests_set)
+        closed_issues, partially_unmuted_issues = close_unmuted_issues(muted_tests_set)
 
     # First, collect all tests into temporary dictionary
     for test in all_tests:
@@ -380,25 +381,60 @@ def create_mute_issues(all_tests, file_path, close_issues=True):
 
     # Add closed issues section if any
     if closed_issues:
-        formatted_results.append("CLOSED ISSUES:")
+        formatted_results.append("🔒 **CLOSED ISSUES**")
+        formatted_results.append("─────────────────────────────")
+        formatted_results.append("")
         for issue in closed_issues:
-            formatted_results.append(f"Closed {issue['url']}")
-            formatted_results.append("Unmuted tests:")
+            formatted_results.append(f"✅ **Closed** {issue['url']}")
+            formatted_results.append("   📝 **Unmuted tests:**")
             for test in issue['tests']:
-                formatted_results.append(f"  - {test}")
+                formatted_results.append(f"   • `{test}`")
+            formatted_results.append("")
+    
+    # Add partially unmuted issues section if any
+    if partially_unmuted_issues:
+        if closed_issues:
+            formatted_results.append("─────────────────────────────")
+            formatted_results.append("")
+        formatted_results.append("🔓 **PARTIALLY UNMUTED ISSUES**")
+        formatted_results.append("─────────────────────────────")
         formatted_results.append("")
-        formatted_results.append("CREATED ISSUES:")
+        for issue in partially_unmuted_issues:
+            formatted_results.append(f"⚠️ **Partially unmuted** {issue['url']}")
+            formatted_results.append("   📝 **Unmuted tests:**")
+            for test in issue['unmuted_tests']:
+                formatted_results.append(f"   • `{test}`")
+            formatted_results.append("   🔒 **Still muted tests:**")
+            for test in issue['still_muted_tests']:
+                formatted_results.append(f"   • `{test}`")
+            formatted_results.append("")
 
-    # Add created issues
-    current_owner = None
-    for result in results:
-        if current_owner != result['owner']:
-            if formatted_results and formatted_results[-1] != "":  # Add blank line between owner groups if last line is not empty
-                formatted_results.append('')
-            current_owner = result['owner']
-            # Add owner header with team URL
-            formatted_results.append(f"TEAM:@ydb-platform/{current_owner} @https://github.com/orgs/ydb-platform/teams/{current_owner}")
-        formatted_results.append(result['message'])
+    # Add created issues section if any
+    if results:
+        if closed_issues or partially_unmuted_issues:
+            formatted_results.append("─────────────────────────────")
+            formatted_results.append("")
+        formatted_results.append("🆕 **CREATED ISSUES**")
+        formatted_results.append("─────────────────────────────")
+        formatted_results.append("")
+    
+        # Add created issues
+        current_owner = None
+        for result in results:
+            if current_owner != result['owner']:
+                if formatted_results and formatted_results[-1] != "":
+                    formatted_results.append('')
+                    formatted_results.append('')
+                current_owner = result['owner']
+                # Add owner header with team URL
+                formatted_results.append(f"👥 **TEAM** @ydb-platform/{current_owner}")
+                formatted_results.append(f"   https://github.com/orgs/ydb-platform/teams/{current_owner}")
+                formatted_results.append("   ┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄┄")
+            
+            # Extract issue URL and title
+            issue_url = result['message'].split('url ')[-1]
+            title = result['message'].split("'")[1]
+            formatted_results.append(f"   🎯 {issue_url} - `{title}`")
 
     print("\n\n")
     print("\n".join(formatted_results))

.github/scripts/tests/update_mute_issues.py

@@ -740,10 +740,11 @@ def close_unmuted_issues(muted_tests_set, do_not_close_issues=False):
         do_not_close_issues (bool): If True, issues will NOT be closed. If False, issues will be closed.
         
     Returns:
-        list: List of dictionaries containing information about closed issues
+        tuple: (closed_issues, partially_unmuted_issues) - lists of dictionaries containing information about closed and partially unmuted issues
     """
     issues = get_muted_tests_from_issues()
     closed_issues = []
+    partially_unmuted_issues = []
 
     # Get status field ID and Unmuted option ID
     _, project_fields = get_project_v2_fields(ORG_NAME, PROJECT_ID)
@@ -760,7 +761,7 @@ def close_unmuted_issues(muted_tests_set, do_not_close_issues=False):
 
     if not status_field_id or not unmuted_option_id:
         print("Warning: Could not find status field or Unmuted option")
-        return closed_issues
+        return closed_issues, partially_unmuted_issues
 
     # First, group tests by issue ID
     tests_by_issue = {}
@@ -781,6 +782,8 @@ def close_unmuted_issues(muted_tests_set, do_not_close_issues=False):
         if issue_info['state'] != 'CLOSED':
             unmuted_tests = [test for test in issue_info['tests'] if test not in muted_tests_set]
             if unmuted_tests:
+                # Get existing comments
+                existing_comments = get_issue_comments(issue_id)
 
                 # If all tests are unmuted, close the issue
                 if len(unmuted_tests) == len(issue_info['tests']):
@@ -797,19 +800,25 @@ def close_unmuted_issues(muted_tests_set, do_not_close_issues=False):
                     print(f"Unmuted tests: {', '.join(sorted(unmuted_tests))}")
                 # If some tests are unmuted but not all, just add a comment if needed
                 else:
-                    # Get existing comments
-                    existing_comments = get_issue_comments(issue_id)
                     if not has_unmute_comment(existing_comments, unmuted_tests):
                         comment = "Some tests have been unmuted:\n" + "\n".join(f"- Test {test}" for test in sorted(unmuted_tests))
                         add_issue_comment(issue_id, comment)
                         print(f"Added comment about unmuted tests to issue: {issue_info['url']}")
                     print(f"Unmuted tests: {', '.join(sorted(unmuted_tests))}")
+                    
+                    # Add to partially unmuted issues list
+                    still_muted_tests = [test for test in issue_info['tests'] if test in muted_tests_set]
+                    partially_unmuted_issues.append({
+                        'url': issue_info['url'],
+                        'unmuted_tests': sorted(unmuted_tests),
+                        'still_muted_tests': sorted(still_muted_tests)
+                    })
 
     # Update status for all closed issues
     print("Updating status for all closed issues...")
     update_all_closed_issues_status(status_field_id, unmuted_option_id)
 
-    return closed_issues
+    return closed_issues, partially_unmuted_issues
 
 
 def main():