From 23054edb230a6b50d176c8412c5a995c13d298df Mon Sep 17 00:00:00 2001 From: Ettore Di Giacinto Date: Tue, 21 Jan 2025 20:38:30 +0100 Subject: [PATCH] Tests improvements Signed-off-by: Ettore Di Giacinto --- core/http/app_test.go | 3 ++- tests/e2e-aio/e2e_test.go | 6 +++--- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/core/http/app_test.go b/core/http/app_test.go index c30de7b8..0a21995c 100644 --- a/core/http/app_test.go +++ b/core/http/app_test.go @@ -787,7 +787,7 @@ var _ = Describe("API test", func() { "application/json", bytes.NewBuffer([]byte(`{ "prompt": "a lovely cat", - "steps": 2, "seed":9000, + "step": 1, "seed":9000, "size": "256x256", "n":2}`))) // The response should contain an URL Expect(err).ToNot(HaveOccurred(), fmt.Sprint(resp)) @@ -796,6 +796,7 @@ var _ = Describe("API test", func() { imgUrlResp := &schema.OpenAIResponse{} err = json.Unmarshal(dat, imgUrlResp) + Expect(err).ToNot(HaveOccurred(), fmt.Sprint(dat)) Expect(imgUrlResp.Data).ToNot(Or(BeNil(), BeZero())) imgUrl := imgUrlResp.Data[0].URL Expect(imgUrl).To(ContainSubstring("http://127.0.0.1:9090/"), imgUrl) diff --git a/tests/e2e-aio/e2e_test.go b/tests/e2e-aio/e2e_test.go index a9c55497..3836f351 100644 --- a/tests/e2e-aio/e2e_test.go +++ b/tests/e2e-aio/e2e_test.go @@ -124,7 +124,7 @@ var _ = Describe("E2E test", func() { resp, err := client.CreateImage(context.TODO(), openai.ImageRequest{ Prompt: "test", - Size: openai.CreateImageSize512x512, + Size: openai.CreateImageSize256x256, }, ) Expect(err).ToNot(HaveOccurred()) @@ -135,7 +135,7 @@ var _ = Describe("E2E test", func() { resp, err := client.CreateImage(context.TODO(), openai.ImageRequest{ Prompt: "test", - Size: openai.CreateImageSize512x512, + Size: openai.CreateImageSize256x256, ResponseFormat: openai.CreateImageResponseFormatURL, }, ) @@ -147,7 +147,7 @@ var _ = Describe("E2E test", func() { resp, err := client.CreateImage(context.TODO(), openai.ImageRequest{ Prompt: "test", - Size: openai.CreateImageSize512x512, + Size: openai.CreateImageSize256x256, ResponseFormat: openai.CreateImageResponseFormatB64JSON, }, )