add service

internal/worker/invite_cleaner.go

@@ -5,7 +5,7 @@ import (
 	"log"
 	"time"
 
-	"smart-search-back/internal/repository"
+	"git.techease.ru/Smart-search/smart-search-back/internal/repository"
 )
 
 type InviteCleaner struct {

internal/worker/session_cleaner.go

@@ -5,7 +5,7 @@ import (
 	"log"
 	"time"
 
-	"smart-search-back/internal/repository"
+	"git.techease.ru/Smart-search/smart-search-back/internal/repository"
 )
 
 type SessionCleaner struct {

internal/worker/worker_test.go

@@ -0,0 +1,213 @@
+package worker
+
+import (
+	"context"
+	"sync/atomic"
+	"testing"
+	"time"
+
+	"github.com/gojuno/minimock/v3"
+	"github.com/stretchr/testify/suite"
+
+	"git.techease.ru/Smart-search/smart-search-back/internal/mocks"
+)
+
+type WorkerSuite struct {
+	suite.Suite
+	ctx    context.Context
+	cancel context.CancelFunc
+	ctrl   *minimock.Controller
+}
+
+func TestWorkerSuite(t *testing.T) {
+	suite.Run(t, new(WorkerSuite))
+}
+
+func (s *WorkerSuite) SetupTest() {
+	s.ctx, s.cancel = context.WithCancel(context.Background())
+	s.ctrl = minimock.NewController(s.T())
+}
+
+func (s *WorkerSuite) TearDownTest() {
+	if s.cancel != nil {
+		s.cancel()
+	}
+}
+
+func (s *WorkerSuite) TestSessionCleaner_StartStop() {
+	var callCount int32
+	sessionRepo := mocks.NewSessionRepositoryMock(s.ctrl)
+	sessionRepo.DeleteExpiredMock.Set(func(_ context.Context) (int, error) {
+		atomic.AddInt32(&callCount, 1)
+		return 5, nil
+	})
+
+	cleaner := NewSessionCleaner(s.ctx, sessionRepo)
+
+	cleaner.Start()
+
+	time.Sleep(50 * time.Millisecond)
+
+	cleaner.Stop()
+
+	s.GreaterOrEqual(int(atomic.LoadInt32(&callCount)), 1,
+		"DeleteExpired должен быть вызван хотя бы раз при старте")
+}
+
+func (s *WorkerSuite) TestSessionCleaner_ContextCancellation() {
+	var callCount int32
+	sessionRepo := mocks.NewSessionRepositoryMock(s.ctrl)
+	sessionRepo.DeleteExpiredMock.Set(func(_ context.Context) (int, error) {
+		atomic.AddInt32(&callCount, 1)
+		return 0, nil
+	})
+
+	cleaner := NewSessionCleaner(s.ctx, sessionRepo)
+
+	cleaner.Start()
+
+	time.Sleep(50 * time.Millisecond)
+
+	s.cancel()
+
+	time.Sleep(50 * time.Millisecond)
+
+	s.GreaterOrEqual(int(atomic.LoadInt32(&callCount)), 1,
+		"DeleteExpired должен быть вызван хотя бы раз")
+}
+
+func (s *WorkerSuite) TestInviteCleaner_StartStop() {
+	var callCount int32
+	inviteRepo := mocks.NewInviteRepositoryMock(s.ctrl)
+	inviteRepo.DeactivateExpiredMock.Set(func(_ context.Context) (int, error) {
+		atomic.AddInt32(&callCount, 1)
+		return 3, nil
+	})
+
+	cleaner := NewInviteCleaner(s.ctx, inviteRepo)
+
+	cleaner.Start()
+
+	time.Sleep(50 * time.Millisecond)
+
+	cleaner.Stop()
+
+	s.GreaterOrEqual(int(atomic.LoadInt32(&callCount)), 1,
+		"DeactivateExpired должен быть вызван хотя бы раз при старте")
+}
+
+func (s *WorkerSuite) TestInviteCleaner_ContextCancellation() {
+	var callCount int32
+	inviteRepo := mocks.NewInviteRepositoryMock(s.ctrl)
+	inviteRepo.DeactivateExpiredMock.Set(func(_ context.Context) (int, error) {
+		atomic.AddInt32(&callCount, 1)
+		return 0, nil
+	})
+
+	cleaner := NewInviteCleaner(s.ctx, inviteRepo)
+
+	cleaner.Start()
+
+	time.Sleep(50 * time.Millisecond)
+
+	s.cancel()
+
+	time.Sleep(50 * time.Millisecond)
+
+	s.GreaterOrEqual(int(atomic.LoadInt32(&callCount)), 1,
+		"DeactivateExpired должен быть вызван хотя бы раз")
+}
+
+func (s *WorkerSuite) TestSessionCleaner_ConcurrentStops() {
+	sessionRepo := mocks.NewSessionRepositoryMock(s.ctrl)
+	sessionRepo.DeleteExpiredMock.Set(func(_ context.Context) (int, error) {
+		return 0, nil
+	})
+
+	cleaner := NewSessionCleaner(s.ctx, sessionRepo)
+
+	cleaner.Start()
+
+	time.Sleep(50 * time.Millisecond)
+
+	done := make(chan struct{})
+	for i := 0; i < 5; i++ {
+		go func() {
+			cleaner.Stop()
+		}()
+	}
+
+	go func() {
+		time.Sleep(100 * time.Millisecond)
+		close(done)
+	}()
+
+	<-done
+}
+
+func (s *WorkerSuite) TestInviteCleaner_ConcurrentStops() {
+	inviteRepo := mocks.NewInviteRepositoryMock(s.ctrl)
+	inviteRepo.DeactivateExpiredMock.Set(func(_ context.Context) (int, error) {
+		return 0, nil
+	})
+
+	cleaner := NewInviteCleaner(s.ctx, inviteRepo)
+
+	cleaner.Start()
+
+	time.Sleep(50 * time.Millisecond)
+
+	done := make(chan struct{})
+	for i := 0; i < 5; i++ {
+		go func() {
+			cleaner.Stop()
+		}()
+	}
+
+	go func() {
+		time.Sleep(100 * time.Millisecond)
+		close(done)
+	}()
+
+	<-done
+}
+
+func (s *WorkerSuite) TestSessionCleaner_MultipleStartStop() {
+	var callCount int32
+	sessionRepo := mocks.NewSessionRepositoryMock(s.ctrl)
+	sessionRepo.DeleteExpiredMock.Set(func(_ context.Context) (int, error) {
+		atomic.AddInt32(&callCount, 1)
+		return 2, nil
+	})
+
+	cleaner := NewSessionCleaner(s.ctx, sessionRepo)
+
+	for i := 0; i < 3; i++ {
+		cleaner.Start()
+		time.Sleep(20 * time.Millisecond)
+		cleaner.Stop()
+	}
+
+	s.GreaterOrEqual(int(atomic.LoadInt32(&callCount)), 3,
+		"DeleteExpired должен быть вызван минимум 3 раза")
+}
+
+func (s *WorkerSuite) TestInviteCleaner_MultipleStartStop() {
+	var callCount int32
+	inviteRepo := mocks.NewInviteRepositoryMock(s.ctrl)
+	inviteRepo.DeactivateExpiredMock.Set(func(_ context.Context) (int, error) {
+		atomic.AddInt32(&callCount, 1)
+		return 1, nil
+	})
+
+	cleaner := NewInviteCleaner(s.ctx, inviteRepo)
+
+	for i := 0; i < 3; i++ {
+		cleaner.Start()
+		time.Sleep(20 * time.Millisecond)
+		cleaner.Stop()
+	}
+
+	s.GreaterOrEqual(int(atomic.LoadInt32(&callCount)), 3,
+		"DeactivateExpired должен быть вызван минимум 3 раза")
+}