mirror of https://github.com/docker/compose.git
50 lines
1.3 KiB
Go
50 lines
1.3 KiB
Go
package amazon
|
|
|
|
import (
|
|
"context"
|
|
"testing"
|
|
|
|
"github.com/docker/ecs-plugin/pkg/amazon/mock"
|
|
"github.com/golang/mock/gomock"
|
|
)
|
|
|
|
func TestDownDontDeleteCluster(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
m := mock.NewMockAPI(ctrl)
|
|
c := &client{
|
|
Cluster: "test_cluster",
|
|
Region: "region",
|
|
api: m,
|
|
}
|
|
ctx := context.TODO()
|
|
recorder := m.EXPECT()
|
|
recorder.DeleteStack(ctx, "test_project").Return(nil)
|
|
recorder.GetStackID(ctx, "test_project").Return("stack-123", nil)
|
|
recorder.WaitStackComplete(ctx, "stack-123", StackDelete).Return(nil)
|
|
recorder.DescribeStackEvents(ctx, "stack-123").Return(nil, nil)
|
|
|
|
c.ComposeDown(ctx, "test_project", false)
|
|
}
|
|
|
|
func TestDownDeleteCluster(t *testing.T) {
|
|
ctrl := gomock.NewController(t)
|
|
defer ctrl.Finish()
|
|
m := mock.NewMockAPI(ctrl)
|
|
c := &client{
|
|
Cluster: "test_cluster",
|
|
Region: "region",
|
|
api: m,
|
|
}
|
|
|
|
ctx := context.TODO()
|
|
recorder := m.EXPECT()
|
|
recorder.DeleteStack(ctx, "test_project").Return(nil)
|
|
recorder.GetStackID(ctx, "test_project").Return("stack-123", nil)
|
|
recorder.WaitStackComplete(ctx, "stack-123", StackDelete).Return(nil)
|
|
recorder.DescribeStackEvents(ctx, "stack-123").Return(nil, nil)
|
|
recorder.DeleteCluster(ctx, "test_cluster").Return(nil)
|
|
|
|
c.ComposeDown(ctx, "test_project", true)
|
|
}
|