diff --git a/src/Refitter.Tests/SwaggerPetstoreTests.cs b/src/Refitter.Tests/SwaggerPetstoreTests.cs index e438dcb8..f51da8b0 100644 --- 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" }, + ResponseTypeOverride = + { + ["getPetById"] = "IApiResponse", // 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>", Exactly.Once()); + generateCode.Should().Contain("Task> GetPetById"); + generateCode.Should().Contain("Task DeletePet"); + generateCode.Should().Contain("Task AddPet"); } [Theory]