zitadel/internal/eventstore/eventstore_querier_test.go

package eventstore_test

import (
	"context"
	"testing"

	"github.com/zitadel/zitadel/internal/eventstore"
)

func TestEventstore_Filter(t *testing.T) {
	type args struct {
		searchQuery *eventstore.SearchQueryBuilder
	}
	type fields struct {
		existingEvents []eventstore.Command
	}
	type res struct {
		eventCount int
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		res     res
		wantErr bool
	}{
		{
			name: "aggregate type filter no events",
			args: args{
				searchQuery: eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
					AddQuery().
					AggregateTypes("not found").
					Builder(),
			},
			fields: fields{
				existingEvents: []eventstore.Command{
					generateCommand(eventstore.AggregateType(t.Name()), "300"),
					generateCommand(eventstore.AggregateType(t.Name()), "300"),
					generateCommand(eventstore.AggregateType(t.Name()), "300"),
				},
			},
			res: res{
				eventCount: 0,
			},
			wantErr: false,
		},
		{
			name: "aggregate type and id filter events found",
			args: args{
				searchQuery: eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
					AddQuery().
					AggregateTypes(eventstore.AggregateType(t.Name())).
					AggregateIDs("303").
					Builder(),
			},
			fields: fields{
				existingEvents: []eventstore.Command{
					generateCommand(eventstore.AggregateType(t.Name()), "303"),
					generateCommand(eventstore.AggregateType(t.Name()), "303"),
					generateCommand(eventstore.AggregateType(t.Name()), "303"),
					generateCommand(eventstore.AggregateType(t.Name()), "305"),
				},
			},
			res: res{
				eventCount: 3,
			},
			wantErr: false,
		},
		{
			name: "exclude aggregate type and event type",
			args: args{
				searchQuery: eventstore.NewSearchQueryBuilder(eventstore.ColumnsEvent).
					AddQuery().
					AggregateTypes(eventstore.AggregateType(t.Name())).
					Builder().
					ExcludeAggregateIDs().
					EventTypes("test.updated").
					AggregateTypes(eventstore.AggregateType(t.Name())).
					Builder(),
			},
			fields: fields{
				existingEvents: []eventstore.Command{
					generateCommand(eventstore.AggregateType(t.Name()), "306"),
					generateCommand(
						eventstore.AggregateType(t.Name()),
						"306",
						func(te *testEvent) {
							te.EventType = "test.updated"
						},
					),
					generateCommand(
						eventstore.AggregateType(t.Name()),
						"308",
					),
				},
			},
			res: res{
				eventCount: 1,
			},
			wantErr: false,
		},
	}
	for _, tt := range tests {
		for querierName, querier := range queriers {
			t.Run(querierName+"/"+tt.name, func(t *testing.T) {
				t.Cleanup(cleanupEventstore(clients[querierName]))

				db := eventstore.NewEventstore(
					&eventstore.Config{
						Querier: querier,
						Pusher:  pushers["v3(inmemory)"],
					},
				)

				// 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("eventstore.query() error = %v, wantErr %v", err, tt.wantErr)
				}

				if len(events) != tt.res.eventCount {
					t.Errorf("eventstore.query() expected event count: %d got %d", tt.res.eventCount, len(events))
				}
			})
		}
	}
}

func TestEventstore_LatestSequence(t *testing.T) {
	type args struct {
		searchQuery *eventstore.SearchQueryBuilder
	}
	type fields struct {
		existingEvents []eventstore.Command
	}
	type res struct {
		sequence float64
	}
	tests := []struct {
		name    string
		fields  fields
		args    args
		res     res
		wantErr bool
	}{
		{
			name: "aggregate type filter no sequence",
			args: args{
				searchQuery: eventstore.NewSearchQueryBuilder(eventstore.ColumnsMaxSequence).
					AddQuery().
					AggregateTypes("not found").
					Builder(),
			},
			fields: fields{
				existingEvents: []eventstore.Command{
					generateCommand(eventstore.AggregateType(t.Name()), "400"),
					generateCommand(eventstore.AggregateType(t.Name()), "400"),
					generateCommand(eventstore.AggregateType(t.Name()), "400"),
				},
			},
			wantErr: false,
		},
		{
			name: "aggregate type filter sequence",
			args: args{
				searchQuery: eventstore.NewSearchQueryBuilder(eventstore.ColumnsMaxSequence).
					AddQuery().
					AggregateTypes(eventstore.AggregateType(t.Name())).
					Builder(),
			},
			fields: fields{
				existingEvents: []eventstore.Command{
					generateCommand(eventstore.AggregateType(t.Name()), "401"),
					generateCommand(eventstore.AggregateType(t.Name()), "401"),
					generateCommand(eventstore.AggregateType(t.Name()), "401"),
				},
			},
			wantErr: false,
		},
	}
	for _, tt := range tests {
		for querierName, querier := range queriers {
			t.Run(querierName+"/"+tt.name, func(t *testing.T) {
				t.Cleanup(cleanupEventstore(clients[querierName]))

				db := eventstore.NewEventstore(
					&eventstore.Config{
						Querier: querier,
						Pusher:  pushers["v3(inmemory)"],
					},
				)

				// setup initial data for query
				_, err := db.Push(context.Background(), tt.fields.existingEvents...)
				if err != nil {
					t.Errorf("error in setup = %v", err)
					return
				}

				sequence, err := db.LatestSequence(context.Background(), tt.args.searchQuery)
				if (err != nil) != tt.wantErr {
					t.Errorf("eventstore.query() error = %v, wantErr %v", err, tt.wantErr)
				}
				if tt.res.sequence > sequence {
					t.Errorf("eventstore.query() expected sequence: %v got %v", tt.res.sequence, sequence)
				}
			})
		}
	}
}