Handle trivia properly in PreferTestCleanupOverDisposeFixer (#4000)

Co-authored-by: Youssef Fahmy <[email protected]>

Author: Youssef1313

src/Analyzers/MSTest.Analyzers.CodeFixes/PreferTestCleanupOverDisposeFixer.cs

@@ -49,7 +49,7 @@ public override async Task RegisterCodeFixesAsync(CodeFixContext context)
             CodeAction.Create(
                 CodeFixResources.ReplaceWithTestCleanuFix,
                 c => ReplaceDisposeWithTestCleanupAsync(context.Document, methodDeclaration, c),
-                nameof(TestMethodShouldBeValidCodeFixProvider)),
+                nameof(PreferTestCleanupOverDisposeFixer)),
             diagnostic);
     }
 
@@ -83,7 +83,9 @@ private static async Task<Document> ReplaceDisposeWithTestCleanupAsync(Document
         else
         {
             // If no interfaces left, remove the base list entirely
+            SyntaxTriviaList trailingTrivia = newParent.BaseList?.GetTrailingTrivia() ?? SyntaxTriviaList.Empty;
             newParent = newParent.WithBaseList(null);
+            newParent = newParent.WithIdentifier(newParent.Identifier.WithTrailingTrivia(trailingTrivia));
         }
 
         editor.ReplaceNode(parentClass, newParent);

test/UnitTests/MSTest.Analyzers.UnitTests/PreferTestCleanupOverDisposeAnalyzerTests.cs

@@ -71,7 +71,8 @@ public class MyTestClass : IAsyncDisposable
             using Microsoft.VisualStudio.TestTools.UnitTesting;
 
             [TestClass]
-            public class MyTestClass {
+            public class MyTestClass
+            {
                 [TestCleanup]
                 public ValueTask TestCleanup()
                 {