|
| 1 | +using System.Net; |
| 2 | +using System.Text; |
| 3 | +using System.Text.Json; |
| 4 | + |
| 5 | +namespace tryAGI.OpenAI.IntegrationTests; |
| 6 | + |
| 7 | +public partial class Tests |
| 8 | +{ |
| 9 | + [TestMethod] |
| 10 | + [DataRow(CreateImageRequestQuality.Medium, "medium")] |
| 11 | + [DataRow(CreateImageRequestQuality.High, "high")] |
| 12 | + [DataRow(CreateImageRequestQuality.Auto, "auto")] |
| 13 | + public async Task CreateImage_SerializesGptImageQualityValues( |
| 14 | + CreateImageRequestQuality quality, |
| 15 | + string expectedWireValue) |
| 16 | + { |
| 17 | + string? capturedBody = null; |
| 18 | + |
| 19 | + using var httpClient = new HttpClient(new ScriptedHttpMessageHandler(async (request, cancellationToken) => |
| 20 | + { |
| 21 | + request.Method.Should().Be(HttpMethod.Post); |
| 22 | + request.RequestUri.Should().Be(new Uri("https://api.openai.com/v1/images/generations")); |
| 23 | + request.Headers.Authorization?.Scheme.Should().Be("Bearer"); |
| 24 | + request.Headers.Authorization?.Parameter.Should().Be("test-api-key"); |
| 25 | + |
| 26 | + capturedBody = await request.Content!.ReadAsStringAsync(cancellationToken).ConfigureAwait(false); |
| 27 | + |
| 28 | + return new HttpResponseMessage(HttpStatusCode.OK) |
| 29 | + { |
| 30 | + Content = new StringContent( |
| 31 | + """ |
| 32 | + { |
| 33 | + "created": 1740000000, |
| 34 | + "data": [ |
| 35 | + { |
| 36 | + "b64_json": "iVBORw0KGgo=" |
| 37 | + } |
| 38 | + ], |
| 39 | + "quality": "medium", |
| 40 | + "size": "1024x1024", |
| 41 | + "output_format": "png" |
| 42 | + } |
| 43 | + """, |
| 44 | + Encoding.UTF8, |
| 45 | + "application/json") |
| 46 | + }; |
| 47 | + })) |
| 48 | + { |
| 49 | + BaseAddress = new Uri(OpenAiClient.DefaultBaseUrl) |
| 50 | + }; |
| 51 | + |
| 52 | + using var api = new OpenAiClient( |
| 53 | + "test-api-key", |
| 54 | + httpClient: httpClient, |
| 55 | + disposeHttpClient: false); |
| 56 | + |
| 57 | + var response = await api.Images.CreateImageAsync( |
| 58 | + prompt: "Generate a test image", |
| 59 | + model: CreateImageRequestModel.GptImage15, |
| 60 | + n: 1, |
| 61 | + quality: quality, |
| 62 | + size: CreateImageRequestSize.x1024x1024, |
| 63 | + outputFormat: CreateImageRequestOutputFormat.Png); |
| 64 | + |
| 65 | + capturedBody.Should().NotBeNullOrWhiteSpace(); |
| 66 | + var requestJson = JsonDocument.Parse(capturedBody!); |
| 67 | + var root = requestJson.RootElement; |
| 68 | + |
| 69 | + root.GetProperty("model").GetString().Should().Be("gpt-image-1.5"); |
| 70 | + root.GetProperty("quality").GetString().Should().Be(expectedWireValue); |
| 71 | + root.GetProperty("size").GetString().Should().Be("1024x1024"); |
| 72 | + root.GetProperty("output_format").GetString().Should().Be("png"); |
| 73 | + capturedBody.Should().NotContain("\"standard\""); |
| 74 | + capturedBody.Should().NotContain("\"hd\""); |
| 75 | + |
| 76 | + response.Data.Should().HaveCount(1); |
| 77 | + response.Data![0].B64Json.Should().Be("iVBORw0KGgo="); |
| 78 | + } |
| 79 | + |
| 80 | + private sealed class ScriptedHttpMessageHandler( |
| 81 | + Func<HttpRequestMessage, CancellationToken, Task<HttpResponseMessage>> handler) |
| 82 | + : HttpMessageHandler |
| 83 | + { |
| 84 | + protected override Task<HttpResponseMessage> SendAsync(HttpRequestMessage request, CancellationToken cancellationToken) |
| 85 | + => handler(request, cancellationToken); |
| 86 | + } |
| 87 | +} |
0 commit comments