mirror of
https://github.com/Wei-Shaw/sub2api.git
synced 2026-03-30 02:09:43 +00:00
104 lines
3.1 KiB
Go
104 lines
3.1 KiB
Go
package service
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestOpsServiceRecordErrorBatch_SanitizesAndBatches(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var captured []*OpsInsertErrorLogInput
|
|
repo := &opsRepoMock{
|
|
BatchInsertErrorLogsFn: func(ctx context.Context, inputs []*OpsInsertErrorLogInput) (int64, error) {
|
|
captured = append(captured, inputs...)
|
|
return int64(len(inputs)), nil
|
|
},
|
|
}
|
|
svc := NewOpsService(repo, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil)
|
|
|
|
msg := " upstream failed: https://example.com?access_token=secret-value "
|
|
detail := `{"authorization":"Bearer secret-token"}`
|
|
entries := []*OpsInsertErrorLogInput{
|
|
{
|
|
ErrorBody: `{"error":"bad","access_token":"secret"}`,
|
|
UpstreamStatusCode: intPtr(-10),
|
|
UpstreamErrorMessage: strPtr(msg),
|
|
UpstreamErrorDetail: strPtr(detail),
|
|
UpstreamErrors: []*OpsUpstreamErrorEvent{
|
|
{
|
|
AccountID: -2,
|
|
UpstreamStatusCode: 429,
|
|
Message: " token leaked ",
|
|
Detail: `{"refresh_token":"secret"}`,
|
|
UpstreamRequestBody: `{"api_key":"secret","messages":[{"role":"user","content":"hello"}]}`,
|
|
},
|
|
},
|
|
},
|
|
{
|
|
ErrorPhase: "upstream",
|
|
ErrorType: "upstream_error",
|
|
CreatedAt: time.Now().UTC(),
|
|
},
|
|
}
|
|
|
|
require.NoError(t, svc.RecordErrorBatch(context.Background(), entries))
|
|
require.Len(t, captured, 2)
|
|
|
|
first := captured[0]
|
|
require.Equal(t, "internal", first.ErrorPhase)
|
|
require.Equal(t, "api_error", first.ErrorType)
|
|
require.Nil(t, first.UpstreamStatusCode)
|
|
require.NotNil(t, first.UpstreamErrorMessage)
|
|
require.NotContains(t, *first.UpstreamErrorMessage, "secret-value")
|
|
require.Contains(t, *first.UpstreamErrorMessage, "access_token=***")
|
|
require.NotNil(t, first.UpstreamErrorDetail)
|
|
require.NotContains(t, *first.UpstreamErrorDetail, "secret-token")
|
|
require.NotContains(t, first.ErrorBody, "secret")
|
|
require.Nil(t, first.UpstreamErrors)
|
|
require.NotNil(t, first.UpstreamErrorsJSON)
|
|
require.NotContains(t, *first.UpstreamErrorsJSON, "secret")
|
|
require.Contains(t, *first.UpstreamErrorsJSON, "[REDACTED]")
|
|
|
|
second := captured[1]
|
|
require.Equal(t, "upstream", second.ErrorPhase)
|
|
require.Equal(t, "upstream_error", second.ErrorType)
|
|
require.False(t, second.CreatedAt.IsZero())
|
|
}
|
|
|
|
func TestOpsServiceRecordErrorBatch_FallsBackToSingleInsert(t *testing.T) {
|
|
t.Parallel()
|
|
|
|
var (
|
|
batchCalls int
|
|
singleCalls int
|
|
)
|
|
repo := &opsRepoMock{
|
|
BatchInsertErrorLogsFn: func(ctx context.Context, inputs []*OpsInsertErrorLogInput) (int64, error) {
|
|
batchCalls++
|
|
return 0, errors.New("batch failed")
|
|
},
|
|
InsertErrorLogFn: func(ctx context.Context, input *OpsInsertErrorLogInput) (int64, error) {
|
|
singleCalls++
|
|
return int64(singleCalls), nil
|
|
},
|
|
}
|
|
svc := NewOpsService(repo, nil, nil, nil, nil, nil, nil, nil, nil, nil, nil)
|
|
|
|
err := svc.RecordErrorBatch(context.Background(), []*OpsInsertErrorLogInput{
|
|
{ErrorMessage: "first"},
|
|
{ErrorMessage: "second"},
|
|
})
|
|
require.NoError(t, err)
|
|
require.Equal(t, 1, batchCalls)
|
|
require.Equal(t, 2, singleCalls)
|
|
}
|
|
|
|
func strPtr(v string) *string {
|
|
return &v
|
|
}
|