Name and order the memory core code

[onechat] / model_test.go

diff --git a/model_test.go b/model_test.go
index b90eaddca24bec8012cee558e4194585cc6942e2..7807400117825b89a8818f7e78b047cb519251f2 100644 (file)
--- 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")