traefik/pkg/provider/ecs/builder_test.go

package ecs

import "github.com/aws/aws-sdk-go/service/ecs"

func instance(ops ...func(*ecsInstance)) ecsInstance {
	e := &ecsInstance{
		containerDefinition: &ecs.ContainerDefinition{},
	}

	for _, op := range ops {
		op(e)
	}

	return *e
}

func name(name string) func(*ecsInstance) {
	return func(e *ecsInstance) {
		e.Name = name
	}
}

func id(id string) func(*ecsInstance) {
	return func(e *ecsInstance) {
		e.ID = id
	}
}

func iMachine(opts ...func(*machine)) func(*ecsInstance) {
	return func(e *ecsInstance) {
		e.machine = &machine{}

		for _, opt := range opts {
			opt(e.machine)
		}
	}
}

func mState(state string) func(*machine) {
	return func(m *machine) {
		m.state = state
	}
}

func mPrivateIP(ip string) func(*machine) {
	return func(m *machine) {
		m.privateIP = ip
	}
}

func mHealthStatus(status string) func(*machine) {
	return func(m *machine) {
		m.healthStatus = status
	}
}

func mPorts(opts ...func(*portMapping)) func(*machine) {
	return func(m *machine) {
		for _, opt := range opts {
			p := &portMapping{}
			opt(p)
			m.ports = append(m.ports, *p)
		}
	}
}

func mPort(containerPort, hostPort int32, protocol string) func(*portMapping) {
	return func(pm *portMapping) {
		pm.containerPort = int64(containerPort)
		pm.hostPort = int64(hostPort)
		pm.protocol = protocol
	}
}

func labels(labels map[string]string) func(*ecsInstance) {
	return func(c *ecsInstance) {
		c.Labels = labels
	}
}