Add WritesWarningDisableCs0618, WritesWarningRestoreCs0618

microsoft · Aug 26, 2024 · df1010d · df1010d
1 parent e519975
commit df1010d
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 0 deletions.
diff --git a/src/Kiota.Builder/Writers/CSharp/CodeBlockEndWriter.cs b/src/Kiota.Builder/Writers/CSharp/CodeBlockEndWriter.cs
@@ -12,6 +12,7 @@ public override void WriteCodeElement(BlockEnd codeElement, LanguageWriter write
         if (codeElement?.Parent is CodeClass codeClass && codeClass.Parent is CodeNamespace)
         {
             writer.CloseBlock();
+            conventions.WritePragmaRestore(writer, CSharpConventionService.CS0618);
         }
     }
 }
diff --git a/tests/Kiota.Builder.Tests/Writers/CSharp/CodeClassDeclarationWriterTests.cs b/tests/Kiota.Builder.Tests/Writers/CSharp/CodeClassDeclarationWriterTests.cs
@@ -54,6 +54,14 @@ public void WritesSimpleDeclaration()
         Assert.Contains("public partial class", result);
     }
 
+    [Fact]
+    public void WritesWarningDisableCs0618()
+    {
+        codeElementWriter.WriteCodeElement(parentClass.StartBlock, writer);
+        var result = tw.ToString();
+        Assert.Contains("#pragma warning disable CS0618", result);
+    }
+
     [Fact]
     public void WritesImplementation()
     {

diff --git a/tests/Kiota.Builder.Tests/Writers/CSharp/CodeClassEndWriterTests.cs b/tests/Kiota.Builder.Tests/Writers/CSharp/CodeClassEndWriterTests.cs
@@ -45,6 +45,14 @@ public void ClosesNestedClasses()
         codeElementWriter.WriteCodeElement(child.EndBlock, writer);
         var result = tw.ToString();
         Assert.Equal(1, result.Count(x => x == '}'));
+        Assert.DoesNotContain("#pragma warning restore CS0618", result);
+    }
+    [Fact]
+    public void WritesWarningRestoreCs0618()
+    {
+        codeElementWriter.WriteCodeElement(parentClass.EndBlock, writer);
+        var result = tw.ToString();
+        Assert.Contains("#pragma warning restore CS0618", result);
     }
     [Fact]
     public void ClosesNonNestedClasses()