update method writer tests

microsoft · Oct 21, 2024 · 3ee244b · 3ee244b
1 parent 9c7eebf
commit 3ee244b
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/tests/Kiota.Builder.Tests/Writers/Python/CodeMethodWriterTests.cs b/tests/Kiota.Builder.Tests/Writers/Python/CodeMethodWriterTests.cs
@@ -791,6 +791,17 @@ public void WritesInheritedDeSerializerBody()
         Assert.Contains("return fields", result);
         Assert.DoesNotContain("defined_in_parent", result, StringComparison.OrdinalIgnoreCase);
     }
+
+    [Fact]
+
+    public void Writes_UnionOfPrimitiveValues_FactoryFunctionAsync()
+    {
+
+        var result = tw.ToString();
+        Assert.Contains("return parseNode?.getNumberValue() ?? parseNode?.getStringValue();", result);
+        AssertExtensions.CurlyBracesAreClosed(result, 1);
+    }
+
     [Fact]
     public void WritesUnionDeSerializerBody()
     {