events: make "WithClock" methods allowed without "unit" build tag (#62)

There are use cases where having those methods available even outside of a unit test is helpful, such as when the objects are instantiated with a clock that could be mocked in the unit test for the parent method.

Signed-off-by: ItalyPaleAle <[email protected]>

events/batcher/batcher.go

@@ -52,6 +52,11 @@ func New[T key](interval time.Duration) *Batcher[T] {
 	}
 }
 
+// WithClock sets the clock used by the batcher. Used for testing.
+func (b *Batcher[T]) WithClock(clock clock.WithDelayedExecution) {
+	b.clock = clock
+}
+
 // Subscribe adds a new event channel subscriber. If the batcher is closed, the
 // subscriber is silently dropped.
 func (b *Batcher[T]) Subscribe(eventCh ...chan<- struct{}) {

events/batcher/batcher_test.go

@@ -30,6 +30,13 @@ func TestNew(t *testing.T) {
 	assert.False(t, b.closed.Load())
 }
 
+func TestWithClock(t *testing.T) {
+	b := New[string](time.Millisecond * 10)
+	fakeClock := testingclock.NewFakeClock(time.Now())
+	b.WithClock(fakeClock)
+	assert.Equal(t, fakeClock, b.clock)
+}
+
 func TestSubscribe(t *testing.T) {
 	t.Parallel()
 

events/queue/processor.go

@@ -51,6 +51,12 @@ func NewProcessor[T queueable](executeFn func(r T)) *Processor[T] {
 	}
 }
 
+// WithClock sets the clock used by the processor. Used for testing.
+func (p *Processor[T]) WithClock(clock kclock.Clock) *Processor[T] {
+	p.clock = clock
+	return p
+}
+
 // Enqueue adds a new item to the queue.
 // If a item with the same ID already exists, it'll be replaced.
 func (p *Processor[T]) Enqueue(r T) error {