Skip to content

Commit 9481947

Browse files
committed
Adds already closed test case
Signed-off-by: Anshul Gupta <[email protected]>
1 parent c11da65 commit 9481947

File tree

1 file changed

+21
-0
lines changed

1 file changed

+21
-0
lines changed

internal/worker/aws/aws_test.go

+21
Original file line numberDiff line numberDiff line change
@@ -538,4 +538,25 @@ func TestAWSWorker_IsReady(t *testing.T) {
538538
m.For(t, "err").Assert(err, m.BeNil())
539539
m.For(t, "ready").Assert(ready, m.Equal(false))
540540
})
541+
542+
t.Run("already closed", func(t *testing.T) {
543+
mAWSClient := NewMockAWSWorkerEC2Client(ctrl)
544+
mAWSClient.EXPECT().
545+
TerminateInstances(gomock.Any(), gomock.Any(), gomock.Any()).
546+
Return(nil, nil)
547+
548+
worker := &AWSWorker{
549+
logger: logger,
550+
client: mAWSClient,
551+
id: "id",
552+
port: port,
553+
}
554+
555+
err := worker.Close()
556+
m.For(t, "close").For("err").Require(err, m.BeNil())
557+
558+
ready, err := worker.IsReady(context.Background())
559+
m.For(t, "err").Assert(err, m.Equal(compute.ErrClosed))
560+
m.For(t, "ready").Assert(ready, m.Equal(false))
561+
})
541562
}

0 commit comments

Comments
 (0)