lb: add DescribeLoadBalancer & CreateLoadBalancer

these functions should force `doWork` to run `updateStack` function depending on lb state

Signed-off-by: Mustafa Abdelrahman <mustafa.abdelrahman@zalando.de>

aws/fake/cf.go

@@ -91,6 +91,10 @@ func (m *CFClient) UpdateStack(params *cloudformation.UpdateStackInput) (*cloudf
 	// Update stack needs to use different variable to register change history,
 	// so createStack and updateStack mocks don't mess with each other states.
 
+	m.tagCreationHistory = append(m.tagCreationHistory, params.Tags)
+	m.paramCreationHistory = append(m.paramCreationHistory, params.Parameters)
+	m.templateCreationHistory = append(m.templateCreationHistory, *params.TemplateBody)
+
 	out, ok := m.Outputs.UpdateStack.response.(*cloudformation.UpdateStackOutput)
 	if !ok {
 		return nil, m.Outputs.UpdateStack.err

aws/fake/elbv2.go

@@ -13,13 +13,16 @@ type ELBv2Outputs struct {
 	DescribeTags         *APIResponse
 	DescribeTargetGroups *APIResponse
 	DescribeTargetHealth *APIResponse
+	CreateLoadBalancer   *APIResponse
+	DescribeLoadBalancer *APIResponse
 }
 
 type ELBv2Client struct {
 	elbv2iface.ELBV2API
-	Outputs  ELBv2Outputs
-	Rtinputs []*elbv2.RegisterTargetsInput
-	Dtinputs []*elbv2.DeregisterTargetsInput
+	Outputs   ELBv2Outputs
+	Rtinputs  []*elbv2.RegisterTargetsInput
+	Dtinputs  []*elbv2.DeregisterTargetsInput
+	LBinputes []*elbv2.CreateLoadBalancerInput
 }
 
 func (m *ELBv2Client) RegisterTargets(in *elbv2.RegisterTargetsInput) (*elbv2.RegisterTargetsOutput, error) {
@@ -67,6 +70,23 @@ func (m *ELBv2Client) DescribeTargetHealth(*elbv2.DescribeTargetHealthInput) (*e
 	return out, m.Outputs.DescribeTargetHealth.err
 }
 
+func (m *ELBv2Client) CreateLoadBalancer(in *elbv2.CreateLoadBalancerInput) (*elbv2.CreateLoadBalancerOutput, error) {
+	m.LBinputes = append(m.LBinputes, in)
+	out, ok := m.Outputs.CreateLoadBalancer.response.(*elbv2.CreateLoadBalancerOutput)
+	if !ok {
+		return nil, m.Outputs.CreateLoadBalancer.err
+	}
+	return out, m.Outputs.CreateLoadBalancer.err
+}
+
+func (m *ELBv2Client) DescribeLoadBalancers(in *elbv2.DescribeLoadBalancersInput) (*elbv2.DescribeLoadBalancersOutput, error) {
+	out, ok := m.Outputs.DescribeLoadBalancer.response.(*elbv2.DescribeLoadBalancersOutput)
+	if !ok {
+		return nil, m.Outputs.DescribeLoadBalancer.err
+	}
+	return out, m.Outputs.DescribeLoadBalancer.err
+}
+
 func MockDeregisterTargetsOutput() *elbv2.DeregisterTargetsOutput {
 	return &elbv2.DeregisterTargetsOutput{}
 }

worker_test.go

@@ -33,7 +33,14 @@ func TestResourceConversionOneToOne(tt *testing.T) {
 	clusterID := "aws:123:eu-central-1:kube-1"
 	vpcID := "1"
 	securityGroupID := "42"
-	running := int64(16) // See https://github.com/aws/aws-sdk-go/blob/master/service/ec2/api.go, type InstanceState
+	// See https://github.com/aws/aws-sdk-go/blob/master/service/ec2/api.go, type InstanceState
+	//    * 0 : pending
+	//    * 16 : running
+	//    * 32 : shutting-down
+	//    * 48 : terminated
+	//    * 64 : stopping
+	//    * 80 : stopped
+	running := int64(16)
 
 	for _, scenario := range []struct {
 		name           string