From 1d4c162fba12aa73b4724ee6d0b696af724932d3 Mon Sep 17 00:00:00 2001 From: Amir Raminfar Date: Fri, 2 Feb 2024 16:48:58 -0800 Subject: [PATCH] Revert "chore: updates tests" This reverts commit 19a83b7b621845835d3ef2bbd8ea075d123a197d. --- internal/web/events_test.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/internal/web/events_test.go b/internal/web/events_test.go index 9bd71fed..aba5be55 100644 --- a/internal/web/events_test.go +++ b/internal/web/events_test.go @@ -2,6 +2,7 @@ package web import ( "context" + "time" "net/http" "net/http/httptest" @@ -22,7 +23,7 @@ func Test_handler_streamEvents_happy(t *testing.T) { mockedClient := new(MockedClient) errChannel := make(chan error) - mockedClient.On("ListContainers").Return([]docker.Container{}, nil).Twice() + mockedClient.On("ListContainers").Return([]docker.Container{}, nil) mockedClient.On("Events", mock.Anything, mock.AnythingOfType("chan<- docker.ContainerEvent")).Return(errChannel).Run(func(args mock.Arguments) { messages := args.Get(1).(chan<- docker.ContainerEvent) go func() { @@ -36,6 +37,7 @@ func Test_handler_streamEvents_happy(t *testing.T) { ActorID: "1234", Host: "localhost", } + time.Sleep(100 * time.Millisecond) cancel() }() }) @@ -44,7 +46,7 @@ func Test_handler_streamEvents_happy(t *testing.T) { Name: "test", Image: "test", Stats: utils.NewRingBuffer[docker.ContainerStat](300), // 300 seconds of stats - }, nil).Once() + }, nil) clients := map[string]docker.Client{ "localhost": mockedClient,