Fix trivia for NUnit2046

nunit · Sep 18, 2024 · 859c2f8 · 859c2f8
1 parent 8b2e749
commit 859c2f8
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 1 deletion.
diff --git a/src/nunit.analyzers.tests/UseCollectionConstraint/UseCollectionConstraintCodeFixTests.cs b/src/nunit.analyzers.tests/UseCollectionConstraint/UseCollectionConstraintCodeFixTests.cs
@@ -51,6 +51,26 @@ public void TestMethod()
             RoslynAssert.CodeFix(analyzer, fix, expectedDiagnostic, code, fixedCode);
         }
 
+        [TestCaseSource(nameof(NumericContraints))]
+        public void VerifyLengthWithNewLine(string constraint)
+        {
+            var code = TestUtility.WrapMethodInClassNamespaceAndAddUsings($@"
+        public void TestMethod()
+        {{
+            var array = new int[] {{ 1 }};
+            Assert.That(↓array.Length,
+                Is.{constraint});
+        }}");
+            var fixedCode = TestUtility.WrapMethodInClassNamespaceAndAddUsings($@"
+        public void TestMethod()
+        {{
+            var array = new int[] {{ 1 }};
+            Assert.That(array,
+                Has.Length.{constraint});
+        }}");
+            RoslynAssert.CodeFix(analyzer, fix, expectedDiagnostic, code, fixedCode);
+        }
+
         [TestCaseSource(nameof(NumericContraints))]
         public void VerifyCount(string constraint)
         {

diff --git a/src/nunit.analyzers/UseCollectionConstraint/UseCollectionConstraintCodeFix.cs b/src/nunit.analyzers/UseCollectionConstraint/UseCollectionConstraintCodeFix.cs
@@ -85,7 +85,7 @@ await IsCollectionType(context, actualExpression.Expression).ConfigureAwait(fals
                 var hasPropertyExpression = SyntaxFactory.MemberAccessExpression(
                         SyntaxKind.SimpleMemberAccessExpression,
                         SyntaxFactory.IdentifierName(NUnitFrameworkConstants.NameOfHas),
-                        propertyName);
+                        propertyName).WithTriviaFrom(innerConstraintExpression);
 
                 nodesToReplace.Add(innerConstraintExpression, hasPropertyExpression);