Extend return type override unit test.

christianhelle · Jan 11, 2024 · 29f2775 · 29f2775
1 parent e15083d
commit 29f2775
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/src/Refitter.Tests/SwaggerPetstoreTests.cs b/src/Refitter.Tests/SwaggerPetstoreTests.cs
@@ -138,10 +138,17 @@ public async Task Can_Generate_Code_With_Type_Override(SampleOpenSpecifications
         var settings = new RefitGeneratorSettings
         {
             ReturnIApiResponse = false,
-            ResponseTypeOverride = { ["getPetById"] = "IApiResponse<Pet>" },
+            ResponseTypeOverride =
+            {
+                ["getPetById"] = "IApiResponse<Pet>", // Wrap existing type
+                ["deletePet"] = "Pet", // Add type where there was none
+                ["addPet"] = "void", // Remove type
+            },
         };
         var generateCode = await GenerateCode(version, filename, settings);
-        generateCode.Should().Contain("Task<IApiResponse<Pet>>", Exactly.Once());
+        generateCode.Should().Contain("Task<IApiResponse<Pet>> GetPetById");
+        generateCode.Should().Contain("Task<Pet> DeletePet");
+        generateCode.Should().Contain("Task AddPet");
     }
 
     [Theory]