From efc2fa7c6153306706b9bb48d4bcac5bf77a7d51 Mon Sep 17 00:00:00 2001 From: Sebastiaan van Stijn Date: Wed, 29 Jul 2026 13:19:43 +0200 Subject: [PATCH] *: don't manually call Events.Evict() in tests Events.Subscribe() returns a cancel func that can handle this. Signed-off-by: Sebastiaan van Stijn --- daemon/cluster/executor/container/health_test.go | 6 +----- daemon/events/events_test.go | 15 ++++++++------- daemon/events_test.go | 8 ++++---- daemon/health_test.go | 4 ++-- 4 files changed, 15 insertions(+), 18 deletions(-) diff --git a/daemon/cluster/executor/container/health_test.go b/daemon/cluster/executor/container/health_test.go index 6d0eb27d6c..415f545be9 100644 --- a/daemon/cluster/executor/container/health_test.go +++ b/daemon/cluster/executor/container/health_test.go @@ -17,10 +17,6 @@ import ( func TestHealthStates(t *testing.T) { // set up environment: events, task, container .... - e := events.New() - _, l, _ := e.Subscribe() - defer e.Evict(l) - task := &api.Task{ ID: "id", ServiceID: "sid", @@ -38,7 +34,7 @@ func TestHealthStates(t *testing.T) { } daemon := &daemon.Daemon{ - EventsService: e, + EventsService: events.New(), } ctrlr, err := newController(daemon, nil, nil, task, nil, nil) diff --git a/daemon/events/events_test.go b/daemon/events/events_test.go index bdf6e671a2..a3bbe87143 100644 --- a/daemon/events/events_test.go +++ b/daemon/events/events_test.go @@ -14,10 +14,10 @@ import ( func TestEventsLog(t *testing.T) { e := New() - _, l1, _ := e.Subscribe() - _, l2, _ := e.Subscribe() - defer e.Evict(l1) - defer e.Evict(l2) + _, l1, cancel1 := e.Subscribe() + defer cancel1() + _, l2, cancel2 := e.Subscribe() + defer cancel2() subscriberCount := e.SubscribersCount() assert.Check(t, is.Equal(subscriberCount, 2)) @@ -53,8 +53,8 @@ func TestEventsLog(t *testing.T) { func TestEventsLogTimeout(t *testing.T) { e := New() - _, l, _ := e.Subscribe() - defer e.Evict(l) + _, _, cancel := e.Subscribe() + defer cancel() c := make(chan struct{}) go func() { @@ -82,7 +82,8 @@ func TestLogEvents(t *testing.T) { }) } time.Sleep(50 * time.Millisecond) - current, l, _ := e.Subscribe() + current, l, cancel := e.Subscribe() + defer cancel() for i := range 10 { num := strconv.Itoa(i + eventsLimit + 16) e.Log(events.Action("action_"+num), events.ContainerEventType, events.Actor{ diff --git a/daemon/events_test.go b/daemon/events_test.go index 98205a1500..d8642f9854 100644 --- a/daemon/events_test.go +++ b/daemon/events_test.go @@ -14,8 +14,8 @@ import ( func TestLogContainerEventCopyLabels(t *testing.T) { e := events.New() - _, l, _ := e.Subscribe() - defer e.Evict(l) + _, l, cancel := e.Subscribe() + defer cancel() ctr := &container.Container{ ID: "container_id", @@ -45,8 +45,8 @@ func TestLogContainerEventCopyLabels(t *testing.T) { func TestLogContainerEventWithAttributes(t *testing.T) { e := events.New() - _, l, _ := e.Subscribe() - defer e.Evict(l) + _, l, cancel := e.Subscribe() + defer cancel() ctr := &container.Container{ ID: "container_id", diff --git a/daemon/health_test.go b/daemon/health_test.go index 52a29bd7ae..880b775e77 100644 --- a/daemon/health_test.go +++ b/daemon/health_test.go @@ -47,8 +47,8 @@ func TestNoneHealthcheck(t *testing.T) { // FIXME(vdemeester) This takes around 3s… This is *way* too long func TestHealthStates(t *testing.T) { e := events.New() - _, l, _ := e.Subscribe() - defer e.Evict(l) + _, l, cancel := e.Subscribe() + defer cancel() expect := func(expected eventtypes.Action) { select {