diff options
Diffstat (limited to 'pkg/pool')
| -rwxr-xr-x | pkg/pool/static_pool_test.go | 7 | ||||
| -rw-r--r-- | pkg/pool/supervisor_test.go | 2 |
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) { |