update unit tests

microsoft · Oct 21, 2024 · 29cf0cf · 29cf0cf
1 parent b0ac2ba
commit 29cf0cf
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/tests/Kiota.Builder.Tests/Writers/Python/CodeMethodWriterTests.cs b/tests/Kiota.Builder.Tests/Writers/Python/CodeMethodWriterTests.cs
@@ -792,12 +792,11 @@ public void WritesInheritedDeSerializerBody()
         Assert.DoesNotContain("defined_in_parent", result, StringComparison.OrdinalIgnoreCase);
     }
     [Fact]
-
-    public void Writes_UnionOfPrimitiveValues_FactoryFunctionAsync()
+    public void WritesFactoryMethodForPrimitiveTypes()
     {
         var result = tw.ToString();
-        Assert.Contains("return parseNode?.getNumberValue() ?? parseNode?.getStringValue();", result);
-        AssertExtensions.CurlyBracesAreClosed(result, 1);
+        Assert.Contains("return numberValue", result);
+        Assert.Contains("return stringValue", result);
     }
     [Fact]
     public void WritesUnionDeSerializerBody()