Update CI

Signed-off-by: Valery Piashchynski <[email protected]>

2 files changed, 4 insertions, 5 deletions

diff --git a/pkg/pool/static_pool_test.go b/pkg/pool/static_pool_test.go
index 53d6b191..f66895dc 100755
--- a/pkg/pool/static_pool_test.go
+++ b/pkg/pool/static_pool_test.go
@@ -459,16 +459,15 @@ func Test_Static_Pool_Handle_Dead(t *testing.T) {
 		},
 	)
 	assert.NoError(t, err)
-	defer p.Destroy(ctx)
-
 	assert.NotNil(t, p)
 
-	for _, w := range p.Workers() {
-		w.State().Set(internal.StateErrored)
+	for i := range p.Workers() {
+		p.Workers()[i].State().Set(internal.StateErrored)
 	}
 
 	_, err = p.Exec(payload.Payload{Body: []byte("hello")})
 	assert.Error(t, err)
+	p.Destroy(ctx)
 }
 
 // identical to replace but controlled on worker side
diff --git a/pkg/pool/supervisor_test.go b/pkg/pool/supervisor_test.go
index 72226bee..b3358965 100644
--- a/pkg/pool/supervisor_test.go
+++ b/pkg/pool/supervisor_test.go
@@ -135,7 +135,6 @@ func TestSupervisedPool_Idle(t *testing.T) {
 
 	assert.NoError(t, err)
 	assert.NotNil(t, p)
-	defer p.Destroy(context.Background())
 
 	pid := p.Workers()[0].Pid()
 
@@ -151,6 +150,7 @@ func TestSupervisedPool_Idle(t *testing.T) {
 	time.Sleep(time.Second * 5)
 	// should be new worker with new pid
 	assert.NotEqual(t, pid, p.Workers()[0].Pid())
+	p.Destroy(context.Background())
 }
 
 func TestSupervisedPool_ExecTTL_OK(t *testing.T) {