mirror of
https://github.com/containrrr/watchtower.git
synced 2025-12-16 15:10:12 +01:00
317 lines
8.3 KiB
Go
317 lines
8.3 KiB
Go
package actions_test
|
|
|
|
import (
|
|
"github.com/containrrr/watchtower/internal/actions"
|
|
. "github.com/containrrr/watchtower/internal/actions/mocks"
|
|
"github.com/containrrr/watchtower/pkg/container"
|
|
"github.com/containrrr/watchtower/pkg/container/mocks"
|
|
"github.com/containrrr/watchtower/pkg/metrics"
|
|
"github.com/containrrr/watchtower/pkg/sorter"
|
|
"github.com/containrrr/watchtower/pkg/types"
|
|
container2 "github.com/docker/docker/api/types/container"
|
|
cli "github.com/docker/docker/client"
|
|
. "github.com/onsi/ginkgo"
|
|
. "github.com/onsi/gomega"
|
|
"time"
|
|
)
|
|
|
|
var _ = Describe("the update action", func() {
|
|
var dockerClient cli.CommonAPIClient
|
|
var client MockClient
|
|
|
|
BeforeEach(func() {
|
|
server := mocks.NewMockAPIServer()
|
|
dockerClient, _ = cli.NewClientWithOpts(
|
|
cli.WithHost(server.URL),
|
|
cli.WithHTTPClient(server.Client()))
|
|
})
|
|
|
|
When("watchtower has been instructed to clean up", func() {
|
|
BeforeEach(func() {
|
|
pullImages := false
|
|
removeVolumes := false
|
|
client = CreateMockClient(
|
|
&TestData{
|
|
NameOfContainerToKeep: "test-container-02",
|
|
Containers: []container.Container{
|
|
CreateMockContainer(
|
|
"test-container-01",
|
|
"test-container-01",
|
|
"fake-image:latest",
|
|
time.Now().AddDate(0, 0, -1),
|
|
nil),
|
|
CreateMockContainer(
|
|
"test-container-02",
|
|
"test-container-02",
|
|
"fake-image:latest",
|
|
time.Now(),
|
|
nil),
|
|
CreateMockContainer(
|
|
"test-container-02",
|
|
"test-container-02",
|
|
"fake-image:latest",
|
|
time.Now(),
|
|
nil),
|
|
},
|
|
},
|
|
dockerClient,
|
|
pullImages,
|
|
removeVolumes,
|
|
)
|
|
})
|
|
|
|
//When("there are multiple containers using the same image", func() {
|
|
// It("should only try to remove the image once", func() {
|
|
//
|
|
// _, err := actions.Update(client, types.UpdateParams{Cleanup: true})
|
|
// Expect(err).NotTo(HaveOccurred())
|
|
// Expect(client.TestData.TriedToRemoveImageCount).To(Equal(1))
|
|
// })
|
|
// When("performing a rolling restart update", func() {
|
|
// It("should try to remove the image once", func() {
|
|
//
|
|
// _, err := actions.Update(client, types.UpdateParams{Cleanup: true, RollingRestart: true})
|
|
// Expect(err).NotTo(HaveOccurred())
|
|
// Expect(client.TestData.TriedToRemoveImageCount).To(Equal(1))
|
|
// })
|
|
// })
|
|
//})
|
|
|
|
When("there are multiple containers using different images", func() {
|
|
It("should try to remove each of them", func() {
|
|
client.TestData.Containers = append(
|
|
client.TestData.Containers,
|
|
CreateMockContainer(
|
|
"unique-test-container",
|
|
"unique-test-container",
|
|
"unique-fake-image:latest",
|
|
time.Now(),
|
|
nil,
|
|
),
|
|
)
|
|
_, err := actions.Update(client, types.UpdateParams{Cleanup: true})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
Expect(client.TestData.TriedToRemoveImageCount).To(Equal(2))
|
|
})
|
|
})
|
|
})
|
|
|
|
When("there are containers with and without links", func() {
|
|
links := [7][]string{
|
|
{},
|
|
{"k-container-01"},
|
|
{"k-container-02"},
|
|
{},
|
|
{"t-container-01"},
|
|
{"t-container-02"},
|
|
{},
|
|
}
|
|
BeforeEach(func() {
|
|
pullImages := false
|
|
removeVolumes := false
|
|
client = CreateMockClient(
|
|
&TestData{
|
|
NameOfContainerToKeep: "",
|
|
Containers: []container.Container{
|
|
CreateMockContainer(
|
|
"k-container-03",
|
|
"k-container-03",
|
|
"fake-image:latest",
|
|
time.Now().Add(time.Second*4),
|
|
links[2]),
|
|
CreateMockContainer(
|
|
"k-container-02",
|
|
"k-container-02",
|
|
"fake-image:latest",
|
|
time.Now().Add(time.Second*2),
|
|
links[1]),
|
|
CreateMockContainer(
|
|
"k-container-01",
|
|
"k-container-01",
|
|
"fake-image:latest",
|
|
time.Now(),
|
|
links[0]),
|
|
|
|
CreateMockContainer(
|
|
"t-container-03",
|
|
"t-container-03",
|
|
"fake-image-2:latest",
|
|
time.Now().Add(time.Second*4),
|
|
links[5]),
|
|
CreateMockContainer(
|
|
"t-container-02",
|
|
"t-container-02",
|
|
"fake-image-2:latest",
|
|
time.Now().Add(time.Second*2),
|
|
links[4]),
|
|
CreateMockContainer(
|
|
"t-container-01",
|
|
"t-container-01",
|
|
"fake-image-2:latest",
|
|
time.Now(),
|
|
links[3]),
|
|
|
|
CreateMockContainer(
|
|
"x-container-01",
|
|
"x-container-01",
|
|
"fake-image-1:latest",
|
|
time.Now(),
|
|
links[6]),
|
|
CreateMockContainer(
|
|
"x-container-02",
|
|
"x-container-02",
|
|
"fake-image-1:latest",
|
|
time.Now().Add(time.Second*2),
|
|
links[6]),
|
|
CreateMockContainer(
|
|
"x-container-03",
|
|
"x-container-03",
|
|
"fake-image-1:latest",
|
|
time.Now().Add(time.Second*4),
|
|
links[6]),
|
|
},
|
|
},
|
|
dockerClient,
|
|
pullImages,
|
|
removeVolumes,
|
|
)
|
|
})
|
|
|
|
When("there are multiple containers with links", func() {
|
|
It("should create appropriate dependency sorted lists", func() {
|
|
metric := &metrics.Metric{}
|
|
containers, err := actions.ScanForContainerUpdates(client, types.UpdateParams{Cleanup: true}, metric)
|
|
undirectedNodes := actions.CreateUndirectedLinks(containers)
|
|
dependencySortedGraphs, err := sorter.SortByDependencies(containers, undirectedNodes)
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
var output [][]string
|
|
|
|
for _, i := range dependencySortedGraphs {
|
|
var inner []string
|
|
for _, j := range i {
|
|
inner = append(inner, j.Name())
|
|
}
|
|
output = append(output, inner)
|
|
}
|
|
|
|
ExpectedOutput := [][]string{
|
|
{"k-container-01", "k-container-02", "k-container-03"},
|
|
{"t-container-01", "t-container-02", "t-container-03"},
|
|
{"x-container-01"},
|
|
{"x-container-02"},
|
|
{"x-container-03"},
|
|
}
|
|
|
|
Expect(output).To(Equal(ExpectedOutput))
|
|
})
|
|
})
|
|
|
|
When("there are multiple containers using the same image", func() {
|
|
It("should stop and restart containers in a correct order", func() {
|
|
_, err := actions.Update(client, types.UpdateParams{Cleanup: true})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
|
|
ExpectedStopOutput := []string{
|
|
"k-container-03",
|
|
"k-container-02",
|
|
"k-container-01",
|
|
"t-container-03",
|
|
"t-container-02",
|
|
"t-container-01",
|
|
"x-container-01",
|
|
"x-container-02",
|
|
"x-container-03",
|
|
}
|
|
|
|
ExpectedRestartOutput := []string{
|
|
"k-container-01",
|
|
"k-container-02",
|
|
"k-container-03",
|
|
"t-container-01",
|
|
"t-container-02",
|
|
"t-container-03",
|
|
"x-container-01",
|
|
"x-container-02",
|
|
"x-container-03",
|
|
}
|
|
|
|
Expect(client.TestData.StopOrder).To(Equal(ExpectedStopOutput))
|
|
Expect(client.TestData.RestartOrder).To(Equal(ExpectedRestartOutput))
|
|
})
|
|
})
|
|
|
|
})
|
|
|
|
When("watchtower has been instructed to monitor only", func() {
|
|
When("certain containers are set to monitor only", func() {
|
|
BeforeEach(func() {
|
|
client = CreateMockClient(
|
|
&TestData{
|
|
NameOfContainerToKeep: "test-container-02",
|
|
Containers: []container.Container{
|
|
CreateMockContainer(
|
|
"test-container-01",
|
|
"test-container-01",
|
|
"fake-image1:latest",
|
|
time.Now(),
|
|
nil),
|
|
CreateMockContainerWithConfig(
|
|
"test-container-02",
|
|
"test-container-02",
|
|
"fake-image2:latest",
|
|
time.Now(),
|
|
&container2.Config{
|
|
Labels: map[string]string{
|
|
"com.centurylinklabs.watchtower.monitor-only": "true",
|
|
},
|
|
}),
|
|
},
|
|
},
|
|
dockerClient,
|
|
false,
|
|
false,
|
|
)
|
|
})
|
|
|
|
It("should not update those containers", func() {
|
|
_, err := actions.Update(client, types.UpdateParams{Cleanup: true})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
Expect(client.TestData.TriedToRemoveImageCount).To(Equal(1))
|
|
})
|
|
})
|
|
|
|
When("monitor only is set globally", func() {
|
|
BeforeEach(func() {
|
|
client = CreateMockClient(
|
|
&TestData{
|
|
Containers: []container.Container{
|
|
CreateMockContainer(
|
|
"test-container-01",
|
|
"test-container-01",
|
|
"fake-image:latest",
|
|
time.Now(),
|
|
nil),
|
|
CreateMockContainer(
|
|
"test-container-02",
|
|
"test-container-02",
|
|
"fake-image:latest",
|
|
time.Now(),
|
|
nil),
|
|
},
|
|
},
|
|
dockerClient,
|
|
false,
|
|
false,
|
|
)
|
|
})
|
|
|
|
It("should not update any containers", func() {
|
|
_, err := actions.Update(client, types.UpdateParams{MonitorOnly: true})
|
|
Expect(err).NotTo(HaveOccurred())
|
|
Expect(client.TestData.TriedToRemoveImageCount).To(Equal(0))
|
|
})
|
|
})
|
|
|
|
})
|
|
})
|