Fix tests

mtglib/internal/dc/updater.go

@@ -2,38 +2,46 @@ package dc
 
 import (
 	"context"
+	"sync"
 	"time"
 )
 
 type updater struct {
+	wg     sync.WaitGroup
 	logger Logger
 	period time.Duration
 }
 
-func (u updater) run(ctx context.Context, callback func() error) {
-	ticker := time.NewTicker(u.period)
+func (u *updater) Wait() {
+	u.wg.Wait()
+}
 
-	defer func() {
-		ticker.Stop()
+func (u *updater) run(ctx context.Context, callback func() error) {
+	u.wg.Go(func() {
+		ticker := time.NewTicker(u.period)
 
-		select {
-		case <-ticker.C:
-		default:
-		}
-	}()
+		defer func() {
+			ticker.Stop()
+
+			select {
+			case <-ticker.C:
+			default:
+			}
+		}()
 
-	for {
-		u.logger.Info("start update")
-		if err := callback(); err != nil {
-			u.logger.WarningError("cannot update: %w", err)
-		}
-		u.logger.Info("updated")
+		for {
+			u.logger.Info("start update")
+			if err := callback(); err != nil {
+				u.logger.WarningError("cannot update: %w", err)
+			}
+			u.logger.Info("updated")
 
-		select {
-		case <-ctx.Done():
-			u.logger.Info("stop updating")
-			return
-		case <-ticker.C:
+			select {
+			case <-ctx.Done():
+				u.logger.Info("stop updating")
+				return
+			case <-ticker.C:
+			}
 		}
-	}
+	})
 }