X-Git-Url: http://git.giacomolli.uk/onechat/blobdiff_plain/ba9faee0cd6c35020997794ef7bf134e8db12e96..d647f33b3e70f9419f6c0740e1d01b210439a519:/responses_test.go?ds=inline diff --git a/responses_test.go b/responses_test.go index de928bd..6017ad7 100644 --- a/responses_test.go +++ b/responses_test.go @@ -151,8 +151,12 @@ func TestClientRecordsEachCall(t *testing.T) { } client := newTestClient(t, server) - _, err := client.Create(t.Context(), testRequest()) + request := testRequest() + request.Node = &nodeKey{2, 5} + _, err := client.Create(t.Context(), request) require.NoError(t, err) + require.NotContains(t, server.Call(0).body, "node") + require.NotContains(t, string(server.Call(0).raw), "[2,5]") _, err = client.Create(t.Context(), testRequest()) require.ErrorContains(t, err, "HTTP 429") @@ -171,9 +175,15 @@ func TestClientRecordsEachCall(t *testing.T) { require.Equal(t, 400, records[0].CachedTokens) require.Equal(t, 7, records[0].OutputTokens) require.Equal(t, quota, records[0].Quota) + require.Equal(t, &[2]int{2, 5}, records[0].Node) + require.Equal(t, len("ok"), records[0].TextBytes) + require.Greater(t, records[0].Seconds, 0.0) require.Empty(t, records[0].Error) require.NotEmpty(t, records[0].Date) + require.Nil(t, records[1].Node) + require.Zero(t, records[1].TextBytes) + require.Equal(t, http.StatusTooManyRequests, records[1].Status) require.Equal(t, quota, records[1].Quota) require.Contains(t, records[1].Error, "HTTP 429")