#387 fix: switch to image id map and add additional tests

internal/actions/mocks/client.go

@@ -0,0 +1,72 @@
+package mocks
+
+import (
+	"errors"
+	"github.com/containrrr/watchtower/pkg/container"
+	"time"
+
+	t "github.com/containrrr/watchtower/pkg/types"
+	cli "github.com/docker/docker/client"
+)
+
+type MockClient struct {
+	TestData      *TestData
+	api           cli.CommonAPIClient
+	pullImages    bool
+	removeVolumes bool
+}
+
+type TestData struct {
+	TriedToRemoveImageCount int
+	NameOfContainerToKeep string
+	Containers            []container.Container
+}
+
+func (testdata *TestData) TriedToRemoveImage() bool {
+	return testdata.TriedToRemoveImageCount > 0
+}
+
+func CreateMockClient(data *TestData, api cli.CommonAPIClient, pullImages bool, removeVolumes bool) MockClient {
+	return MockClient {
+		data,
+		api,
+		pullImages,
+		removeVolumes,
+	}
+}
+
+func (client MockClient) ListContainers(f t.Filter) ([]container.Container, error) {
+	return client.TestData.Containers, nil
+}
+
+func (client MockClient) StopContainer(c container.Container, d time.Duration) error {
+	if c.Name() == client.TestData.NameOfContainerToKeep {
+		return errors.New("tried to stop the instance we want to keep")
+	}
+	return nil
+}
+func (client MockClient) StartContainer(c container.Container) (string, error) {
+	return "", nil
+}
+
+func (client MockClient) RenameContainer(c container.Container, s string) error {
+	return nil
+}
+
+func (client MockClient) RemoveImageByID(id string) error {
+	client.TestData.TriedToRemoveImageCount++
+	return nil
+}
+
+func (client MockClient) GetContainer(containerID string) (container.Container, error) {
+	return container.Container{}, nil
+}
+
+func (client MockClient) ExecuteCommand(containerID string, command string) error {
+	return nil
+}
+
+func (client MockClient) IsContainerStale(c container.Container) (bool, error) {
+	return true, nil
+}
+

internal/actions/mocks/container.go

@@ -0,0 +1,28 @@
+package mocks
+
+import (
+	"github.com/containrrr/watchtower/pkg/container"
+	"github.com/docker/docker/api/types"
+	container2 "github.com/docker/docker/api/types/container"
+	"time"
+)
+
+func CreateMockContainer(id string, name string, image string, created time.Time) 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),
+		},
+	}
+	return *container.NewContainer(
+		&content,
+		&types.ImageInspect{
+			ID: image,
+		},
+	)
+}