TheArchive/zitadel
1
0
Fork 0
mirror of https://github.com/zitadel/zitadel.git synced 2024-12-14 11:58:02 +00:00
Code Issues Packages Projects Releases Wiki Activity
dfb8c266d7
zitadel/internal/eventstore/v2/repository/sql/crdb_test.go

1236 lines
30 KiB
Go
Raw Normal View History

test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
package sql
import (
test(eventstore): push
2020-10-15 06:44:17 +00:00
"context"
try with goroutines
2020-10-19 07:53:32 +00:00
"sync"
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
"testing"
"github.com/caos/zitadel/internal/eventstore/v2/repository"
push tests
2020-10-15 11:25:25 +00:00
"github.com/lib/pq"
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
_ "github.com/lib/pq"
)
func TestCRDB_placeholder(t *testing.T) {
type args struct {
query string
}
type res struct {
query string
}
tests := []struct {
name string
args args
res res
}{
{
name: "no placeholders",
args: args{
query: "SELECT * FROM eventstore.events",
},
res: res{
query: "SELECT * FROM eventstore.events",
},
},
{
name: "one placeholder",
args: args{
query: "SELECT * FROM eventstore.events WHERE aggregate_type = ?",
},
res: res{
query: "SELECT * FROM eventstore.events WHERE aggregate_type = $1",
},
},
{
name: "multiple placeholders",
args: args{
query: "SELECT * FROM eventstore.events WHERE aggregate_type = ? AND aggregate_id = ? LIMIT ?",
},
res: res{
query: "SELECT * FROM eventstore.events WHERE aggregate_type = $1 AND aggregate_id = $2 LIMIT $3",
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{}
if query := db.placeholder(tt.args.query); query != tt.res.query {
t.Errorf("CRDB.placeholder() = %v, want %v", query, tt.res.query)
}
})
}
}
func TestCRDB_operation(t *testing.T) {
type res struct {
op string
}
type args struct {
operation repository.Operation
}
tests := []struct {
name string
args args
res res
}{
{
name: "no op",
args: args{
operation: repository.Operation(-1),
},
res: res{
op: "",
},
},
{
name: "greater",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
operation: repository.OperationGreater,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
op: ">",
},
},
{
name: "less",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
operation: repository.OperationLess,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
op: "<",
},
},
{
name: "equals",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
operation: repository.OperationEquals,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
op: "=",
},
},
{
name: "in",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
operation: repository.OperationIn,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
op: "=",
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{}
if got := db.operation(tt.args.operation); got != tt.res.op {
t.Errorf("CRDB.operation() = %v, want %v", got, tt.res.op)
}
})
}
}
func TestCRDB_conditionFormat(t *testing.T) {
type res struct {
format string
}
type args struct {
operation repository.Operation
}
tests := []struct {
name string
args args
res res
}{
{
name: "default",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
operation: repository.OperationEquals,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
format: "%s %s ?",
},
},
{
name: "in",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
operation: repository.OperationIn,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
format: "%s %s ANY(?)",
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{}
if got := db.conditionFormat(tt.args.operation); got != tt.res.format {
t.Errorf("CRDB.conditionFormat() = %v, want %v", got, tt.res.format)
}
})
}
}
func TestCRDB_columnName(t *testing.T) {
type res struct {
name string
}
type args struct {
field repository.Field
}
tests := []struct {
name string
args args
res res
}{
{
name: "invalid field",
args: args{
field: repository.Field(-1),
},
res: res{
name: "",
},
},
{
name: "aggregate id",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
field: repository.FieldAggregateID,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
name: "aggregate_id",
},
},
{
name: "aggregate type",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
field: repository.FieldAggregateType,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
name: "aggregate_type",
},
},
{
name: "editor service",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
field: repository.FieldEditorService,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
name: "editor_service",
},
},
{
name: "editor user",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
field: repository.FieldEditorUser,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
name: "editor_user",
},
},
{
name: "event type",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
field: repository.FieldEventType,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
name: "event_type",
},
},
{
name: "latest sequence",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
field: repository.FieldSequence,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
name: "event_sequence",
},
},
{
name: "resource owner",
args: args{
refactor: eventstore v2
2020-10-06 19:28:09 +00:00
field: repository.FieldResourceOwner,
test(eventstore): sql unit tests
2020-10-05 18:39:36 +00:00
},
res: res{
name: "resource_owner",
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{}
if got := db.columnName(tt.args.field); got != tt.res.name {
t.Errorf("CRDB.operation() = %v, want %v", got, tt.res.name)
}
})
}
}
test(eventstore): push
2020-10-15 06:44:17 +00:00
push tests
2020-10-15 11:25:25 +00:00
func TestCRDB_Push_OneAggregate(t *testing.T) {
test(eventstore): push
2020-10-15 06:44:17 +00:00
type args struct {
try with goroutines
2020-10-19 07:53:32 +00:00
ctx context.Context
test(eventstore): push
2020-10-15 06:44:17 +00:00
events []*repository.Event
}
push tests
2020-10-15 11:25:25 +00:00
type eventsRes struct {
pushedEventsCount int
aggType repository.AggregateType
aggID []string
}
type res struct {
wantErr bool
eventsRes eventsRes
}
test(eventstore): push
2020-10-15 06:44:17 +00:00
tests := []struct {
push tests
2020-10-15 11:25:25 +00:00
name string
args args
res res
test(eventstore): push
2020-10-15 06:44:17 +00:00
}{
{
name: "push 1 event with check previous",
args: args{
try with goroutines
2020-10-19 07:53:32 +00:00
ctx: context.Background(),
test(eventstore): push
2020-10-15 06:44:17 +00:00
events: []*repository.Event{
push tests
2020-10-15 11:25:25 +00:00
generateEvent(t, "1", true, 0),
},
},
res: res{
wantErr: false,
eventsRes: eventsRes{
pushedEventsCount: 1,
aggID: []string{"1"},
aggType: repository.AggregateType(t.Name()),
}},
},
{
name: "fail push 1 event with check previous wrong sequence",
args: args{
try with goroutines
2020-10-19 07:53:32 +00:00
ctx: context.Background(),
push tests
2020-10-15 11:25:25 +00:00
events: []*repository.Event{
generateEvent(t, "2", true, 5),
},
},
res: res{
wantErr: true,
eventsRes: eventsRes{
pushedEventsCount: 0,
aggID: []string{"2"},
aggType: repository.AggregateType(t.Name()),
},
},
},
{
name: "push 1 event without check previous",
args: args{
try with goroutines
2020-10-19 07:53:32 +00:00
ctx: context.Background(),
push tests
2020-10-15 11:25:25 +00:00
events: []*repository.Event{
generateEvent(t, "3", false, 0),
},
},
res: res{
wantErr: false,
eventsRes: eventsRes{
pushedEventsCount: 1,
aggID: []string{"3"},
aggType: repository.AggregateType(t.Name()),
},
},
},
{
name: "push 1 event without check previous wrong sequence",
args: args{
try with goroutines
2020-10-19 07:53:32 +00:00
ctx: context.Background(),
push tests
2020-10-15 11:25:25 +00:00
events: []*repository.Event{
generateEvent(t, "4", false, 5),
},
},
res: res{
wantErr: false,
eventsRes: eventsRes{
pushedEventsCount: 1,
aggID: []string{"4"},
aggType: repository.AggregateType(t.Name()),
},
},
},
{
name: "fail on push two events on agg without linking",
args: args{
try with goroutines
2020-10-19 07:53:32 +00:00
ctx: context.Background(),
push tests
2020-10-15 11:25:25 +00:00
events: []*repository.Event{
generateEvent(t, "5", true, 0),
generateEvent(t, "5", true, 0),
},
},
res: res{
wantErr: true,
eventsRes: eventsRes{
pushedEventsCount: 0,
aggID: []string{"5"},
aggType: repository.AggregateType(t.Name()),
},
},
},
{
name: "push two events on agg with linking",
args: args{
try with goroutines
2020-10-19 07:53:32 +00:00
ctx: context.Background(),
push tests
2020-10-15 11:25:25 +00:00
events: linkEvents(
generateEvent(t, "6", true, 0),
generateEvent(t, "6", true, 0),
),
},
res: res{
wantErr: false,
eventsRes: eventsRes{
pushedEventsCount: 2,
aggID: []string{"6"},
aggType: repository.AggregateType(t.Name()),
},
},
},
{
name: "push two events on agg with linking without check previous",
args: args{
try with goroutines
2020-10-19 07:53:32 +00:00
ctx: context.Background(),
push tests
2020-10-15 11:25:25 +00:00
events: linkEvents(
generateEvent(t, "7", false, 0),
generateEvent(t, "7", false, 0),
),
},
res: res{
wantErr: false,
eventsRes: eventsRes{
pushedEventsCount: 2,
aggID: []string{"7"},
aggType: repository.AggregateType(t.Name()),
test(eventstore): push
2020-10-15 06:44:17 +00:00
},
},
},
{
push tests
2020-10-15 11:25:25 +00:00
name: "push two events on agg with linking mixed check previous",
args: args{
try with goroutines
2020-10-19 07:53:32 +00:00
ctx: context.Background(),
push tests
2020-10-15 11:25:25 +00:00
events: linkEvents(
generateEvent(t, "8", false, 0),
generateEvent(t, "8", true, 0),
generateEvent(t, "8", false, 0),
generateEvent(t, "8", true, 0),
generateEvent(t, "8", true, 0),
),
},
res: res{
wantErr: false,
eventsRes: eventsRes{
pushedEventsCount: 5,
aggID: []string{"8"},
aggType: repository.AggregateType(t.Name()),
},
},
},
try with goroutines
2020-10-19 07:53:32 +00:00
{
name: "failed push because context canceled",
args: args{
ctx: canceledCtx(),
events: []*repository.Event{
generateEvent(t, "9", true, 0),
},
},
res: res{
wantErr: true,
eventsRes: eventsRes{
pushedEventsCount: 0,
aggID: []string{"9"},
aggType: repository.AggregateType(t.Name()),
},
},
},
push tests
2020-10-15 11:25:25 +00:00
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{
client: testCRDBClient,
}
try with goroutines
2020-10-19 07:53:32 +00:00
if err := db.Push(tt.args.ctx, tt.args.events...); (err != nil) != tt.res.wantErr {
push tests
2020-10-15 11:25:25 +00:00
t.Errorf("CRDB.Push() error = %v, wantErr %v", err, tt.res.wantErr)
}
countRow := testCRDBClient.QueryRow("SELECT COUNT(*) FROM eventstore.events where aggregate_type = $1 AND aggregate_id = ANY($2)", tt.res.eventsRes.aggType, pq.Array(tt.res.eventsRes.aggID))
var count int
err := countRow.Scan(&count)
if err != nil {
t.Error("unable to query inserted rows: ", err)
return
}
if count != tt.res.eventsRes.pushedEventsCount {
t.Errorf("expected push count %d got %d", tt.res.eventsRes.pushedEventsCount, count)
}
})
}
}
func TestCRDB_Push_MultipleAggregate(t *testing.T) {
type args struct {
events []*repository.Event
}
type eventsRes struct {
pushedEventsCount int
aggType []repository.AggregateType
aggID []string
}
type res struct {
wantErr bool
eventsRes eventsRes
}
tests := []struct {
name string
args args
res res
}{
{
name: "push two aggregates both check previous",
test(eventstore): push
2020-10-15 06:44:17 +00:00
args: args{
events: []*repository.Event{
push tests
2020-10-15 11:25:25 +00:00
generateEvent(t, "100", true, 0),
generateEvent(t, "101", true, 0),
},
},
res: res{
wantErr: false,
eventsRes: eventsRes{
pushedEventsCount: 2,
aggID: []string{"100", "101"},
aggType: []repository.AggregateType{repository.AggregateType(t.Name())},
},
},
},
{
name: "push two aggregates both check previous multiple events",
args: args{
events: combineEventLists(
linkEvents(
generateEvent(t, "102", true, 0),
generateEvent(t, "102", true, 0),
),
linkEvents(
generateEvent(t, "103", true, 0),
generateEvent(t, "103", true, 0),
),
),
},
res: res{
wantErr: false,
eventsRes: eventsRes{
pushedEventsCount: 4,
aggID: []string{"102", "103"},
aggType: []repository.AggregateType{repository.AggregateType(t.Name())},
},
},
},
{
name: "fail push linked events of different aggregates",
args: args{
events: linkEvents(
generateEvent(t, "104", false, 0),
try with goroutines
2020-10-19 07:53:32 +00:00
generateEvent(t, "105", false, 0),
push tests
2020-10-15 11:25:25 +00:00
),
},
res: res{
try with goroutines
2020-10-19 07:53:32 +00:00
wantErr: true,
push tests
2020-10-15 11:25:25 +00:00
eventsRes: eventsRes{
pushedEventsCount: 0,
try with goroutines
2020-10-19 07:53:32 +00:00
aggID: []string{"104", "105"},
aggType: []repository.AggregateType{repository.AggregateType(t.Name())},
},
},
},
{
name: "push two aggregates mixed check previous multiple events",
args: args{
events: combineEventLists(
linkEvents(
generateEvent(t, "106", true, 0),
generateEvent(t, "106", false, 0),
generateEvent(t, "106", false, 0),
generateEvent(t, "106", true, 0),
),
linkEvents(
generateEvent(t, "107", false, 0),
generateEvent(t, "107", true, 0),
generateEvent(t, "107", false, 0),
generateEvent(t, "107", true, 0),
),
linkEvents(
generateEvent(t, "108", true, 0),
generateEvent(t, "108", false, 0),
generateEvent(t, "108", false, 0),
generateEvent(t, "108", true, 0),
),
),
},
},
{
name: "failed push same aggregate in two transactions",
args: args{
events: combineEventLists(
linkEvents(
generateEvent(t, "109", true, 0),
),
linkEvents(
generateEvent(t, "109", true, 0),
),
),
},
res: res{
wantErr: true,
eventsRes: eventsRes{
pushedEventsCount: 0,
aggID: []string{"109"},
push tests
2020-10-15 11:25:25 +00:00
aggType: []repository.AggregateType{repository.AggregateType(t.Name())},
test(eventstore): push
2020-10-15 06:44:17 +00:00
},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{
client: testCRDBClient,
}
push tests
2020-10-15 11:25:25 +00:00
if err := db.Push(context.Background(), tt.args.events...); (err != nil) != tt.res.wantErr {
t.Errorf("CRDB.Push() error = %v, wantErr %v", err, tt.res.wantErr)
}
countRow := testCRDBClient.QueryRow("SELECT COUNT(*) FROM eventstore.events where aggregate_type = ANY($1) AND aggregate_id = ANY($2)", pq.Array(tt.res.eventsRes.aggType), pq.Array(tt.res.eventsRes.aggID))
var count int
err := countRow.Scan(&count)
if err != nil {
t.Error("unable to query inserted rows: ", err)
return
}
if count != tt.res.eventsRes.pushedEventsCount {
t.Errorf("expected push count %d got %d", tt.res.eventsRes.pushedEventsCount, count)
test(eventstore): push
2020-10-15 06:44:17 +00:00
}
})
}
}
push tests
2020-10-15 11:25:25 +00:00
try with goroutines
2020-10-19 07:53:32 +00:00
func TestCRDB_Push_Parallel(t *testing.T) {
type args struct {
events [][]*repository.Event
}
type eventsRes struct {
pushedEventsCount int
aggTypes []repository.AggregateType
aggIDs []string
}
type res struct {
errCount int
eventsRes eventsRes
}
tests := []struct {
name string
args args
res res
}{
{
name: "clients push different aggregates",
args: args{
events: [][]*repository.Event{
linkEvents(
generateEvent(t, "200", false, 0),
generateEvent(t, "200", true, 0),
generateEvent(t, "200", false, 0),
),
linkEvents(
generateEvent(t, "201", false, 0),
generateEvent(t, "201", true, 0),
generateEvent(t, "201", false, 0),
),
combineEventLists(
linkEvents(
generateEvent(t, "202", false, 0),
),
linkEvents(
generateEvent(t, "203", true, 0),
generateEvent(t, "203", false, 0),
),
),
},
},
res: res{
errCount: 0,
eventsRes: eventsRes{
aggIDs: []string{"200", "201", "202", "203"},
pushedEventsCount: 9,
aggTypes: []repository.AggregateType{repository.AggregateType(t.Name())},
},
},
},
{
name: "clients push same aggregates no check previous",
args: args{
events: [][]*repository.Event{
linkEvents(
generateEvent(t, "204", false, 0),
generateEvent(t, "204", false, 0),
),
linkEvents(
generateEvent(t, "204", false, 0),
generateEvent(t, "204", false, 0),
),
combineEventLists(
linkEvents(
generateEvent(t, "205", false, 0),
generateEvent(t, "205", false, 0),
generateEvent(t, "205", false, 0),
),
linkEvents(
generateEvent(t, "206", false, 0),
generateEvent(t, "206", false, 0),
generateEvent(t, "206", false, 0),
),
),
combineEventLists(
linkEvents(
generateEvent(t, "204", false, 0),
),
linkEvents(
generateEvent(t, "205", false, 0),
generateEvent(t, "205", false, 0),
),
linkEvents(
generateEvent(t, "206", false, 0),
),
),
},
},
res: res{
errCount: 0,
eventsRes: eventsRes{
aggIDs: []string{"204", "205", "206"},
pushedEventsCount: 14,
aggTypes: []repository.AggregateType{repository.AggregateType(t.Name())},
},
},
},
{
name: "clients push different aggregates one with check previous",
args: args{
events: [][]*repository.Event{
linkEvents(
generateEvent(t, "207", false, 0),
generateEvent(t, "207", false, 0),
generateEvent(t, "207", false, 0),
generateEvent(t, "207", false, 0),
generateEvent(t, "207", false, 0),
generateEvent(t, "207", false, 0),
),
linkEvents(
generateEvent(t, "208", true, 0),
generateEvent(t, "208", true, 0),
generateEvent(t, "208", true, 0),
generateEvent(t, "208", true, 0),
generateEvent(t, "208", true, 0),
),
},
},
res: res{
errCount: 0,
eventsRes: eventsRes{
aggIDs: []string{"207", "208"},
pushedEventsCount: 11,
aggTypes: []repository.AggregateType{repository.AggregateType(t.Name())},
},
},
},
{
name: "clients push different aggregates all with check previous on first event fail",
args: args{
events: [][]*repository.Event{
linkEvents(
generateEventWithData(t, "210", true, 0, []byte(`{ "transaction": 1 }`)),
generateEventWithData(t, "210", false, 0, []byte(`{ "transaction": 1.1 }`)),
),
linkEvents(
generateEventWithData(t, "210", true, 0, []byte(`{ "transaction": 2 }`)),
generateEventWithData(t, "210", false, 0, []byte(`{ "transaction": 2.1 }`)),
),
linkEvents(
generateEventWithData(t, "210", true, 0, []byte(`{ "transaction": 3 }`)),
generateEventWithData(t, "210", false, 0, []byte(`{ "transaction": 30.1 }`)),
),
},
},
res: res{
errCount: 2,
eventsRes: eventsRes{
aggIDs: []string{"210"},
pushedEventsCount: 2,
aggTypes: []repository.AggregateType{repository.AggregateType(t.Name())},
},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{
client: testCRDBClient,
}
wg := sync.WaitGroup{}
errs := make([]error, 0, tt.res.errCount)
errsMu := sync.Mutex{}
for _, events := range tt.args.events {
wg.Add(1)
go func(events []*repository.Event) {
err := db.Push(context.Background(), events...)
if err != nil {
errsMu.Lock()
errs = append(errs, err)
errsMu.Unlock()
}
wg.Done()
}(events)
}
wg.Wait()
if len(errs) != tt.res.errCount {
t.Errorf("CRDB.Push() error count = %d, wanted err count %d, errs: %v", len(errs), tt.res.errCount, errs)
}
rows, err := testCRDBClient.Query("SELECT event_data FROM eventstore.events where aggregate_type = ANY($1) AND aggregate_id = ANY($2) order by event_sequence", pq.Array(tt.res.eventsRes.aggTypes), pq.Array(tt.res.eventsRes.aggIDs))
if err != nil {
t.Error("unable to query inserted rows: ", err)
return
}
var count int
for rows.Next() {
count++
data := make(Data, 0)
err := rows.Scan(&data)
if err != nil {
t.Error("unable to query inserted rows: ", err)
return
}
t.Logf("inserted data: %v", string(data))
}
if count != tt.res.eventsRes.pushedEventsCount {
t.Errorf("expected push count %d got %d", tt.res.eventsRes.pushedEventsCount, count)
}
})
}
}
test: eventstore
2020-10-21 17:00:41 +00:00
func TestCRDB_Filter(t *testing.T) {
try with goroutines
2020-10-19 07:53:32 +00:00
type args struct {
searchQuery *repository.SearchQuery
}
type fields struct {
existingEvents []*repository.Event
}
type res struct {
test: eventstore query
2020-10-21 07:39:24 +00:00
eventCount int
try with goroutines
2020-10-19 07:53:32 +00:00
}
tests := []struct {
name string
fields fields
args args
res res
wantErr bool
}{
{
name: "aggregate type filter no events",
test: eventstore query
2020-10-21 07:39:24 +00:00
args: args{
searchQuery: &repository.SearchQuery{
Columns: repository.ColumnsEvent,
Filters: []*repository.Filter{
repository.NewFilter(repository.FieldAggregateType, "not found", repository.OperationEquals),
},
},
},
fields: fields{
existingEvents: []*repository.Event{
generateEvent(t, "300", false, 0),
generateEvent(t, "300", false, 0),
generateEvent(t, "300", false, 0),
},
},
res: res{
eventCount: 0,
},
wantErr: false,
},
{
test: eventstore
2020-10-21 17:00:41 +00:00
name: "aggregate type and id filter events found",
test: eventstore query
2020-10-21 07:39:24 +00:00
args: args{
searchQuery: &repository.SearchQuery{
Columns: repository.ColumnsEvent,
Filters: []*repository.Filter{
repository.NewFilter(repository.FieldAggregateType, t.Name(), repository.OperationEquals),
test: eventstore
2020-10-21 17:00:41 +00:00
repository.NewFilter(repository.FieldAggregateID, "303", repository.OperationEquals),
test: eventstore query
2020-10-21 07:39:24 +00:00
},
},
},
fields: fields{
existingEvents: []*repository.Event{
test: eventstore
2020-10-21 17:00:41 +00:00
generateEvent(t, "303", false, 0),
generateEvent(t, "303", false, 0),
generateEvent(t, "303", false, 0),
generateEvent(t, "305", false, 0),
test: eventstore query
2020-10-21 07:39:24 +00:00
},
},
res: res{
eventCount: 3,
},
wantErr: false,
},
test: eventstore
2020-10-21 17:00:41 +00:00
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{
client: testCRDBClient,
}
// setup initial data for query
if err := db.Push(context.Background(), tt.fields.existingEvents...); err != nil {
t.Errorf("error in setup = %v", err)
return
}
events, err := db.Filter(context.Background(), tt.args.searchQuery)
if (err != nil) != tt.wantErr {
t.Errorf("CRDB.query() error = %v, wantErr %v", err, tt.wantErr)
}
if len(events) != tt.res.eventCount {
t.Errorf("CRDB.query() expected event count: %d got %d", tt.res.eventCount, len(events))
}
})
}
}
func TestCRDB_LatestSequence(t *testing.T) {
type args struct {
searchQuery *repository.SearchQuery
}
type fields struct {
existingEvents []*repository.Event
}
type res struct {
sequence uint64
}
tests := []struct {
name string
fields fields
args args
res res
wantErr bool
}{
test: eventstore query
2020-10-21 07:39:24 +00:00
{
test: eventstore
2020-10-21 17:00:41 +00:00
name: "aggregate type filter no sequence",
test: eventstore query
2020-10-21 07:39:24 +00:00
args: args{
searchQuery: &repository.SearchQuery{
test: eventstore
2020-10-21 17:00:41 +00:00
Columns: repository.ColumnsMaxSequence,
test: eventstore query
2020-10-21 07:39:24 +00:00
Filters: []*repository.Filter{
test: eventstore
2020-10-21 17:00:41 +00:00
repository.NewFilter(repository.FieldAggregateType, "not found", repository.OperationEquals),
test: eventstore query
2020-10-21 07:39:24 +00:00
},
},
},
fields: fields{
existingEvents: []*repository.Event{
test: eventstore
2020-10-21 17:00:41 +00:00
generateEvent(t, "400", false, 0),
generateEvent(t, "400", false, 0),
generateEvent(t, "400", false, 0),
test: eventstore query
2020-10-21 07:39:24 +00:00
},
},
res: res{
test: eventstore
2020-10-21 17:00:41 +00:00
sequence: 0,
test: eventstore query
2020-10-21 07:39:24 +00:00
},
wantErr: false,
},
{
test: eventstore
2020-10-21 17:00:41 +00:00
name: "aggregate type filter sequence",
try with goroutines
2020-10-19 07:53:32 +00:00
args: args{
test: eventstore
2020-10-21 17:00:41 +00:00
searchQuery: &repository.SearchQuery{
Columns: repository.ColumnsMaxSequence,
Filters: []*repository.Filter{
repository.NewFilter(repository.FieldAggregateType, t.Name(), repository.OperationEquals),
},
},
try with goroutines
2020-10-19 07:53:32 +00:00
},
fields: fields{
test: eventstore
2020-10-21 17:00:41 +00:00
existingEvents: []*repository.Event{
generateEvent(t, "401", false, 0),
generateEvent(t, "401", false, 0),
generateEvent(t, "401", false, 0),
},
try with goroutines
2020-10-19 07:53:32 +00:00
},
res: res{
test: eventstore
2020-10-21 17:00:41 +00:00
sequence: 3,
try with goroutines
2020-10-19 07:53:32 +00:00
},
wantErr: false,
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{
client: testCRDBClient,
}
// setup initial data for query
if err := db.Push(context.Background(), tt.fields.existingEvents...); err != nil {
t.Errorf("error in setup = %v", err)
return
}
test: eventstore
2020-10-21 17:00:41 +00:00
sequence, err := db.LatestSequence(context.Background(), tt.args.searchQuery)
if (err != nil) != tt.wantErr {
try with goroutines
2020-10-19 07:53:32 +00:00
t.Errorf("CRDB.query() error = %v, wantErr %v", err, tt.wantErr)
}
test: eventstore
2020-10-21 17:00:41 +00:00
if sequence < tt.res.sequence {
t.Errorf("CRDB.query() expected sequence: %d got %d", tt.res.sequence, sequence)
}
try with goroutines
2020-10-19 07:53:32 +00:00
})
}
}
fix(eventstore): resource owner from previous event
2020-10-22 16:13:31 +00:00
func TestCRDB_Push_ResourceOwner(t *testing.T) {
type args struct {
events []*repository.Event
}
type res struct {
resourceOwners []string
}
type fields struct {
aggregateIDs []string
aggregateType string
}
tests := []struct {
name string
args args
res res
fields fields
}{
{
name: "two events of same aggregate same resource owner",
args: args{
events: []*repository.Event{
generateEvent(t, "500", false, 0, func(e *repository.Event) { e.ResourceOwner = "caos" }),
generateEvent(t, "500", false, 0, func(e *repository.Event) { e.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{
events: []*repository.Event{
generateEvent(t, "501", false, 0, func(e *repository.Event) { e.ResourceOwner = "caos" }),
generateEvent(t, "502", false, 0, func(e *repository.Event) { e.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{
events: []*repository.Event{
generateEvent(t, "503", false, 0, func(e *repository.Event) { e.ResourceOwner = "caos" }),
generateEvent(t, "504", false, 0, func(e *repository.Event) { e.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{
events: combineEventLists(
linkEvents(
generateEvent(t, "505", false, 0, func(e *repository.Event) { e.ResourceOwner = "caos" }),
generateEvent(t, "505", false, 0, func(e *repository.Event) { e.ResourceOwner = "caos" }),
),
linkEvents(
generateEvent(t, "506", false, 0, func(e *repository.Event) { e.ResourceOwner = "zitadel" }),
generateEvent(t, "506", false, 0, func(e *repository.Event) { e.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{
events: combineEventLists(
linkEvents(
generateEvent(t, "507", false, 0, func(e *repository.Event) { e.ResourceOwner = "caos" }),
generateEvent(t, "507", false, 0, func(e *repository.Event) { e.ResourceOwner = "ignored" }),
),
linkEvents(
generateEvent(t, "508", false, 0, func(e *repository.Event) { e.ResourceOwner = "zitadel" }),
generateEvent(t, "508", false, 0, func(e *repository.Event) { e.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{
events: combineEventLists(
linkEvents(
generateEvent(t, "509", false, 0, func(e *repository.Event) { e.ResourceOwner = "caos" }),
generateEvent(t, "509", false, 0, func(e *repository.Event) { e.ResourceOwner = "ignored" }),
),
linkEvents(
generateEvent(t, "509", false, 0, func(e *repository.Event) { e.ResourceOwner = "ignored" }),
generateEvent(t, "509", false, 0, func(e *repository.Event) { e.ResourceOwner = "ignored" }),
),
),
},
fields: fields{
aggregateIDs: []string{"509"},
aggregateType: t.Name(),
},
res: res{
resourceOwners: []string{"caos", "caos", "caos", "caos"},
},
},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
db := &CRDB{
client: testCRDBClient,
}
if err := db.Push(context.Background(), tt.args.events...); err != nil {
t.Errorf("CRDB.Push() error = %v", err)
}
if len(tt.args.events) != len(tt.res.resourceOwners) {
t.Errorf("length of events (%d) and resource owners (%d) must be equal", len(tt.args.events), len(tt.res.resourceOwners))
return
}
for i, event := range tt.args.events {
if event.ResourceOwner != tt.res.resourceOwners[i] {
t.Errorf("resource owner not expected want: %q got: %q", tt.res.resourceOwners[i], event.ResourceOwner)
}
}
rows, err := testCRDBClient.Query("SELECT resource_owner FROM eventstore.events WHERE aggregate_type = $1 AND aggregate_id = ANY($2) ORDER BY event_sequence", tt.fields.aggregateType, pq.Array(tt.fields.aggregateIDs))
if err != nil {
t.Error("unable to query inserted rows: ", err)
return
}
eventCount := 0
for i := 0; rows.Next(); i++ {
var resourceOwner string
err = rows.Scan(&resourceOwner)
if err != nil {
t.Error("unable to scan row: ", err)
return
}
if resourceOwner != tt.res.resourceOwners[i] {
t.Errorf("unexpected resource owner in queried event. want %q, got: %q", tt.res.resourceOwners[i], resourceOwner)
}
eventCount++
}
if eventCount != len(tt.res.resourceOwners) {
t.Errorf("wrong queried event count: want %d, got %d", len(tt.res.resourceOwners), eventCount)
}
})
}
}
try with goroutines
2020-10-19 07:53:32 +00:00
func canceledCtx() context.Context {
ctx, cancel := context.WithCancel(context.Background())
cancel()
return ctx
}
func combineEventLists(lists ...[]*repository.Event) []*repository.Event {
combined := make([]*repository.Event, 0)
for _, list := range lists {
combined = append(combined, list...)
}
return combined
push tests
2020-10-15 11:25:25 +00:00
}
func linkEvents(events ...*repository.Event) []*repository.Event {
for i := 1; i < len(events); i++ {
events[i].PreviousEvent = events[i-1]
}
return events
}
test: eventstore
2020-10-21 17:00:41 +00:00
func generateEvent(t *testing.T, aggregateID string, checkPrevious bool, previousSeq uint64, opts ...func(*repository.Event)) *repository.Event {
push tests
2020-10-15 11:25:25 +00:00
t.Helper()
test: eventstore
2020-10-21 17:00:41 +00:00
e := &repository.Event{
push tests
2020-10-15 11:25:25 +00:00
AggregateID: aggregateID,
AggregateType: repository.AggregateType(t.Name()),
CheckPreviousSequence: checkPrevious,
EditorService: "svc",
EditorUser: "user",
PreviousEvent: nil,
PreviousSequence: previousSeq,
ResourceOwner: "ro",
Type: "test.created",
Version: "v1",
}
test: eventstore
2020-10-21 17:00:41 +00:00
for _, opt := range opts {
opt(e)
}
return e
push tests
2020-10-15 11:25:25 +00:00
}
try with goroutines
2020-10-19 07:53:32 +00:00
func generateEventWithData(t *testing.T, aggregateID string, checkPrevious bool, previousSeq uint64, data []byte) *repository.Event {
t.Helper()
return &repository.Event{
AggregateID: aggregateID,
AggregateType: repository.AggregateType(t.Name()),
CheckPreviousSequence: checkPrevious,
EditorService: "svc",
EditorUser: "user",
PreviousEvent: nil,
PreviousSequence: previousSeq,
ResourceOwner: "ro",
Type: "test.created",
Version: "v1",
Data: data,
}
}
Reference in New Issue Copy Permalink