Added global ticker and timeout

Signed-off-by: Effi-S <effi.szt@gmail.com>

Author: Effi-S

token/services/storage/db/dbtest/notifier.go

@@ -60,13 +60,21 @@ func (c *dbEventsCollector[T]) AssertSize(size int) error {
 		c.close <- true
 	}()
 
+	// The overall timeout must be computed once: allocating time.After inside
+	// the loop would reset it on every poll tick, so it could never elapse and
+	// AssertSize would spin forever when the expected size is never reached.
+	timeout := time.NewTimer(time.Second)
+	defer timeout.Stop()
+	ticker := time.NewTicker(20 * time.Millisecond)
+	defer ticker.Stop()
+
 	for {
 		select {
 		case <-c.close:
 			return errors.Errorf("db events collector closed")
-		case <-time.After(time.Second):
+		case <-timeout.C:
 			return errors.Errorf("db events collector timeout")
-		case <-time.After(20 * time.Millisecond):
+		case <-ticker.C:
 			c.mu.RLock()
 			resultSize := len(c.result)
 			c.mu.RUnlock()