X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/02cdaff8be27f37f44db8892bb12554f2d702323..ce0eacc2fe5090810cf22e022af3ab83006068a4:/model_test.go

diff --git a/model_test.go b/model_test.go
index b90eadd..7807400 100644
--- a/model_test.go
+++ b/model_test.go
@@ -62,9 +62,9 @@ func TestClientSendsTheFixedRequestFields(t *testing.T) {
 }
 
 func TestClientSendsTheSessionHeaderToChatGPT(t *testing.T) {
-	login := newFakeAuth(t)
-	auth := newTestAuth(t, login)
-	writeTestTokens(t, auth, login.token(time.Hour), "refresh-0")
+	login := newChatGPTFake(t)
+	auth := newTestChatGPTAuth(t, login)
+	chatgptWriteTestTokens(t, auth, login.token(time.Hour), "refresh-0")
 
 	testCases := []struct {
 		name string
@@ -152,7 +152,7 @@ func TestClientRecordsEachCall(t *testing.T) {
 
 	client := newTestModelClient(t, server)
 	request := modelTestRequest()
-	request.Node = &nodeKey{2, 5}
+	request.Node = &NodeKey{2, 5}
 	_, err := client.Create(t.Context(), request)
 	require.NoError(t, err)
 	require.NotContains(t, server.Call(0).body, "node")