repository_test.go

package midgard_test

import (
	"fmt"
	"testing"

	"errors"

	"os"

	"github.com/it-chain/midgard"
	"github.com/it-chain/midgard/store"
	"github.com/it-chain/midgard/store/leveldb"
	"github.com/stretchr/testify/assert"
)

//aggregate
type UserAggregate struct {
	Name string
	midgard.EventModel
}

func (u *UserAggregate) On(event midgard.Event) error {

	switch v := event.(type) {

	case *UserCreated:
		u.ID = v.ID

	case *UserUpdated:
		u.Name = v.Name

	default:
		return errors.New(fmt.Sprintf("unhandled event [%s]", v))
	}

	return nil
}

//event
type UserCreated struct {
	midgard.EventModel
}

//event
type UserUpdated struct {
	Name string
	midgard.EventModel
}

func TestNewRepository(t *testing.T) {

	path := "test"
	defer os.RemoveAll(path)

	store := leveldb.NewEventStore(path, store.NewSerializer(UserCreated{}, UserUpdated{}))
	r := midgard.NewRepo(store, nil)

	aggregateID := "123"

	err := r.Save(aggregateID, UserCreated{
		EventModel: midgard.EventModel{
			ID:   aggregateID,
			Type: "User",
		},
	})

	assert.NoError(t, err)

	err = r.Save(aggregateID, UserUpdated{
		EventModel: midgard.EventModel{
			ID:   aggregateID,
			Type: "User",
		},
		Name: "jun",
	})

	assert.NoError(t, err)

	err = r.Save(aggregateID, UserUpdated{
		EventModel: midgard.EventModel{
			ID:   aggregateID,
			Type: "User",
		},
		Name: "jun2",
	})

	assert.NoError(t, err)

	user := &UserAggregate{}

	err = r.Load(user, aggregateID)
	assert.NoError(t, err)

	assert.Equal(t, user.ID, aggregateID)
	assert.Equal(t, user.Name, "jun2")

	fmt.Println(user)
}