refactoring tests with mock servers (#30) (#356)

2023-06-12 22:40:26 +09:00
parent a243e7331f
commit b616090e69
20 changed files with 732 additions and 1061 deletions
--- a/openai_test.go
+++ b/openai_test.go
@@ -0,0 +1,28 @@
+package openai_test
+
+import (
+	. "github.com/sashabaranov/go-openai"
+	"github.com/sashabaranov/go-openai/internal/test"
+)
+
+func setupOpenAITestServer() (client *Client, server *test.ServerTest, teardown func()) {
+	server = test.NewTestServer()
+	ts := server.OpenAITestServer()
+	ts.Start()
+	teardown = ts.Close
+	config := DefaultConfig(test.GetTestToken())
+	config.BaseURL = ts.URL + "/v1"
+	client = NewClientWithConfig(config)
+	return
+}
+
+func setupAzureTestServer() (client *Client, server *test.ServerTest, teardown func()) {
+	server = test.NewTestServer()
+	ts := server.OpenAITestServer()
+	ts.Start()
+	teardown = ts.Close
+	config := DefaultAzureConfig(test.GetTestToken(), "https://dummylab.openai.azure.com/")
+	config.BaseURL = ts.URL
+	client = NewClientWithConfig(config)
+	return
+}