zitadel/internal/user/repository/eventsourcing/model/user_test.go

package model

import (
	"encoding/json"
	es_models "github.com/caos/zitadel/internal/eventstore/models"
	"github.com/caos/zitadel/internal/user/model"
	"testing"
	"time"
)

func TestAppendDeactivatedEvent(t *testing.T) {
	type args struct {
		user *User
	}
	tests := []struct {
		name   string
		args   args
		result *User
	}{
		{
			name: "append deactivate event",
			args: args{
				user: &User{},
			},
			result: &User{State: int32(model.USERSTATE_INACTIVE)},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.args.user.appendDeactivatedEvent()
			if tt.args.user.State != tt.result.State {
				t.Errorf("got wrong result: expected: %v, actual: %v ", tt.result, tt.args.user)
			}
		})
	}
}

func TestAppendReactivatedEvent(t *testing.T) {
	type args struct {
		user *User
	}
	tests := []struct {
		name   string
		args   args
		result *User
	}{
		{
			name: "append reactivate event",
			args: args{
				user: &User{},
			},
			result: &User{State: int32(model.USERSTATE_ACTIVE)},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.args.user.appendReactivatedEvent()
			if tt.args.user.State != tt.result.State {
				t.Errorf("got wrong result: expected: %v, actual: %v ", tt.result, tt.args.user)
			}
		})
	}
}

func TestAppendLockEvent(t *testing.T) {
	type args struct {
		user *User
	}
	tests := []struct {
		name   string
		args   args
		result *User
	}{
		{
			name: "append lock event",
			args: args{
				user: &User{},
			},
			result: &User{State: int32(model.USERSTATE_LOCKED)},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.args.user.appendLockedEvent()
			if tt.args.user.State != tt.result.State {
				t.Errorf("got wrong result: expected: %v, actual: %v ", tt.result, tt.args.user)
			}
		})
	}
}

func TestAppendUnlockEvent(t *testing.T) {
	type args struct {
		user *User
	}
	tests := []struct {
		name   string
		args   args
		result *User
	}{
		{
			name: "append unlock event",
			args: args{
				user: &User{},
			},
			result: &User{State: int32(model.USERSTATE_ACTIVE)},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			tt.args.user.appendUnlockedEvent()
			if tt.args.user.State != tt.result.State {
				t.Errorf("got wrong result: expected: %v, actual: %v ", tt.result, tt.args.user)
			}
		})
	}
}

func TestAppendInitUserCodeEvent(t *testing.T) {
	type args struct {
		user  *User
		code  *InitUserCode
		event *es_models.Event
	}
	tests := []struct {
		name   string
		args   args
		result *User
	}{
		{
			name: "append init user code event",
			args: args{
				user:  &User{},
				code:  &InitUserCode{Expiry: time.Hour * 30},
				event: &es_models.Event{},
			},
			result: &User{InitCode: &InitUserCode{Expiry: time.Hour * 30}},
		},
	}
	for _, tt := range tests {
		t.Run(tt.name, func(t *testing.T) {
			if tt.args.code != nil {
				data, _ := json.Marshal(tt.args.code)
				tt.args.event.Data = data
			}
			tt.args.user.appendInitUsercodeCreatedEvent(tt.args.event)
			if tt.args.user.InitCode.Expiry != tt.result.InitCode.Expiry {
				t.Errorf("got wrong result: expected: %v, actual: %v ", tt.result, tt.args.user)
			}
		})
	}
}