李建琦
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
99 lines
2.9 KiB
Go
99 lines
2.9 KiB
Go
package migrate
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"entgo.io/ent/dialect/entsql"
|
|
entschema "entgo.io/ent/dialect/sql/schema"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestAuthIdentityFoundationForeignKeyOnDeleteActions(t *testing.T) {
|
|
require.Equal(
|
|
t,
|
|
entschema.Cascade,
|
|
findForeignKeyBySymbol(t, AuthIdentitiesTable, "auth_identities_users_auth_identities").OnDelete,
|
|
)
|
|
require.Equal(
|
|
t,
|
|
entschema.Cascade,
|
|
findForeignKeyBySymbol(t, AuthIdentityChannelsTable, "auth_identity_channels_auth_identities_channels").OnDelete,
|
|
)
|
|
require.Equal(
|
|
t,
|
|
entschema.Cascade,
|
|
findForeignKeyBySymbol(t, IdentityAdoptionDecisionsTable, "identity_adoption_decisions_pending_auth_sessions_adoption_decision").OnDelete,
|
|
)
|
|
|
|
require.Equal(
|
|
t,
|
|
entschema.SetNull,
|
|
findForeignKeyBySymbol(t, PendingAuthSessionsTable, "pending_auth_sessions_users_pending_auth_sessions").OnDelete,
|
|
)
|
|
require.Equal(
|
|
t,
|
|
entschema.SetNull,
|
|
findForeignKeyBySymbol(t, IdentityAdoptionDecisionsTable, "identity_adoption_decisions_auth_identities_adoption_decisions").OnDelete,
|
|
)
|
|
}
|
|
|
|
func TestPaymentOrdersOutTradeNoPartialUniqueIndex(t *testing.T) {
|
|
idx := findIndexByName(t, PaymentOrdersTable, "paymentorder_out_trade_no")
|
|
require.True(t, idx.Unique)
|
|
require.Len(t, idx.Columns, 1)
|
|
require.Equal(t, "out_trade_no", idx.Columns[0].Name)
|
|
require.NotNil(t, idx.Annotation)
|
|
require.Equal(t, (&entsql.IndexAnnotation{Where: "out_trade_no <> ''"}).Where, idx.Annotation.Where)
|
|
}
|
|
|
|
func TestAccountsParentAccountForeignKey(t *testing.T) {
|
|
fk := findForeignKeyByColumn(t, AccountsTable, "parent_account_id")
|
|
require.Len(t, fk.Columns, 1)
|
|
require.Equal(t, "parent_account_id", fk.Columns[0].Name)
|
|
require.False(t, fk.Columns[0].Unique, "active-shadow uniqueness is enforced by the partial uq_accounts_spark_shadow_per_parent index")
|
|
require.Len(t, fk.RefColumns, 1)
|
|
require.Equal(t, "id", fk.RefColumns[0].Name)
|
|
require.Equal(t, entschema.Restrict, fk.OnDelete)
|
|
}
|
|
|
|
func findForeignKeyBySymbol(t *testing.T, table *entschema.Table, symbol string) *entschema.ForeignKey {
|
|
t.Helper()
|
|
|
|
for _, fk := range table.ForeignKeys {
|
|
if fk.Symbol == symbol {
|
|
return fk
|
|
}
|
|
}
|
|
|
|
require.Failf(t, "missing foreign key", "table %s should include foreign key %s", table.Name, symbol)
|
|
return nil
|
|
}
|
|
|
|
func findIndexByName(t *testing.T, table *entschema.Table, name string) *entschema.Index {
|
|
t.Helper()
|
|
|
|
for _, idx := range table.Indexes {
|
|
if idx.Name == name {
|
|
return idx
|
|
}
|
|
}
|
|
|
|
require.Failf(t, "missing index", "table %s should include index %s", table.Name, name)
|
|
return nil
|
|
}
|
|
|
|
func findForeignKeyByColumn(t *testing.T, table *entschema.Table, column string) *entschema.ForeignKey {
|
|
t.Helper()
|
|
|
|
for _, fk := range table.ForeignKeys {
|
|
for _, col := range fk.Columns {
|
|
if col.Name == column {
|
|
return fk
|
|
}
|
|
}
|
|
}
|
|
|
|
require.Failf(t, "missing foreign key", "table %s should include foreign key for column %s", table.Name, column)
|
|
return nil
|
|
}
|