func()

in pkg/cloudprovider/aws/fake/ec2api.go [77:136]
57 lines of code
11 McCabe index (conditional complexity)

func (e *EC2API) CreateFleetWithContext(_ context.Context, input *ec2.CreateFleetInput, _ ...request.Option) (*ec2.CreateFleetOutput, error) {
	e.CalledWithCreateFleetInput.Add(input)
	if input.LaunchTemplateConfigs[0].LaunchTemplateSpecification.LaunchTemplateName == nil {
		return nil, fmt.Errorf("missing launch template name")
	}
	instances := []*ec2.Instance{}
	instanceIds := []*string{}
	skippedPools := []CapacityPool{}
	var spotInstanceRequestID *string

	if aws.StringValue(input.TargetCapacitySpecification.DefaultTargetCapacityType) == v1alpha1.CapacityTypeSpot {
		spotInstanceRequestID = aws.String(randomdata.SillyName())
	}

	for i := 0; i < int(*input.TargetCapacitySpecification.TotalTargetCapacity); i++ {
		skipInstance := false
		for _, pool := range e.InsufficientCapacityPools {
			if pool.InstanceType == aws.StringValue(input.LaunchTemplateConfigs[0].Overrides[0].InstanceType) &&
				pool.Zone == aws.StringValue(input.LaunchTemplateConfigs[0].Overrides[0].AvailabilityZone) &&
				pool.CapacityType == aws.StringValue(input.TargetCapacitySpecification.DefaultTargetCapacityType) {
				skippedPools = append(skippedPools, pool)
				skipInstance = true
				break
			}
		}
		if skipInstance {
			continue
		}
		instances = append(instances, &ec2.Instance{
			InstanceId:            aws.String(randomdata.SillyName()),
			Placement:             &ec2.Placement{AvailabilityZone: input.LaunchTemplateConfigs[0].Overrides[0].AvailabilityZone},
			PrivateDnsName:        aws.String(randomdata.IpV4Address()),
			InstanceType:          input.LaunchTemplateConfigs[0].Overrides[0].InstanceType,
			SpotInstanceRequestId: spotInstanceRequestID,
		})
		e.Instances.Store(*instances[i].InstanceId, instances[i])
		instanceIds = append(instanceIds, instances[i].InstanceId)
	}

	result := &ec2.CreateFleetOutput{
		Instances: []*ec2.CreateFleetInstance{{InstanceIds: instanceIds}}}
	if len(skippedPools) > 0 {
		for _, pool := range skippedPools {
			result.Errors = append(result.Errors, &ec2.CreateFleetError{
				ErrorCode: aws.String("InsufficientInstanceCapacity"),
				LaunchTemplateAndOverrides: &ec2.LaunchTemplateAndOverridesResponse{
					LaunchTemplateSpecification: &ec2.FleetLaunchTemplateSpecification{
						LaunchTemplateId:   input.LaunchTemplateConfigs[0].LaunchTemplateSpecification.LaunchTemplateId,
						LaunchTemplateName: input.LaunchTemplateConfigs[0].LaunchTemplateSpecification.LaunchTemplateName,
					},
					Overrides: &ec2.FleetLaunchTemplateOverrides{
						InstanceType:     aws.String(pool.InstanceType),
						AvailabilityZone: aws.String(pool.Zone),
					},
				},
			})
		}
	}
	return result, nil
}