X-Git-Url: https://git.giacomolli.uk/onechat/blobdiff_plain/087d725b5ab41ad34ffd24938482d45a8b99a080..3cdbd017da9ae7a7045f4272f16d4b262fd01c79:/model_test.go?ds=sidebyside diff --git a/model_test.go b/model_test.go index e932107..90d0e93 100644 --- a/model_test.go +++ b/model_test.go @@ -145,8 +145,10 @@ func TestClientRecordsEachCall(t *testing.T) { w.Header().Set("X-Codex-Plan-Type", "plus") w.Header().Set("X-Other", "not quota") w.Header().Set("X-Codex-Turn-State", "opaque state") + if call.number == 1 { w.WriteHeader(http.StatusTooManyRequests) + _, _ = io.WriteString(w, `{"detail": "slow down"}`) return @@ -176,6 +178,7 @@ func TestClientRecordsEachCall(t *testing.T) { "x-codex-primary-used-percent": "12", "x-codex-plan-type": "plus", } + require.Equal(t, "gpt-test", records[0].Model) require.Equal(t, http.StatusOK, records[0].Status) require.Equal(t, 500, records[0].InputTokens) @@ -213,11 +216,13 @@ func TestClientRefusesCallsOverTheLimit(t *testing.T) { for range 2 { require.False(t, client.LimitReached()) + _, err := client.Create(t.Context(), modelTestRequest()) require.NoError(t, err) } require.True(t, client.LimitReached()) + _, err := client.Create(t.Context(), modelTestRequest()) require.ErrorIs(t, err, errModelCallLimit) require.Equal(t, 2, server.Calls())