mirrors/prometheus-operator
mirrors/prometheus-operator
1
0
Fork 0
mirror of https://github.com/prometheus-operator/prometheus-operator.git synced 2025-04-20 19:29:10 +00:00
Code Activity

pkg/prometheus: add test case for extra containers

Browse source 
add a very basic test case asserting that additional container specs in
the prometheus crd are added to the statefulset that reflects the
current behavior
This commit is contained in:
Steve Mitchell 2019-02-27 19:58:59 -05:00 • committed by Steve Mitchell
parent e05267a050
commit ad35d199b0
1 changed files with 50 additions and 0 deletions

50
pkg/prometheus/statefulset_test.go
View file

@ -698,3 +698,53 @@ func TestSidecarsNoMemoryLimits(t *testing.T) {
}
}
}
func TestAdditionalContainers(t *testing.T) {
// The base to compare everything against
baseSet, err := makeStatefulSet(monitoringv1.Prometheus{}, defaultTestConfig, nil, "")
// Add an extra container
addSset, err := makeStatefulSet(monitoringv1.Prometheus{
Spec: monitoringv1.PrometheusSpec{
Containers: []v1.Container{
{
Name: "extra-container",
},
},
},
}, defaultTestConfig, nil, "")
require.NoError(t, err)
if len(baseSet.Spec.Template.Spec.Containers)+1 != len(addSset.Spec.Template.Spec.Containers) {
t.Fatalf("container count mismatch")
}
// Adding a new container with the same name results in
const existingContainerName = "prometheus"
modSset, err := makeStatefulSet(monitoringv1.Prometheus{
Spec: monitoringv1.PrometheusSpec{
Containers: []v1.Container{
{
Name: existingContainerName,
},
},
},
}, defaultTestConfig, nil, "")
require.NoError(t, err)
if len(baseSet.Spec.Template.Spec.Containers)+1 != len(modSset.Spec.Template.Spec.Containers) {
t.Fatalf("container count mismatch")
}
// Check that adding a container with an existing name results in 2 containers.
// Note: this is a valid CRD but k8s will not start the pod because there are 2 containers with the same name
var containersNamedPrometheus int
for _, c := range modSset.Spec.Template.Spec.Containers {
if c.Name == existingContainerName {
containersNamedPrometheus++
}
}
if containersNamedPrometheus != 2 {
t.Fatalf("expected 2 containers named prometheus but only found %d", containersNamedPrometheus)
}
}