Function TestClassifyModelCallError

pkg/rag/strategy/vector_store_test.go:27–54

(t *testing.T)

)

func TestClassifyModelCallError(t *testing.T) {
	t.Parallel()
	tests := []struct {
		name        string
		err         error
		wantAborted bool
	}{
		{name: "nil", err: nil, wantAborted: false},
		{name: "404 model not found", err: &modelerrors.StatusError{StatusCode: http.StatusNotFound, Err: errors.New("not_found_error: model: claude-sonnet-4-7")}, wantAborted: true},
		{name: "401 unauthorized", err: &modelerrors.StatusError{StatusCode: http.StatusUnauthorized, Err: errors.New("unauthorized")}, wantAborted: true},
		{name: "429 rate limited", err: &modelerrors.StatusError{StatusCode: http.StatusTooManyRequests, Err: errors.New("too many requests")}, wantAborted: true},
		{name: "500 server error", err: &modelerrors.StatusError{StatusCode: http.StatusInternalServerError, Err: errors.New("server error")}, wantAborted: false},
		{name: "timeout message", err: errors.New("request timeout"), wantAborted: false},
		{name: "context canceled", err: context.Canceled, wantAborted: false},
		{name: "context deadline", err: context.DeadlineExceeded, wantAborted: false},
		{name: "wrapped 404", err: fmt.Errorf("batch 1 failed: %w", &modelerrors.StatusError{StatusCode: http.StatusNotFound, Err: errors.New("no such model")}), wantAborted: true},
	}

	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			got := classifyModelCallError(tt.err)
			assert.Equal(t, tt.wantAborted, isIndexingAborted(got))
			if tt.err != nil {
				assert.ErrorIs(t, got, tt.err)
			}
		})
	}
}

// fakeEmbeddingProvider counts embedding calls and always fails with a fixed error.
type fakeEmbeddingProvider struct {