gardener/pkg/gardenlet/controller/bastion/add_test.go at master · stackitcloud/gardener · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
// SPDX-FileCopyrightText: SAP SE or an SAP affiliate company and Gardener contributors
//
// SPDX-License-Identifier: Apache-2.0

package bastion_test

import (
	"context"

	"github.com/go-logr/logr"
	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"
	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"
	fakeclient "sigs.k8s.io/controller-runtime/pkg/client/fake"
	"sigs.k8s.io/controller-runtime/pkg/reconcile"

	gardencorev1beta1 "github.com/gardener/gardener/pkg/apis/core/v1beta1"
	extensionsv1alpha1 "github.com/gardener/gardener/pkg/apis/extensions/v1alpha1"
	operationsv1alpha1 "github.com/gardener/gardener/pkg/apis/operations/v1alpha1"
	. "github.com/gardener/gardener/pkg/gardenlet/controller/bastion"
)

const (
	bastionName = "foo"
	projectName = "project"
)

var _ = Describe("Add", func() {
	var (
		ctx        = context.TODO()
		log        logr.Logger
		fakeClient client.Client
		reconciler *Reconciler

		operationsBastion *operationsv1alpha1.Bastion
		extensionsBastion *extensionsv1alpha1.Bastion
		cluster           *extensionsv1alpha1.Cluster

		shootTechnicalID = "shoot--" + projectName + "--shootName"
		projectNamespace = "garden" + projectName
	)

	BeforeEach(func() {
		testScheme := runtime.NewScheme()
		Expect(extensionsv1alpha1.AddToScheme(testScheme)).To(Succeed())
		fakeClient = fakeclient.NewClientBuilder().WithScheme(testScheme).Build()

		cluster = &extensionsv1alpha1.Cluster{
			ObjectMeta: metav1.ObjectMeta{
				Name: shootTechnicalID,
			},
			Spec: extensionsv1alpha1.ClusterSpec{
				Shoot: runtime.RawExtension{
					Object: &gardencorev1beta1.Shoot{
						ObjectMeta: metav1.ObjectMeta{
							Name:      "shoot",
							Namespace: projectNamespace,
						},
					},
				},
			},
		}

		reconciler = &Reconciler{
			SeedClient: fakeClient,
		}
	})

	Describe("#MapExtensionsBastionToOperationsBastion", func() {
		BeforeEach(func() {
			log = logr.Discard()

			operationsBastion = &operationsv1alpha1.Bastion{
				ObjectMeta: metav1.ObjectMeta{
					Name:      bastionName,
					Namespace: projectNamespace,
				},
			}

			extensionsBastion = &extensionsv1alpha1.Bastion{
				ObjectMeta: metav1.ObjectMeta{
					Name:      bastionName,
					Namespace: shootTechnicalID,
				},
			}
		})

		It("should map the extensions Bastion to operations Bastion", func() {
			Expect(fakeClient.Create(ctx, cluster)).To(Succeed())
			Expect(reconciler.MapExtensionsBastionToOperationsBastion(log)(ctx, extensionsBastion)).To(ConsistOf(
				reconcile.Request{NamespacedName: types.NamespacedName{Namespace: operationsBastion.Namespace, Name: operationsBastion.Name}},
			))
		})

		It("should return nil if the cluster is not found", func() {
			Expect(reconciler.MapExtensionsBastionToOperationsBastion(log)(ctx, extensionsBastion)).To(BeNil())
		})

		It("should return nil when shoot is not present in the cluster", func() {
			cluster.Spec.Shoot.Object = nil
			Expect(fakeClient.Create(ctx, cluster)).To(Succeed())

			Expect(reconciler.MapExtensionsBastionToOperationsBastion(log)(ctx, extensionsBastion)).To(BeNil())
		})
	})
})