From f7b6ad901d7021046038f69e0d4d1055c41e9914 Mon Sep 17 00:00:00 2001 From: Nikola Jokic Date: Thu, 25 Jan 2024 15:45:07 +0100 Subject: [PATCH] Add listener graceful termination period and background context after the message is received (#3187) --- cmd/ghalistener/listener/listener_test.go | 2 ++ cmd/ghalistener/listener/metrics_test.go | 1 + 2 files changed, 3 insertions(+) diff --git a/cmd/ghalistener/listener/listener_test.go b/cmd/ghalistener/listener/listener_test.go index 9e0ac1dc..ff5208de 100644 --- a/cmd/ghalistener/listener/listener_test.go +++ b/cmd/ghalistener/listener/listener_test.go @@ -420,6 +420,7 @@ func TestListener_Listen(t *testing.T) { } client.On("CreateMessageSession", ctx, mock.Anything, mock.Anything).Return(session, nil).Once() client.On("DeleteMessageSession", mock.Anything, session.RunnerScaleSet.Id, session.SessionId).Return(nil).Once() + config.Client = client l, err := New(config) @@ -463,6 +464,7 @@ func TestListener_Listen(t *testing.T) { Statistics: &actions.RunnerScaleSetStatistic{}, } client.On("CreateMessageSession", ctx, mock.Anything, mock.Anything).Return(session, nil).Once() + client.On("DeleteMessageSession", mock.Anything, session.RunnerScaleSet.Id, session.SessionId).Return(nil).Once() msg := &actions.RunnerScaleSetMessage{ MessageId: 1, diff --git a/cmd/ghalistener/listener/metrics_test.go b/cmd/ghalistener/listener/metrics_test.go index 96ca6ac2..0f94c28d 100644 --- a/cmd/ghalistener/listener/metrics_test.go +++ b/cmd/ghalistener/listener/metrics_test.go @@ -82,6 +82,7 @@ func TestInitialMetrics(t *testing.T) { client := listenermocks.NewClient(t) client.On("CreateMessageSession", mock.Anything, mock.Anything, mock.Anything).Return(session, nil).Once() + client.On("DeleteMessageSession", mock.Anything, session.RunnerScaleSet.Id, session.SessionId).Return(nil).Once() config.Client = client handler := listenermocks.NewHandler(t)