ci: fixes preview generated code unit test failure introduced by #4907

Signed-off-by: Vincent Biret <[email protected]>
microsoft · Jul 5, 2024 · 8f5a624 · 8f5a624
1 parent 4adb397
commit 8f5a624
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/tests/Kiota.Builder.Tests/Writers/CSharp/CodeClassDeclarationWriterTests.cs b/tests/Kiota.Builder.Tests/Writers/CSharp/CodeClassDeclarationWriterTests.cs
@@ -113,7 +113,6 @@ public void WritesGeneratedCodeAttribute()
     {
         codeElementWriter.WriteCodeElement(parentClass.StartBlock, writer);
         var result = tw.ToString();
-        var pattern = @"\s+\[global::System\.CodeDom\.Compiler\.GeneratedCode\(""Kiota"", ""[0-9]+\.[0-9]+\.[0-9]+\""\)\]";
-        Assert.Matches(pattern, result);
+        Assert.Matches(CodeEnumWriterTests.GeneratedCodePattern, result);
     }
 }
diff --git a/tests/Kiota.Builder.Tests/Writers/CSharp/CodeEnumWriterTests.cs b/tests/Kiota.Builder.Tests/Writers/CSharp/CodeEnumWriterTests.cs
@@ -130,13 +130,13 @@ public void DoesntWriteAnythingOnNoOption()
         Assert.Empty(result);
     }
 
+    internal const string GeneratedCodePattern = @"\s+\[global::System\.CodeDom\.Compiler\.GeneratedCode\(""Kiota"", ""[0-9]+\.[0-9]+\.[0-9]+(?:-preview\.\d+)?""\)\]";
     [Fact]
     public void WritesGeneratedCodeAttribute()
     {
         currentEnum.AddOption(new CodeEnumOption { Name = "option2" });
         writer.Write(currentEnum);
         var result = tw.ToString();
-        var pattern = @"\s+\[global::System\.CodeDom\.Compiler\.GeneratedCode\(""Kiota"", ""[0-9]+\.[0-9]+\.[0-9]+\""\)\]";
-        Assert.Matches(pattern, result);
+        Assert.Matches(GeneratedCodePattern, result);
     }
 }