package headsync

import (
	"github.com/anyproto/any-sync/app/ldiff"
	"github.com/anyproto/any-sync/app/ldiff/mock_ldiff"
	"github.com/anyproto/any-sync/app/logger"
	"github.com/anyproto/any-sync/commonspace/headsync/mock_headsync"
	"github.com/anyproto/any-sync/commonspace/object/tree/treestorage/mock_treestorage"
	"github.com/anyproto/any-sync/commonspace/settings/settingsstate/mock_settingsstate"
	"github.com/anyproto/any-sync/commonspace/spacestorage/mock_spacestorage"
	"github.com/anyproto/any-sync/util/periodicsync/mock_periodicsync"
	"github.com/golang/mock/gomock"
	"testing"
)

func TestDiffService(t *testing.T) {
	ctrl := gomock.NewController(t)
	defer ctrl.Finish()

	spaceId := "spaceId"
	l := logger.NewNamed("sync")
	pSyncMock := mock_periodicsync.NewMockPeriodicSync(ctrl)
	storageMock := mock_spacestorage.NewMockSpaceStorage(ctrl)
	treeStorageMock := mock_treestorage.NewMockTreeStorage(ctrl)
	diffMock := mock_ldiff.NewMockDiff(ctrl)
	syncer := mock_headsync.NewMockDiffSyncer(ctrl)
	delState := mock_settingsstate.NewMockObjectDeletionState(ctrl)
	syncPeriod := 1
	initId := "initId"

	service := &headSync{
		spaceId:      spaceId,
		storage:      storageMock,
		periodicSync: pSyncMock,
		syncer:       syncer,
		diff:         diffMock,
		log:          l,
		syncPeriod:   syncPeriod,
	}

	t.Run("init", func(t *testing.T) {
		storageMock.EXPECT().TreeStorage(initId).Return(treeStorageMock, nil)
		treeStorageMock.EXPECT().Heads().Return([]string{"h1", "h2"}, nil)
		syncer.EXPECT().Init(delState)
		diffMock.EXPECT().Set(ldiff.Element{
			Id:   initId,
			Head: "h1h2",
		})
		hash := "123"
		diffMock.EXPECT().Hash().Return(hash)
		storageMock.EXPECT().WriteSpaceHash(hash)
		pSyncMock.EXPECT().Run()
		service.Init([]string{initId}, delState)
	})

	t.Run("update heads", func(t *testing.T) {
		syncer.EXPECT().UpdateHeads(initId, []string{"h1", "h2"})
		service.UpdateHeads(initId, []string{"h1", "h2"})
	})

	t.Run("remove objects", func(t *testing.T) {
		syncer.EXPECT().RemoveObjects([]string{"h1", "h2"})
		service.RemoveObjects([]string{"h1", "h2"})
	})

	t.Run("close", func(t *testing.T) {
		pSyncMock.EXPECT().Close()
		service.Close()
	})
}