mirror of
https://github.com/edgelesssys/constellation.git
synced 2024-12-14 18:34:31 -05:00
2afddcb0f8
* deps: update K8s dependencies * deps: bump controller runtime * chore: tidy * bump helm and migrate controller runtime * fix helm deprecation --------- Signed-off-by: Paul Meyer <49727155+katexochen@users.noreply.github.com> Co-authored-by: renovate[bot] <29139614+renovate[bot]@users.noreply.github.com> Co-authored-by: Paul Meyer <49727155+katexochen@users.noreply.github.com> Co-authored-by: Leonard Cohnen <lc@edgeless.systems>
292 lines
7.7 KiB
Go
292 lines
7.7 KiB
Go
/*
|
|
Copyright (c) Edgeless Systems GmbH
|
|
|
|
SPDX-License-Identifier: AGPL-3.0-only
|
|
*/
|
|
|
|
package controllers
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
corev1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
"k8s.io/apimachinery/pkg/types"
|
|
"sigs.k8s.io/controller-runtime/pkg/client"
|
|
"sigs.k8s.io/controller-runtime/pkg/event"
|
|
"sigs.k8s.io/controller-runtime/pkg/reconcile"
|
|
|
|
nodemaintenancev1beta1 "github.com/edgelesssys/constellation/v2/3rdparty/node-maintenance-operator/api/v1beta1"
|
|
updatev1alpha1 "github.com/edgelesssys/constellation/v2/operators/constellation-node-operator/v2/api/v1alpha1"
|
|
)
|
|
|
|
func TestScalingGroupImageChangedPredicate(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
event event.UpdateEvent
|
|
wantProcessing bool
|
|
}{
|
|
"old object is not a scaling group": {
|
|
event: event.UpdateEvent{
|
|
ObjectNew: &updatev1alpha1.ScalingGroup{},
|
|
},
|
|
},
|
|
"new object is not a scaling group": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &updatev1alpha1.ScalingGroup{},
|
|
},
|
|
},
|
|
"image reference is unchanged": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &updatev1alpha1.ScalingGroup{
|
|
Status: updatev1alpha1.ScalingGroupStatus{ImageReference: "image-reference"},
|
|
},
|
|
ObjectNew: &updatev1alpha1.ScalingGroup{
|
|
Status: updatev1alpha1.ScalingGroupStatus{ImageReference: "image-reference"},
|
|
},
|
|
},
|
|
},
|
|
"image reference has changed": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &updatev1alpha1.ScalingGroup{
|
|
Status: updatev1alpha1.ScalingGroupStatus{ImageReference: "old-image-reference"},
|
|
},
|
|
ObjectNew: &updatev1alpha1.ScalingGroup{
|
|
Status: updatev1alpha1.ScalingGroupStatus{ImageReference: "new-image-reference"},
|
|
},
|
|
},
|
|
wantProcessing: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
predicate := scalingGroupImageChangedPredicate()
|
|
assert.Equal(tc.wantProcessing, predicate.Update(tc.event))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestAutoscalerEnabledStatusChangedPredicate(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
event event.UpdateEvent
|
|
wantProcessing bool
|
|
}{
|
|
"old object is not an autoscaling strategy": {
|
|
event: event.UpdateEvent{
|
|
ObjectNew: &updatev1alpha1.AutoscalingStrategy{},
|
|
},
|
|
},
|
|
"new object is not an autoscaling strategy": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &updatev1alpha1.AutoscalingStrategy{},
|
|
},
|
|
},
|
|
"status is unchanged": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &updatev1alpha1.AutoscalingStrategy{
|
|
Status: updatev1alpha1.AutoscalingStrategyStatus{Enabled: true},
|
|
},
|
|
ObjectNew: &updatev1alpha1.AutoscalingStrategy{
|
|
Status: updatev1alpha1.AutoscalingStrategyStatus{Enabled: true},
|
|
},
|
|
},
|
|
},
|
|
"status has changed": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &updatev1alpha1.AutoscalingStrategy{
|
|
Status: updatev1alpha1.AutoscalingStrategyStatus{},
|
|
},
|
|
ObjectNew: &updatev1alpha1.AutoscalingStrategy{
|
|
Status: updatev1alpha1.AutoscalingStrategyStatus{Enabled: true},
|
|
},
|
|
},
|
|
wantProcessing: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
predicate := autoscalerEnabledStatusChangedPredicate()
|
|
assert.Equal(tc.wantProcessing, predicate.Update(tc.event))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNodeReadyPredicate(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
event event.UpdateEvent
|
|
wantProcessing bool
|
|
}{
|
|
"old object is not a node": {
|
|
event: event.UpdateEvent{
|
|
ObjectNew: &corev1.Node{},
|
|
},
|
|
},
|
|
"new object is not a node": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &corev1.Node{},
|
|
},
|
|
},
|
|
"status is unchanged": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &corev1.Node{},
|
|
ObjectNew: &corev1.Node{},
|
|
},
|
|
},
|
|
"node became ready": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &corev1.Node{
|
|
Status: corev1.NodeStatus{
|
|
Conditions: []corev1.NodeCondition{
|
|
{Type: corev1.NodeReady, Status: corev1.ConditionFalse},
|
|
},
|
|
},
|
|
},
|
|
ObjectNew: &corev1.Node{
|
|
Status: corev1.NodeStatus{
|
|
Conditions: []corev1.NodeCondition{
|
|
{Type: corev1.NodeReady, Status: corev1.ConditionTrue},
|
|
},
|
|
},
|
|
},
|
|
},
|
|
wantProcessing: true,
|
|
},
|
|
"node acquired provider id": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &corev1.Node{},
|
|
ObjectNew: &corev1.Node{
|
|
Spec: corev1.NodeSpec{
|
|
ProviderID: "provider-id",
|
|
},
|
|
},
|
|
},
|
|
wantProcessing: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
predicate := nodeReadyPredicate()
|
|
assert.Equal(tc.wantProcessing, predicate.Update(tc.event))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestNodeMaintenanceSucceededPredicate(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
event event.UpdateEvent
|
|
wantProcessing bool
|
|
}{
|
|
"old object is not a node maintenance resource": {
|
|
event: event.UpdateEvent{
|
|
ObjectNew: &nodemaintenancev1beta1.NodeMaintenance{},
|
|
},
|
|
},
|
|
"new object is not a node maintenance resource": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &nodemaintenancev1beta1.NodeMaintenance{},
|
|
},
|
|
},
|
|
"status is unchanged": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &nodemaintenancev1beta1.NodeMaintenance{
|
|
Status: nodemaintenancev1beta1.NodeMaintenanceStatus{
|
|
Phase: nodemaintenancev1beta1.MaintenanceRunning,
|
|
},
|
|
},
|
|
ObjectNew: &nodemaintenancev1beta1.NodeMaintenance{
|
|
Status: nodemaintenancev1beta1.NodeMaintenanceStatus{
|
|
Phase: nodemaintenancev1beta1.MaintenanceRunning,
|
|
},
|
|
},
|
|
},
|
|
},
|
|
"status has changed": {
|
|
event: event.UpdateEvent{
|
|
ObjectOld: &nodemaintenancev1beta1.NodeMaintenance{
|
|
Status: nodemaintenancev1beta1.NodeMaintenanceStatus{
|
|
Phase: nodemaintenancev1beta1.MaintenanceRunning,
|
|
},
|
|
},
|
|
ObjectNew: &nodemaintenancev1beta1.NodeMaintenance{
|
|
Status: nodemaintenancev1beta1.NodeMaintenanceStatus{
|
|
Phase: nodemaintenancev1beta1.MaintenanceSucceeded,
|
|
},
|
|
},
|
|
},
|
|
wantProcessing: true,
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
predicate := nodeMaintenanceSucceededPredicate()
|
|
assert.Equal(tc.wantProcessing, predicate.Update(tc.event))
|
|
})
|
|
}
|
|
}
|
|
|
|
func TestFindObjectsForScalingGroup(t *testing.T) {
|
|
scalingGroup := updatev1alpha1.ScalingGroup{
|
|
Spec: updatev1alpha1.ScalingGroupSpec{
|
|
NodeVersion: "nodeversion",
|
|
},
|
|
}
|
|
wantRequests := []reconcile.Request{
|
|
{
|
|
NamespacedName: types.NamespacedName{
|
|
Name: "nodeversion",
|
|
},
|
|
},
|
|
}
|
|
assert := assert.New(t)
|
|
reconciler := NodeVersionReconciler{}
|
|
requests := reconciler.findObjectsForScalingGroup(context.TODO(), &scalingGroup)
|
|
assert.ElementsMatch(wantRequests, requests)
|
|
}
|
|
|
|
func TestFindAllNodeVersions(t *testing.T) {
|
|
testCases := map[string]struct {
|
|
nodeVersion client.Object
|
|
listNodeVersionsErr error
|
|
wantRequests []reconcile.Request
|
|
}{
|
|
"getting the corresponding node images fails": {
|
|
listNodeVersionsErr: errors.New("get-node-version-err"),
|
|
},
|
|
"node image reconcile request is returned": {
|
|
nodeVersion: &updatev1alpha1.NodeVersion{
|
|
ObjectMeta: metav1.ObjectMeta{Name: "nodeversion"},
|
|
},
|
|
wantRequests: []reconcile.Request{
|
|
{
|
|
NamespacedName: types.NamespacedName{
|
|
Name: "nodeversion",
|
|
},
|
|
},
|
|
},
|
|
},
|
|
}
|
|
|
|
for name, tc := range testCases {
|
|
t.Run(name, func(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
reconciler := NodeVersionReconciler{
|
|
Client: newStubReaderClient(t, []runtime.Object{tc.nodeVersion}, nil, tc.listNodeVersionsErr),
|
|
}
|
|
requests := reconciler.findAllNodeVersions(context.TODO(), nil)
|
|
assert.ElementsMatch(tc.wantRequests, requests)
|
|
})
|
|
}
|
|
}
|