opensource/podman
1
0
Fork 0
mirror of https://github.com/containers/podman.git synced 2025-06-25 20:26:51 +08:00
Code Issues Packages Projects Releases Wiki Activity

Add more tests for liveness probes with default hostname & named ports

Browse Source 
Signed-off-by: Hedayat Vatankhah <hedayat.fwd@gmail.com>
This commit is contained in:
Hedayat Vatankhah
2023-07-13 18:15:02 +03:30
parent 600de05e3b
commit a8d8c9497a
1 changed files with 129 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

130
pkg/specgen/generate/kube/play_test.go
View File

@ -1247,6 +1247,131 @@ func TestHttpLivenessProbe(t *testing.T) {
true, true,
"http://localhost:80/", "http://localhost:80/",
}, },
{
"HttpLivenessProbeNamedPort",
specgen.SpecGenerator{},
v1.Container{
LivenessProbe: &v1.Probe{
Handler: v1.Handler{
HTTPGet: &v1.HTTPGetAction{
Port: intstr.FromString("httpPort"),
},
},
},
Ports: []v1.ContainerPort{
{Name: "servicePort", ContainerPort: 7000},
{Name: "httpPort", ContainerPort: 8000},
},
},
"always",
true,
"http://localhost:8000/",
},
}
for _, test := range tests {
test := test
t.Run(test.name, func(t *testing.T) {
err := setupLivenessProbe(&test.specGenerator, test.container, test.restartPolicy)
if err == nil {
assert.Equal(t, err == nil, test.succeed)
assert.Contains(t, test.specGenerator.ContainerHealthCheckConfig.HealthConfig.Test, test.expectedURL)
}
})
}
}
func TestTCPLivenessProbe(t *testing.T) {
tests := []struct {
name string
specGenerator specgen.SpecGenerator
container v1.Container
restartPolicy string
succeed bool
expectedHost string
expectedPort string
}{
{
"TCPLivenessProbeNormal",
specgen.SpecGenerator{},
v1.Container{
LivenessProbe: &v1.Probe{
Handler: v1.Handler{
TCPSocket: &v1.TCPSocketAction{
Host: "127.0.0.1",
Port: intstr.FromInt(8080),
},
},
},
},
"always",
true,
"127.0.0.1",
"8080",
},
{
"TCPLivenessProbeHostUsesDefault",
specgen.SpecGenerator{},
v1.Container{
LivenessProbe: &v1.Probe{
Handler: v1.Handler{
TCPSocket: &v1.TCPSocketAction{
Port: intstr.FromInt(200),
},
},
},
},
"always",
true,
"localhost",
"200",
},
{
"TCPLivenessProbeUseNamedPort",
specgen.SpecGenerator{},
v1.Container{
LivenessProbe: &v1.Probe{
Handler: v1.Handler{
TCPSocket: &v1.TCPSocketAction{
Port: intstr.FromString("servicePort"),
Host: "myservice.domain.com",
},
},
},
Ports: []v1.ContainerPort{
{ContainerPort: 6000},
{Name: "servicePort", ContainerPort: 4000},
{Name: "2ndServicePort", ContainerPort: 3000},
},
},
"always",
true,
"myservice.domain.com",
"4000",
},
{
"TCPLivenessProbeInvalidPortName",
specgen.SpecGenerator{},
v1.Container{
LivenessProbe: &v1.Probe{
Handler: v1.Handler{
TCPSocket: &v1.TCPSocketAction{
Port: intstr.FromString("3rdservicePort"),
Host: "myservice.domain.com",
},
},
},
Ports: []v1.ContainerPort{
{ContainerPort: 6000},
{Name: "servicePort", ContainerPort: 4000},
{Name: "2ndServicePort", ContainerPort: 3000},
},
},
"always",
false,
"myservice.domain.com",
"4000",
},
} }
for _, test := range tests { for _, test := range tests {
@ -1254,7 +1379,10 @@ func TestHttpLivenessProbe(t *testing.T) {
t.Run(test.name, func(t *testing.T) { t.Run(test.name, func(t *testing.T) {
err := setupLivenessProbe(&test.specGenerator, test.container, test.restartPolicy) err := setupLivenessProbe(&test.specGenerator, test.container, test.restartPolicy)
assert.Equal(t, err == nil, test.succeed) assert.Equal(t, err == nil, test.succeed)
assert.Contains(t, test.specGenerator.ContainerHealthCheckConfig.HealthConfig.Test, test.expectedURL) if err == nil {
assert.Contains(t, test.specGenerator.ContainerHealthCheckConfig.HealthConfig.Test, test.expectedHost)
assert.Contains(t, test.specGenerator.ContainerHealthCheckConfig.HealthConfig.Test, test.expectedPort)
}
}) })
} }
} }