123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215 |
- package container
- import (
- "fmt"
- "strings"
- containertypes "github.com/docker/docker/api/types/container"
- mounttypes "github.com/docker/docker/api/types/mount"
- networktypes "github.com/docker/docker/api/types/network"
- "github.com/docker/docker/api/types/strslice"
- "github.com/docker/go-connections/nat"
- specs "github.com/opencontainers/image-spec/specs-go/v1"
- )
- // WithName sets the name of the container
- func WithName(name string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.Name = name
- }
- }
- // WithLinks sets the links of the container
- func WithLinks(links ...string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.HostConfig.Links = links
- }
- }
- // WithImage sets the image of the container
- func WithImage(image string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.Config.Image = image
- }
- }
- // WithCmd sets the comannds of the container
- func WithCmd(cmds ...string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.Config.Cmd = strslice.StrSlice(cmds)
- }
- }
- // WithNetworkMode sets the network mode of the container
- func WithNetworkMode(mode string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.HostConfig.NetworkMode = containertypes.NetworkMode(mode)
- }
- }
- // WithExposedPorts sets the exposed ports of the container
- func WithExposedPorts(ports ...string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.Config.ExposedPorts = map[nat.Port]struct{}{}
- for _, port := range ports {
- c.Config.ExposedPorts[nat.Port(port)] = struct{}{}
- }
- }
- }
- // WithTty sets the TTY mode of the container
- func WithTty(tty bool) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.Config.Tty = tty
- }
- }
- // WithWorkingDir sets the working dir of the container
- func WithWorkingDir(dir string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.Config.WorkingDir = dir
- }
- }
- // WithMount adds an mount
- func WithMount(m mounttypes.Mount) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.HostConfig.Mounts = append(c.HostConfig.Mounts, m)
- }
- }
- // WithVolume sets the volume of the container
- func WithVolume(target string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- if c.Config.Volumes == nil {
- c.Config.Volumes = map[string]struct{}{}
- }
- c.Config.Volumes[target] = struct{}{}
- }
- }
- // WithBind sets the bind mount of the container
- func WithBind(src, target string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.HostConfig.Binds = append(c.HostConfig.Binds, fmt.Sprintf("%s:%s", src, target))
- }
- }
- // WithTmpfs sets a target path in the container to a tmpfs
- func WithTmpfs(target string) func(config *TestContainerConfig) {
- return func(c *TestContainerConfig) {
- if c.HostConfig.Tmpfs == nil {
- c.HostConfig.Tmpfs = make(map[string]string)
- }
- spec := strings.SplitN(target, ":", 2)
- var opts string
- if len(spec) > 1 {
- opts = spec[1]
- }
- c.HostConfig.Tmpfs[spec[0]] = opts
- }
- }
- // WithIPv4 sets the specified ip for the specified network of the container
- func WithIPv4(network, ip string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- if c.NetworkingConfig.EndpointsConfig == nil {
- c.NetworkingConfig.EndpointsConfig = map[string]*networktypes.EndpointSettings{}
- }
- if v, ok := c.NetworkingConfig.EndpointsConfig[network]; !ok || v == nil {
- c.NetworkingConfig.EndpointsConfig[network] = &networktypes.EndpointSettings{}
- }
- if c.NetworkingConfig.EndpointsConfig[network].IPAMConfig == nil {
- c.NetworkingConfig.EndpointsConfig[network].IPAMConfig = &networktypes.EndpointIPAMConfig{}
- }
- c.NetworkingConfig.EndpointsConfig[network].IPAMConfig.IPv4Address = ip
- }
- }
- // WithIPv6 sets the specified ip6 for the specified network of the container
- func WithIPv6(network, ip string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- if c.NetworkingConfig.EndpointsConfig == nil {
- c.NetworkingConfig.EndpointsConfig = map[string]*networktypes.EndpointSettings{}
- }
- if v, ok := c.NetworkingConfig.EndpointsConfig[network]; !ok || v == nil {
- c.NetworkingConfig.EndpointsConfig[network] = &networktypes.EndpointSettings{}
- }
- if c.NetworkingConfig.EndpointsConfig[network].IPAMConfig == nil {
- c.NetworkingConfig.EndpointsConfig[network].IPAMConfig = &networktypes.EndpointIPAMConfig{}
- }
- c.NetworkingConfig.EndpointsConfig[network].IPAMConfig.IPv6Address = ip
- }
- }
- // WithLogDriver sets the log driver to use for the container
- func WithLogDriver(driver string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.HostConfig.LogConfig.Type = driver
- }
- }
- // WithAutoRemove sets the container to be removed on exit
- func WithAutoRemove(c *TestContainerConfig) {
- c.HostConfig.AutoRemove = true
- }
- // WithPidsLimit sets the container's "pids-limit
- func WithPidsLimit(limit *int64) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- if c.HostConfig == nil {
- c.HostConfig = &containertypes.HostConfig{}
- }
- c.HostConfig.PidsLimit = limit
- }
- }
- // WithRestartPolicy sets container's restart policy
- func WithRestartPolicy(policy string) func(c *TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.HostConfig.RestartPolicy.Name = policy
- }
- }
- // WithUser sets the user
- func WithUser(user string) func(c *TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.Config.User = user
- }
- }
- // WithPrivileged sets privileged mode for the container
- func WithPrivileged(privileged bool) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- if c.HostConfig == nil {
- c.HostConfig = &containertypes.HostConfig{}
- }
- c.HostConfig.Privileged = privileged
- }
- }
- // WithCgroupnsMode sets the cgroup namespace mode for the container
- func WithCgroupnsMode(mode string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- if c.HostConfig == nil {
- c.HostConfig = &containertypes.HostConfig{}
- }
- c.HostConfig.CgroupnsMode = containertypes.CgroupnsMode(mode)
- }
- }
- // WithExtraHost sets the user defined IP:Host mappings in the container's
- // /etc/hosts file
- func WithExtraHost(extraHost string) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.HostConfig.ExtraHosts = append(c.HostConfig.ExtraHosts, extraHost)
- }
- }
- // WithPlatform specifies the desired platform the image should have.
- func WithPlatform(p *specs.Platform) func(*TestContainerConfig) {
- return func(c *TestContainerConfig) {
- c.Platform = p
- }
- }
|