96 lines
3.0 KiB
Go
96 lines
3.0 KiB
Go
package settingsstate
|
|
|
|
import (
|
|
"github.com/anytypeio/any-sync/commonspace/object/tree/objecttree"
|
|
"github.com/anytypeio/any-sync/commonspace/object/tree/objecttree/mock_objecttree"
|
|
"github.com/anytypeio/any-sync/commonspace/spacesyncproto"
|
|
"github.com/golang/mock/gomock"
|
|
"github.com/stretchr/testify/require"
|
|
"testing"
|
|
)
|
|
|
|
func TestStateBuilder_ProcessChange(t *testing.T) {
|
|
sb := &stateBuilder{}
|
|
rootId := "rootId"
|
|
deletedId := "deletedId"
|
|
|
|
t.Run("empty model", func(t *testing.T) {
|
|
ch := &objecttree.Change{}
|
|
newSt := sb.processChange(ch, rootId, &State{
|
|
DeletedIds: []string{deletedId},
|
|
})
|
|
require.Equal(t, []string{deletedId}, newSt.DeletedIds)
|
|
})
|
|
|
|
t.Run("changeId is equal to startId, LastIteratedId is equal to startId", func(t *testing.T) {
|
|
ch := &objecttree.Change{}
|
|
ch.Model = &spacesyncproto.SettingsData{
|
|
Content: []*spacesyncproto.SpaceSettingsContent{
|
|
{Value: &spacesyncproto.SpaceSettingsContent_ObjectDelete{
|
|
ObjectDelete: &spacesyncproto.ObjectDelete{Id: deletedId},
|
|
}},
|
|
},
|
|
}
|
|
ch.Id = "startId"
|
|
startId := "startId"
|
|
newSt := sb.processChange(ch, rootId, &State{
|
|
DeletedIds: []string{deletedId},
|
|
LastIteratedId: startId,
|
|
})
|
|
require.Equal(t, []string{deletedId}, newSt.DeletedIds)
|
|
})
|
|
|
|
t.Run("changeId is equal to rootId", func(t *testing.T) {
|
|
ch := &objecttree.Change{}
|
|
ch.PreviousIds = []string{"someId"}
|
|
ch.Model = &spacesyncproto.SettingsData{
|
|
Snapshot: &spacesyncproto.SpaceSettingsSnapshot{
|
|
DeletedIds: []string{"id1", "id2"},
|
|
DeleterPeerId: "peerId",
|
|
},
|
|
}
|
|
ch.Id = "rootId"
|
|
newSt := sb.processChange(ch, rootId, &State{})
|
|
require.Equal(t, []string{"id1", "id2"}, newSt.DeletedIds)
|
|
require.Equal(t, "peerId", newSt.DeleterId)
|
|
})
|
|
|
|
t.Run("changeId is not equal to lastIteratedId or rootId", func(t *testing.T) {
|
|
ch := &objecttree.Change{}
|
|
ch.PreviousIds = []string{"someId"}
|
|
ch.Model = &spacesyncproto.SettingsData{
|
|
Content: []*spacesyncproto.SpaceSettingsContent{
|
|
{Value: &spacesyncproto.SpaceSettingsContent_ObjectDelete{
|
|
ObjectDelete: &spacesyncproto.ObjectDelete{Id: deletedId},
|
|
}},
|
|
},
|
|
}
|
|
ch.Id = "someId"
|
|
newSt := sb.processChange(ch, rootId, &State{})
|
|
require.Equal(t, []string{deletedId}, newSt.DeletedIds)
|
|
})
|
|
}
|
|
|
|
func TestStateBuilder_Build(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
objTree := mock_objecttree.NewMockObjectTree(ctrl)
|
|
sb := &stateBuilder{}
|
|
defer ctrl.Finish()
|
|
|
|
t.Run("state is nil", func(t *testing.T) {
|
|
ch := &objecttree.Change{Id: "rootId"}
|
|
objTree.EXPECT().Root().Return(ch)
|
|
objTree.EXPECT().IterateFrom("rootId", gomock.Any(), gomock.Any()).Return(nil)
|
|
_, err := sb.Build(objTree, nil)
|
|
require.NoError(t, err)
|
|
})
|
|
|
|
t.Run("state is non-empty", func(t *testing.T) {
|
|
ch := &objecttree.Change{Id: "rootId"}
|
|
objTree.EXPECT().Root().Return(ch)
|
|
objTree.EXPECT().IterateFrom("someId", gomock.Any(), gomock.Any()).Return(nil)
|
|
_, err := sb.Build(objTree, &State{LastIteratedId: "someId"})
|
|
require.NoError(t, err)
|
|
})
|
|
}
|