Add assert for nullable return type tests

christianhelle · Jan 26, 2024 · 060d88d · 060d88d
1 parent bbf9705
commit 060d88d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/Refitter.Tests/SwaggerPetstoreTests.cs b/src/Refitter.Tests/SwaggerPetstoreTests.cs
@@ -576,6 +576,7 @@ public async Task Can_Generate_Code_With_NonNullable_Return_Types(SampleOpenSpec
             GenerateOptionalPropertiesAsNullable = true
         };
         var generateCode = await GenerateCode(version, filename, settings);
+        generateCode.Should().Contain("Task<Pet>");
         generateCode.Should().NotContain("Task<Pet?>");
     }
 
@@ -594,6 +595,7 @@ public async Task Can_Generate_Code_With_NonNullable_Return_Types_In_ApiResponse
             GenerateOptionalPropertiesAsNullable = true
         };
         var generateCode = await GenerateCode(version, filename, settings);
+        generateCode.Should().Contain("Task<IApiResponse<Pet>>");
         generateCode.Should().NotContain("Task<IApiResponse<Pet?>>");
     }
 }