watchtower/internal/actions/mocks/container.go

package mocks

import (
	"github.com/containrrr/watchtower/pkg/container"
	"github.com/docker/docker/api/types"
	container2 "github.com/docker/docker/api/types/container"
	"time"
)

// CreateMockContainer creates a container substitute valid for testing
func CreateMockContainer(id string, name string, image string, created time.Time, depends []string) container.Container {
	content := types.ContainerJSON{
		ContainerJSONBase: &types.ContainerJSONBase{
			ID:      id,
			Image:   image,
			Name:    name,
			Created: created.String(),
		},
		Config: &container2.Config{
			Labels: make(map[string]string),
		},
	}
	dependencyString := ""
	for ind, i := range depends {
		if ind == 0 {
			dependencyString += i;
		}else{
			dependencyString += "," + i;
		}
	}
	content.Config.Labels["com.centurylinklabs.watchtower.depends-on"] = dependencyString
	return *container.NewContainer(
		&content,
		&types.ImageInspect{
			ID: image,
		},
	)
}

// CreateMockContainerWithConfig creates a container substitute valid for testing
func CreateMockContainerWithConfig(id string, name string, image string, created time.Time, config *container2.Config) container.Container {
	content := types.ContainerJSON{
		ContainerJSONBase: &types.ContainerJSONBase{
			ID:      id,
			Image:   image,
			Name:    name,
			Created: created.String(),
		},
		Config: config,
	}
	return *container.NewContainer(
		&content,
		&types.ImageInspect{
			ID: image,
		},
	)
}