X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/9b1705c5647864ef482078e1e169d6dfeade98b6..02cdaff8be27f37f44db8892bb12554f2d702323:/auth_test.go?ds=sidebyside

diff --git a/auth_test.go b/auth_test.go
index ccc7e66..0621605 100644
--- a/auth_test.go
+++ b/auth_test.go
@@ -169,9 +169,9 @@ func TestForcedRefreshLetsTheNextCallWork(t *testing.T) {
 	auth := newTestAuth(t, server)
 	writeTestTokens(t, auth, server.token(-time.Hour), "refresh-0")
 
-	responses := newFakeResponses(t)
-	client := newClient(responses.URL, auth, "agent-key", filepath.Join(t.TempDir(), "usage.jsonl"))
-	_, err := client.Create(t.Context(), testRequest())
+	responses := newModelFake(t)
+	client := newModelClient(responses.URL, auth, "agent-key", filepath.Join(t.TempDir(), "usage.jsonl"))
+	_, err := client.Create(t.Context(), modelTestRequest())
 	require.NoError(t, err)
 
 	require.Equal(t, 1, server.refreshes)
@@ -185,18 +185,18 @@ func TestModelEndpointFollowsTheConfig(t *testing.T) {
 	dir := t.TempDir()
 	config := defaultConfig()
 
-	t.Setenv("OPENAI_API_KEY", testKey)
+	t.Setenv("OPENAI_API_KEY", modelTestKey)
 	address, auth, err := modelEndpoint(config, dir)
 	require.NoError(t, err)
-	require.Equal(t, chatgptURL, address)
+	require.Equal(t, modelChatGPTURL, address)
 	require.IsType(t, &chatgptAuth{}, auth)
 	require.Equal(t, filepath.Join(dir, tokenFileName), auth.(*chatgptAuth).path)
 
 	config.Endpoint = endpointAPI
 	address, auth, err = modelEndpoint(config, dir)
 	require.NoError(t, err)
-	require.Equal(t, apiURL, address)
-	require.Equal(t, apiKey(testKey), auth)
+	require.Equal(t, modelAPIURL, address)
+	require.Equal(t, ModelAPIKey(modelTestKey), auth)
 
 	t.Setenv("OPENAI_API_KEY", "")
 	_, _, err = modelEndpoint(config, dir)