mirror of
https://github.com/zitadel/zitadel.git
synced 2024-12-12 11:04:25 +00:00
713 lines
21 KiB
Go
713 lines
21 KiB
Go
|
package eventstore_test
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"sync"
|
||
|
|
"testing"
|
||
|
|
"time"
|
||
|
|
|
||
|
|
"github.com/zitadel/zitadel/internal/database"
|
||
|
|
"github.com/zitadel/zitadel/internal/eventstore"
|
||
|
|
)
|
||
|
|
|
||
|
|
func TestCRDB_Push_OneAggregate(t *testing.T) {
|
||
|
|
type args struct {
|
||
|
|
ctx context.Context
|
||
|
|
commands []eventstore.Command
|
||
|
|
uniqueDataType string
|
||
|
|
uniqueDataField string
|
||
|
|
uniqueDataInstanceID string
|
||
|
|
}
|
||
|
|
type eventsRes struct {
|
||
|
|
pushedEventsCount int
|
||
|
|
uniqueCount int
|
||
|
|
assetCount int
|
||
|
|
aggType eventstore.AggregateType
|
||
|
|
aggIDs database.TextArray[string]
|
||
|
|
}
|
||
|
|
type res struct {
|
||
|
|
wantErr bool
|
||
|
|
eventsRes eventsRes
|
||
|
|
}
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
args args
|
||
|
|
res res
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "push 1 event",
|
||
|
|
args: args{
|
||
|
|
ctx: context.Background(),
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "1"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 1,
|
||
|
|
aggIDs: []string{"1"},
|
||
|
|
aggType: eventstore.AggregateType(t.Name()),
|
||
|
|
}},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "push two events on agg",
|
||
|
|
args: args{
|
||
|
|
ctx: context.Background(),
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "6"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "6"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 2,
|
||
|
|
aggIDs: []string{"6"},
|
||
|
|
aggType: eventstore.AggregateType(t.Name()),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "failed push because context canceled",
|
||
|
|
args: args{
|
||
|
|
ctx: canceledCtx(),
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "9"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: true,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 0,
|
||
|
|
aggIDs: []string{"9"},
|
||
|
|
aggType: eventstore.AggregateType(t.Name()),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "push 1 event and add unique constraint",
|
||
|
|
args: args{
|
||
|
|
ctx: context.Background(),
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "10",
|
||
|
|
generateAddUniqueConstraint("usernames", "field"),
|
||
|
|
),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 1,
|
||
|
|
uniqueCount: 1,
|
||
|
|
aggIDs: []string{"10"},
|
||
|
|
aggType: eventstore.AggregateType(t.Name()),
|
||
|
|
}},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "push 1 event and remove unique constraint",
|
||
|
|
args: args{
|
||
|
|
ctx: context.Background(),
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "11",
|
||
|
|
generateRemoveUniqueConstraint("usernames", "testremove"),
|
||
|
|
),
|
||
|
|
},
|
||
|
|
uniqueDataType: "usernames",
|
||
|
|
uniqueDataField: "testremove",
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 1,
|
||
|
|
uniqueCount: 0,
|
||
|
|
aggIDs: []string{"11"},
|
||
|
|
aggType: eventstore.AggregateType(t.Name()),
|
||
|
|
}},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "push 1 event and remove instance unique constraints",
|
||
|
|
args: args{
|
||
|
|
ctx: context.Background(),
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "12",
|
||
|
|
generateRemoveUniqueConstraint("instance", "instanceID"),
|
||
|
|
),
|
||
|
|
},
|
||
|
|
uniqueDataType: "usernames",
|
||
|
|
uniqueDataField: "testremove",
|
||
|
|
uniqueDataInstanceID: "instanceID",
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 1,
|
||
|
|
uniqueCount: 0,
|
||
|
|
aggIDs: []string{"12"},
|
||
|
|
aggType: eventstore.AggregateType(t.Name()),
|
||
|
|
}},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "push 1 event and add asset",
|
||
|
|
args: args{
|
||
|
|
ctx: context.Background(),
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "13"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 1,
|
||
|
|
assetCount: 1,
|
||
|
|
aggIDs: []string{"13"},
|
||
|
|
aggType: eventstore.AggregateType(t.Name()),
|
||
|
|
}},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "push 1 event and remove asset",
|
||
|
|
args: args{
|
||
|
|
ctx: context.Background(),
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "14"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 1,
|
||
|
|
assetCount: 0,
|
||
|
|
aggIDs: []string{"14"},
|
||
|
|
aggType: eventstore.AggregateType(t.Name()),
|
||
|
|
}},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
for pusherName, pusher := range pushers {
|
||
|
|
t.Run(pusherName+"/"+tt.name, func(t *testing.T) {
|
||
|
|
t.Cleanup(cleanupEventstore(clients[pusherName]))
|
||
|
|
db := eventstore.NewEventstore(
|
||
|
|
&eventstore.Config{
|
||
|
|
Querier: queriers["v2(inmemory)"],
|
||
|
|
Pusher: pusher,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
if tt.args.uniqueDataType != "" && tt.args.uniqueDataField != "" {
|
||
|
|
err := fillUniqueData(tt.args.uniqueDataType, tt.args.uniqueDataField, tt.args.uniqueDataInstanceID)
|
||
|
|
if err != nil {
|
||
|
|
t.Error("unable to prefill insert unique data: ", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if _, err := db.Push(tt.args.ctx, tt.args.commands...); (err != nil) != tt.res.wantErr {
|
||
|
|
t.Errorf("CRDB.Push() error = %v, wantErr %v", err, tt.res.wantErr)
|
||
|
|
}
|
||
|
|
|
||
|
|
assertEventCount(t,
|
||
|
|
clients[pusherName],
|
||
|
|
database.TextArray[eventstore.AggregateType]{tt.res.eventsRes.aggType},
|
||
|
|
tt.res.eventsRes.aggIDs,
|
||
|
|
tt.res.eventsRes.pushedEventsCount,
|
||
|
|
)
|
||
|
|
|
||
|
|
assertUniqueConstraint(t, clients[pusherName], tt.args.commands, tt.res.eventsRes.uniqueCount)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCRDB_Push_MultipleAggregate(t *testing.T) {
|
||
|
|
type args struct {
|
||
|
|
commands []eventstore.Command
|
||
|
|
}
|
||
|
|
type eventsRes struct {
|
||
|
|
pushedEventsCount int
|
||
|
|
aggType database.TextArray[eventstore.AggregateType]
|
||
|
|
aggID database.TextArray[string]
|
||
|
|
}
|
||
|
|
type res struct {
|
||
|
|
wantErr bool
|
||
|
|
eventsRes eventsRes
|
||
|
|
}
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
args args
|
||
|
|
res res
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "push two aggregates",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "100"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "101"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 2,
|
||
|
|
aggID: []string{"100", "101"},
|
||
|
|
aggType: database.TextArray[eventstore.AggregateType]{eventstore.AggregateType(t.Name())},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "push two aggregates both multiple events",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "102"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "102"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "103"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "103"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 4,
|
||
|
|
aggID: []string{"102", "103"},
|
||
|
|
aggType: database.TextArray[eventstore.AggregateType]{eventstore.AggregateType(t.Name())},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "push two aggregates mixed multiple events",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "106"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "106"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "106"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "106"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "107"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "107"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "107"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "107"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "108"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "108"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "108"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "108"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
wantErr: false,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
pushedEventsCount: 12,
|
||
|
|
aggID: []string{"106", "107", "108"},
|
||
|
|
aggType: database.TextArray[eventstore.AggregateType]{eventstore.AggregateType(t.Name())},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
for pusherName, pusher := range pushers {
|
||
|
|
t.Run(pusherName+"/"+tt.name, func(t *testing.T) {
|
||
|
|
t.Cleanup(cleanupEventstore(clients[pusherName]))
|
||
|
|
|
||
|
|
db := eventstore.NewEventstore(
|
||
|
|
&eventstore.Config{
|
||
|
|
Querier: queriers["v2(inmemory)"],
|
||
|
|
Pusher: pusher,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
if _, err := db.Push(context.Background(), tt.args.commands...); (err != nil) != tt.res.wantErr {
|
||
|
|
t.Errorf("CRDB.Push() error = %v, wantErr %v", err, tt.res.wantErr)
|
||
|
|
}
|
||
|
|
|
||
|
|
assertEventCount(t, clients[pusherName], tt.res.eventsRes.aggType, tt.res.eventsRes.aggID, tt.res.eventsRes.pushedEventsCount)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCRDB_Push_Parallel(t *testing.T) {
|
||
|
|
type args struct {
|
||
|
|
commands [][]eventstore.Command
|
||
|
|
}
|
||
|
|
type eventsRes struct {
|
||
|
|
pushedEventsCount int
|
||
|
|
aggTypes database.TextArray[eventstore.AggregateType]
|
||
|
|
aggIDs database.TextArray[string]
|
||
|
|
}
|
||
|
|
type res struct {
|
||
|
|
minErrCount int
|
||
|
|
eventsRes eventsRes
|
||
|
|
}
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
args args
|
||
|
|
res res
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "clients push different aggregates",
|
||
|
|
args: args{
|
||
|
|
commands: [][]eventstore.Command{
|
||
|
|
{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "200"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "200"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "200"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "201"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "201"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "201"),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "202"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "203"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "203"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
minErrCount: 0,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
aggIDs: []string{"200", "201", "202", "203"},
|
||
|
|
pushedEventsCount: 9,
|
||
|
|
aggTypes: database.TextArray[eventstore.AggregateType]{eventstore.AggregateType(t.Name())},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "clients push same aggregates",
|
||
|
|
args: args{
|
||
|
|
commands: [][]eventstore.Command{
|
||
|
|
{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "204"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "204"),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "204"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "204"),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "204"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "204"),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "204"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "204"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
minErrCount: 1,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
aggIDs: []string{"204"},
|
||
|
|
pushedEventsCount: 6,
|
||
|
|
aggTypes: database.TextArray[eventstore.AggregateType]{eventstore.AggregateType(t.Name())},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "clients push different aggregates",
|
||
|
|
args: args{
|
||
|
|
commands: [][]eventstore.Command{
|
||
|
|
{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "207"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "207"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "207"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "207"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "207"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "207"),
|
||
|
|
},
|
||
|
|
{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "208"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "208"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "208"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "208"),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "208"),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
minErrCount: 0,
|
||
|
|
eventsRes: eventsRes{
|
||
|
|
aggIDs: []string{"207", "208"},
|
||
|
|
pushedEventsCount: 11,
|
||
|
|
aggTypes: database.TextArray[eventstore.AggregateType]{eventstore.AggregateType(t.Name())},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
for pusherName, pusher := range pushers {
|
||
|
|
t.Run(pusherName+"/"+tt.name, func(t *testing.T) {
|
||
|
|
t.Cleanup(cleanupEventstore(clients[pusherName]))
|
||
|
|
|
||
|
|
db := eventstore.NewEventstore(
|
||
|
|
&eventstore.Config{
|
||
|
|
Querier: queriers["v2(inmemory)"],
|
||
|
|
Pusher: pusher,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
errs := pushAggregates(db, tt.args.commands)
|
||
|
|
|
||
|
|
if len(errs) < tt.res.minErrCount {
|
||
|
|
t.Errorf("eventstore.Push() error count = %d, wanted err count %d, errs: %v", len(errs), tt.res.minErrCount, errs)
|
||
|
|
}
|
||
|
|
|
||
|
|
assertEventCount(t, clients[pusherName], tt.res.eventsRes.aggTypes, tt.res.eventsRes.aggIDs, tt.res.eventsRes.pushedEventsCount)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestCRDB_Push_ResourceOwner(t *testing.T) {
|
||
|
|
type args struct {
|
||
|
|
commands []eventstore.Command
|
||
|
|
}
|
||
|
|
type res struct {
|
||
|
|
resourceOwners database.TextArray[string]
|
||
|
|
}
|
||
|
|
type fields struct {
|
||
|
|
aggregateIDs database.TextArray[string]
|
||
|
|
aggregateType string
|
||
|
|
}
|
||
|
|
tests := []struct {
|
||
|
|
name string
|
||
|
|
args args
|
||
|
|
res res
|
||
|
|
fields fields
|
||
|
|
}{
|
||
|
|
{
|
||
|
|
name: "two events of same aggregate same resource owner",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "500", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "500", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
fields: fields{
|
||
|
|
aggregateIDs: []string{"500"},
|
||
|
|
aggregateType: t.Name(),
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
resourceOwners: []string{"caos", "caos"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "two events of different aggregate same resource owner",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "501", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "502", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
fields: fields{
|
||
|
|
aggregateIDs: []string{"501", "502"},
|
||
|
|
aggregateType: t.Name(),
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
resourceOwners: []string{"caos", "caos"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "two events of different aggregate different resource owner",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "503", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "504", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "zitadel" }),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
fields: fields{
|
||
|
|
aggregateIDs: []string{"503", "504"},
|
||
|
|
aggregateType: t.Name(),
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
resourceOwners: []string{"caos", "zitadel"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "events of different aggregate different resource owner",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "505", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "505", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "506", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "zitadel" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "506", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "zitadel" }),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
fields: fields{
|
||
|
|
aggregateIDs: []string{"505", "506"},
|
||
|
|
aggregateType: t.Name(),
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
resourceOwners: []string{"caos", "caos", "zitadel", "zitadel"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "events of different aggregate different resource owner per event",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "507", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "507", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "ignored" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "508", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "zitadel" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "508", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "ignored" }),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
fields: fields{
|
||
|
|
aggregateIDs: []string{"507", "508"},
|
||
|
|
aggregateType: t.Name(),
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
resourceOwners: []string{"caos", "caos", "zitadel", "zitadel"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
{
|
||
|
|
name: "events of one aggregate different resource owner per event",
|
||
|
|
args: args{
|
||
|
|
commands: []eventstore.Command{
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "509", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "caos" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "509", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "ignored" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "509", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "ignored" }),
|
||
|
|
generateCommand(eventstore.AggregateType(t.Name()), "509", func(e *testEvent) { e.BaseEvent.Agg.ResourceOwner = "ignored" }),
|
||
|
|
},
|
||
|
|
},
|
||
|
|
fields: fields{
|
||
|
|
aggregateIDs: []string{"509"},
|
||
|
|
aggregateType: t.Name(),
|
||
|
|
},
|
||
|
|
res: res{
|
||
|
|
resourceOwners: []string{"caos", "caos", "caos", "caos"},
|
||
|
|
},
|
||
|
|
},
|
||
|
|
}
|
||
|
|
for _, tt := range tests {
|
||
|
|
for pusherName, pusher := range pushers {
|
||
|
|
t.Run(pusherName+"/"+tt.name, func(t *testing.T) {
|
||
|
|
t.Cleanup(cleanupEventstore(clients[pusherName]))
|
||
|
|
|
||
|
|
db := eventstore.NewEventstore(
|
||
|
|
&eventstore.Config{
|
||
|
|
Querier: queriers["v2(inmemory)"],
|
||
|
|
Pusher: pusher,
|
||
|
|
},
|
||
|
|
)
|
||
|
|
|
||
|
|
events, err := db.Push(context.Background(), tt.args.commands...)
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("CRDB.Push() error = %v", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if len(events) != len(tt.res.resourceOwners) {
|
||
|
|
t.Errorf("length of events (%d) and resource owners (%d) must be equal", len(events), len(tt.res.resourceOwners))
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
for i, event := range events {
|
||
|
|
if event.Aggregate().ResourceOwner != tt.res.resourceOwners[i] {
|
||
|
|
t.Errorf("resource owner not expected want: %q got: %q", tt.res.resourceOwners[i], event.Aggregate().ResourceOwner)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
assertResourceOwners(t, clients[pusherName], tt.res.resourceOwners, tt.fields.aggregateIDs, tt.fields.aggregateType)
|
||
|
|
})
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func pushAggregates(pusher eventstore.Pusher, aggregateCommands [][]eventstore.Command) []error {
|
||
|
|
wg := sync.WaitGroup{}
|
||
|
|
errs := make([]error, 0)
|
||
|
|
errsMu := sync.Mutex{}
|
||
|
|
wg.Add(len(aggregateCommands))
|
||
|
|
|
||
|
|
ctx, cancel := context.WithCancel(context.Background())
|
||
|
|
|
||
|
|
for _, commands := range aggregateCommands {
|
||
|
|
go func(events []eventstore.Command) {
|
||
|
|
<-ctx.Done()
|
||
|
|
|
||
|
|
_, err := pusher.Push(context.Background(), events...) //nolint:contextcheck
|
||
|
|
if err != nil {
|
||
|
|
errsMu.Lock()
|
||
|
|
errs = append(errs, err)
|
||
|
|
errsMu.Unlock()
|
||
|
|
}
|
||
|
|
|
||
|
|
wg.Done()
|
||
|
|
}(commands)
|
||
|
|
}
|
||
|
|
|
||
|
|
// wait till all routines are started
|
||
|
|
time.Sleep(100 * time.Millisecond)
|
||
|
|
cancel()
|
||
|
|
wg.Wait()
|
||
|
|
|
||
|
|
return errs
|
||
|
|
}
|
||
|
|
|
||
|
|
func assertResourceOwners(t *testing.T, db *database.DB, resourceOwners, aggregateIDs database.TextArray[string], aggregateType string) {
|
||
|
|
t.Helper()
|
||
|
|
|
||
|
|
eventCount := 0
|
||
|
|
err := db.Query(func(rows *sql.Rows) error {
|
||
|
|
for i := 0; rows.Next(); i++ {
|
||
|
|
var resourceOwner string
|
||
|
|
err := rows.Scan(&resourceOwner)
|
||
|
|
if err != nil {
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
if resourceOwner != resourceOwners[i] {
|
||
|
|
t.Errorf("unexpected resource owner in queried event. want %q, got: %q", resourceOwners[i], resourceOwner)
|
||
|
|
}
|
||
|
|
eventCount++
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}, "SELECT owner FROM eventstore.events2 WHERE aggregate_type = $1 AND aggregate_id = ANY($2) ORDER BY position, in_tx_order", aggregateType, aggregateIDs)
|
||
|
|
if err != nil {
|
||
|
|
t.Error("query failed: ", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if eventCount != len(resourceOwners) {
|
||
|
|
t.Errorf("wrong queried event count: want %d, got %d", len(resourceOwners), eventCount)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func assertEventCount(t *testing.T, db *database.DB, aggTypes database.TextArray[eventstore.AggregateType], aggIDs database.TextArray[string], maxPushedEventsCount int) {
|
||
|
|
t.Helper()
|
||
|
|
|
||
|
|
var count int
|
||
|
|
err := db.QueryRow(func(row *sql.Row) error {
|
||
|
|
return row.Scan(&count)
|
||
|
|
}, "SELECT count(*) FROM eventstore.events2 where aggregate_type = ANY($1) AND aggregate_id = ANY($2)", aggTypes, aggIDs)
|
||
|
|
|
||
|
|
if err != nil {
|
||
|
|
t.Errorf("unexpected err in row.Scan: %v", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
if count > maxPushedEventsCount {
|
||
|
|
t.Errorf("expected push count %d got %d", maxPushedEventsCount, count)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func assertUniqueConstraint(t *testing.T, db *database.DB, commands []eventstore.Command, expectedCount int) {
|
||
|
|
t.Helper()
|
||
|
|
|
||
|
|
var uniqueConstraint *eventstore.UniqueConstraint
|
||
|
|
for _, command := range commands {
|
||
|
|
if e := command.(*testEvent); len(e.uniqueConstraints) > 0 {
|
||
|
|
uniqueConstraint = e.uniqueConstraints[0]
|
||
|
|
break
|
||
|
|
}
|
||
|
|
}
|
||
|
|
if uniqueConstraint == nil {
|
||
|
|
return
|
||
|
|
}
|
||
|
|
|
||
|
|
var uniqueCount int
|
||
|
|
err := db.QueryRow(func(row *sql.Row) error {
|
||
|
|
return row.Scan(&uniqueCount)
|
||
|
|
}, "SELECT COUNT(*) FROM eventstore.unique_constraints where unique_type = $1 AND unique_field = $2", uniqueConstraint.UniqueType, uniqueConstraint.UniqueField)
|
||
|
|
if err != nil {
|
||
|
|
t.Error("unable to query inserted rows: ", err)
|
||
|
|
return
|
||
|
|
}
|
||
|
|
if uniqueCount != expectedCount {
|
||
|
|
t.Errorf("expected unique count %d got %d", expectedCount, uniqueCount)
|
||
|
|
}
|
||
|
|
}
|