李建琦
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
83 lines
2.7 KiB
Go
83 lines
2.7 KiB
Go
package repository
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"strings"
|
|
"testing"
|
|
|
|
"github.com/DATA-DOG/go-sqlmock"
|
|
dbent "github.com/Wei-Shaw/sub2api/ent"
|
|
_ "github.com/Wei-Shaw/sub2api/ent/runtime"
|
|
"github.com/stretchr/testify/require"
|
|
|
|
"entgo.io/ent/dialect"
|
|
entsql "entgo.io/ent/dialect/sql"
|
|
)
|
|
|
|
type captureEntQueryMatcher struct {
|
|
actual *string
|
|
}
|
|
|
|
func (m captureEntQueryMatcher) Match(_, actual string) error {
|
|
if m.actual == nil {
|
|
return fmt.Errorf("query capture target is nil")
|
|
}
|
|
*m.actual = actual
|
|
return nil
|
|
}
|
|
|
|
func TestListSchedulableAccountLoadsUsesSingleProjectionQuery(t *testing.T) {
|
|
var capturedSQL string
|
|
db, mock, err := sqlmock.New(sqlmock.QueryMatcherOption(captureEntQueryMatcher{actual: &capturedSQL}))
|
|
require.NoError(t, err)
|
|
t.Cleanup(func() { _ = db.Close() })
|
|
|
|
driver := entsql.OpenDB(dialect.Postgres, db)
|
|
client := dbent.NewClient(dbent.Driver(driver))
|
|
t.Cleanup(func() { _ = client.Close() })
|
|
repo := newAccountRepositoryWithSQL(client, db, nil)
|
|
|
|
mock.ExpectQuery("schedulable account load projection").
|
|
WillReturnRows(sqlmock.NewRows([]string{"id", "concurrency", "load_factor"}).
|
|
AddRow(int64(11), 3, nil).
|
|
AddRow(int64(12), 2, 7))
|
|
|
|
loads, err := repo.ListSchedulableAccountLoads(context.Background())
|
|
require.NoError(t, err)
|
|
require.Len(t, loads, 2)
|
|
require.Equal(t, int64(11), loads[0].ID)
|
|
require.Equal(t, 3, loads[0].MaxConcurrency)
|
|
require.Equal(t, int64(12), loads[1].ID)
|
|
require.Equal(t, 7, loads[1].MaxConcurrency)
|
|
require.NoError(t, mock.ExpectationsWereMet(), "projection path must execute exactly one query")
|
|
|
|
normalized := normalizeSQLWhitespace(capturedSQL)
|
|
selectClause, _, found := strings.Cut(normalized, " FROM ")
|
|
require.True(t, found, "unexpected projection SQL: %s", normalized)
|
|
require.Equal(t, 2, strings.Count(selectClause, ","), "projection must select exactly three columns: %s", selectClause)
|
|
require.Contains(t, selectClause, `"id"`)
|
|
require.Contains(t, selectClause, `"concurrency"`)
|
|
require.Contains(t, selectClause, `"load_factor"`)
|
|
require.NotContains(t, selectClause, "credentials")
|
|
require.NotContains(t, selectClause, "extra")
|
|
require.NotContains(t, selectClause, "proxy_id")
|
|
require.NotContains(t, normalized, "account_groups")
|
|
require.NotContains(t, normalized, "proxies")
|
|
for _, predicateColumn := range []string{
|
|
"status",
|
|
"schedulable",
|
|
"temp_unschedulable_until",
|
|
"expires_at",
|
|
"auto_pause_on_expired",
|
|
"overload_until",
|
|
"rate_limit_reset_at",
|
|
"deleted_at",
|
|
} {
|
|
require.Contains(t, normalized, predicateColumn)
|
|
}
|
|
_, orderClause, hasOrder := strings.Cut(normalized, " ORDER BY ")
|
|
require.True(t, hasOrder, "projection query must preserve schedulable account order: %s", normalized)
|
|
require.Contains(t, orderClause, `"priority" ASC`)
|
|
}
|