李建琦
6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
33 lines
1.2 KiB
Go
33 lines
1.2 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
"github.com/Wei-Shaw/sub2api/internal/service"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestBatchUpsertIngressRejectsUsesFixedMultiRowChunks(t *testing.T) {
|
|
db, mock, err := sqlmock.New()
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { _ = db.Close() })
|
|
repo := &opsRepository{db: db}
|
|
now := time.Now().UTC().Truncate(time.Minute)
|
|
items := make([]*service.OpsIngressRejectAggregate, ingressRejectUpsertChunkSize+1)
|
|
for i := range items {
|
|
items[i] = &service.OpsIngressRejectAggregate{
|
|
BucketStart: now, RejectReason: "invalid_api_key", RouteFamily: "messages",
|
|
Protocol: "anthropic", ClientIP: "192.0.2.1", RequestCount: 1, FirstSeen: now, LastSeen: now,
|
|
}
|
|
}
|
|
mock.ExpectBegin()
|
|
mock.ExpectExec("INSERT INTO ops_ingress_reject_aggregates").WillReturnResult(sqlmock.NewResult(0, int64(ingressRejectUpsertChunkSize)))
|
|
mock.ExpectExec("INSERT INTO ops_ingress_reject_aggregates").WillReturnResult(sqlmock.NewResult(0, 1))
|
|
mock.ExpectCommit()
|
|
require.NoError(t, repo.BatchUpsertIngressRejects(context.Background(), items))
|
|
require.NoError(t, mock.ExpectationsWereMet())
|
|
}
|