@@ -24,7 +24,7 @@ import (
2424 "testing"
2525
2626 "github.com/stretchr/testify/assert"
27- v1 "k8s.io/api/core/v1"
27+ core "k8s.io/api/core/v1"
2828
2929 api "github.com/arangodb/kube-arangodb/pkg/apis/deployment/v1"
3030 "github.com/arangodb/kube-arangodb/pkg/apis/shared"
@@ -63,10 +63,10 @@ func Test_InitContainers(t *testing.T) {
6363 testCases := []TestCase {
6464 {
6565 name : "Same containers" ,
66- spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * v1 .Container ) {
66+ spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * core .Container ) {
6767 c .Image = "local:1.0"
6868 })),
69- status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * v1 .Container ) {
69+ status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * core .Container ) {
7070 c .Image = "local:1.0"
7171 })),
7272
@@ -80,10 +80,10 @@ func Test_InitContainers(t *testing.T) {
8080 },
8181 {
8282 name : "Containers with different image" ,
83- spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * v1 .Container ) {
83+ spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * core .Container ) {
8484 c .Image = "local:1.0"
8585 })),
86- status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * v1 .Container ) {
86+ status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * core .Container ) {
8787 c .Image = "local:2.0"
8888 })),
8989
@@ -104,10 +104,10 @@ func Test_InitContainers(t *testing.T) {
104104 testCases := []TestCase {
105105 {
106106 name : "Containers with different image but init rotation enforced" ,
107- spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * v1 .Container ) {
107+ spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * core .Container ) {
108108 c .Image = "local:1.0"
109109 })),
110- status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * v1 .Container ) {
110+ status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , nil ), addInitContainer ("sidecar" , func (c * core .Container ) {
111111 c .Image = "local:2.0"
112112 })),
113113
@@ -121,14 +121,14 @@ func Test_InitContainers(t *testing.T) {
121121 },
122122 {
123123 name : "Core Containers with different image but init rotation enforced" ,
124- spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * v1 .Container ) {
124+ spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * core .Container ) {
125125 c .Image = "local:1.0"
126- }), addInitContainer ("sidecar" , func (c * v1 .Container ) {
126+ }), addInitContainer ("sidecar" , func (c * core .Container ) {
127127 c .Image = "local:1.0"
128128 })),
129- status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * v1 .Container ) {
129+ status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * core .Container ) {
130130 c .Image = "local:2.0"
131- }), addInitContainer ("sidecar" , func (c * v1 .Container ) {
131+ }), addInitContainer ("sidecar" , func (c * core .Container ) {
132132 c .Image = "local:1.0"
133133 })),
134134
@@ -142,12 +142,12 @@ func Test_InitContainers(t *testing.T) {
142142 },
143143 {
144144 name : "Only core container change" ,
145- spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * v1 .Container ) {
145+ spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * core .Container ) {
146146 c .Image = "local:1.0"
147- }), addInitContainer (api .ServerGroupReservedInitContainerNameUpgrade , func (c * v1 .Container ) {
147+ }), addInitContainer (api .ServerGroupReservedInitContainerNameUpgrade , func (c * core .Container ) {
148148 c .Image = "local:1.0"
149149 })),
150- status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * v1 .Container ) {
150+ status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * core .Container ) {
151151 c .Image = "local:2.0"
152152 })),
153153
@@ -161,16 +161,16 @@ func Test_InitContainers(t *testing.T) {
161161 },
162162 {
163163 name : "Only core container change with sidecar" ,
164- spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * v1 .Container ) {
164+ spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * core .Container ) {
165165 c .Image = "local:1.0"
166- }), addInitContainer (api .ServerGroupReservedInitContainerNameUpgrade , func (c * v1 .Container ) {
166+ }), addInitContainer (api .ServerGroupReservedInitContainerNameUpgrade , func (c * core .Container ) {
167167 c .Image = "local:1.0"
168- }), addInitContainer ("sidecar" , func (c * v1 .Container ) {
168+ }), addInitContainer ("sidecar" , func (c * core .Container ) {
169169 c .Image = "local:1.0"
170170 })),
171- status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * v1 .Container ) {
171+ status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * core .Container ) {
172172 c .Image = "local:2.0"
173- }), addInitContainer ("sidecar" , func (c * v1 .Container ) {
173+ }), addInitContainer ("sidecar" , func (c * core .Container ) {
174174 c .Image = "local:1.0"
175175 })),
176176
@@ -184,16 +184,16 @@ func Test_InitContainers(t *testing.T) {
184184 },
185185 {
186186 name : "Only core container change with sidecar change" ,
187- spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * v1 .Container ) {
187+ spec : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * core .Container ) {
188188 c .Image = "local:1.0"
189- }), addInitContainer (api .ServerGroupReservedInitContainerNameUpgrade , func (c * v1 .Container ) {
189+ }), addInitContainer (api .ServerGroupReservedInitContainerNameUpgrade , func (c * core .Container ) {
190190 c .Image = "local:1.0"
191- }), addInitContainer ("sidecar" , func (c * v1 .Container ) {
191+ }), addInitContainer ("sidecar" , func (c * core .Container ) {
192192 c .Image = "local:1.0"
193193 })),
194- status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * v1 .Container ) {
194+ status : buildPodSpec (addInitContainer (api .ServerGroupReservedInitContainerNameUUID , func (c * core .Container ) {
195195 c .Image = "local:2.0"
196- }), addInitContainer ("sidecar" , func (c * v1 .Container ) {
196+ }), addInitContainer ("sidecar" , func (c * core .Container ) {
197197 c .Image = "local:2.0"
198198 })),
199199
@@ -271,6 +271,57 @@ func Test_Container_Args(t *testing.T) {
271271 runTestCases (t )(testCases ... )
272272}
273273
274+ func Test_Container_Ports (t * testing.T ) {
275+ testCases := []TestCase {
276+ {
277+ name : "Ports of server pod changed" ,
278+ spec : buildPodSpec (addContainer ("server" , func (c * core.Container ) {
279+ c .Ports = []core.ContainerPort {
280+ {
281+ Name : "server" ,
282+ ContainerPort : 8526 ,
283+ Protocol : "tcp" ,
284+ },
285+ }
286+ })),
287+ status : buildPodSpec (addContainer ("server" , func (c * core.Container ) {
288+ c .Ports = []core.ContainerPort {
289+ {
290+ Name : "server" ,
291+ ContainerPort : 8626 ,
292+ Protocol : "tcp" ,
293+ },
294+ }
295+ })),
296+ expectedMode : SilentRotation ,
297+ },
298+ {
299+ name : "Ports of sidecar pod changed" ,
300+ spec : buildPodSpec (addContainer ("sidecar" , func (c * core.Container ) {
301+ c .Ports = []core.ContainerPort {
302+ {
303+ Name : "server" ,
304+ ContainerPort : 8526 ,
305+ Protocol : "tcp" ,
306+ },
307+ }
308+ })),
309+ status : buildPodSpec (addContainer ("sidecar" , func (c * core.Container ) {
310+ c .Ports = []core.ContainerPort {
311+ {
312+ Name : "server" ,
313+ ContainerPort : 8626 ,
314+ Protocol : "tcp" ,
315+ },
316+ }
317+ })),
318+ expectedMode : GracefulRotation ,
319+ },
320+ }
321+
322+ runTestCases (t )(testCases ... )
323+ }
324+
274325func TestIsOnlyLogLevelChanged (t * testing.T ) {
275326 type args struct {
276327 specArgs []string
0 commit comments