Курнат Андрей
64 lines
2.2 KiB
Go
64 lines
2.2 KiB
Go
package event_test
|
|
|
|
import (
|
|
"context"
|
|
"sync/atomic"
|
|
|
|
userv1beta1 "github.com/cs3org/go-cs3apis/cs3/identity/user/v1beta1"
|
|
. "github.com/onsi/ginkgo/v2"
|
|
. "github.com/onsi/gomega"
|
|
"github.com/qsfera/server/pkg/log"
|
|
searchMocks "github.com/qsfera/server/services/search/pkg/search/mocks"
|
|
"github.com/qsfera/server/services/search/pkg/service/event"
|
|
"github.com/opencloud-eu/reva/v2/pkg/events"
|
|
"github.com/opencloud-eu/reva/v2/pkg/events/raw"
|
|
rawMocks "github.com/opencloud-eu/reva/v2/pkg/events/raw/mocks"
|
|
"github.com/stretchr/testify/mock"
|
|
)
|
|
|
|
var _ = DescribeTable("event",
|
|
func(mcks []string, e any, asyncUploads bool) {
|
|
var (
|
|
s = &searchMocks.Searcher{}
|
|
calls atomic.Int32
|
|
)
|
|
|
|
stream := rawMocks.NewStream(GinkgoT())
|
|
ch := make(chan raw.Event, 1)
|
|
stream.EXPECT().Consume(mock.Anything, mock.Anything).Return((<-chan raw.Event)(ch), nil)
|
|
|
|
event, err := event.New(context.Background(), stream, log.NewLogger(), nil, nil, s, 50, 1, asyncUploads)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
go func() {
|
|
err := event.Run()
|
|
Expect(err).NotTo(HaveOccurred())
|
|
}()
|
|
defer event.Close()
|
|
|
|
for _, mck := range mcks {
|
|
s.On(mck, mock.Anything, mock.Anything).Return(nil).Run(func(args mock.Arguments) {
|
|
calls.Add(1)
|
|
})
|
|
}
|
|
|
|
ch <- raw.Event{
|
|
Event: events.Event{Event: e},
|
|
}
|
|
|
|
Eventually(func() int {
|
|
return int(calls.Load())
|
|
}, "2s").Should(Equal(len(mcks)))
|
|
},
|
|
Entry("ItemTrashed", []string{"TrashItem", "IndexSpace"}, events.ItemTrashed{}, false),
|
|
Entry("ItemMoved", []string{"MoveItem", "IndexSpace"}, events.ItemMoved{}, false),
|
|
Entry("ItemRestored", []string{"RestoreItem", "IndexSpace"}, events.ItemRestored{}, false),
|
|
Entry("ContainerCreated", []string{"IndexSpace"}, events.ContainerCreated{}, false),
|
|
Entry("FileTouched", []string{"IndexSpace"}, events.FileTouched{}, false),
|
|
Entry("FileVersionRestored", []string{"IndexSpace"}, events.FileVersionRestored{}, false),
|
|
Entry("TagsAdded", []string{"UpsertItem", "IndexSpace"}, events.TagsAdded{}, false),
|
|
Entry("TagsRemoved", []string{"UpsertItem", "IndexSpace"}, events.TagsRemoved{}, false),
|
|
Entry("FileUploaded", []string{"IndexSpace"}, events.FileUploaded{}, false),
|
|
Entry("UploadReady", []string{"IndexSpace"}, events.UploadReady{ExecutingUser: &userv1beta1.User{}}, true),
|
|
)
|