|
| 1 | +package controllerstest |
| 2 | + |
| 3 | +import ( |
| 4 | + "context" |
| 5 | + "testing" |
| 6 | + "time" |
| 7 | + |
| 8 | + "github.com/loft-sh/cluster-api-provider-vcluster/api/v1alpha1" |
| 9 | + "github.com/loft-sh/cluster-api-provider-vcluster/controllers" |
| 10 | + "github.com/onsi/ginkgo/v2" |
| 11 | + "github.com/onsi/gomega" |
| 12 | + corev1 "k8s.io/api/core/v1" |
| 13 | + metav1 "k8s.io/apimachinery/pkg/apis/meta/v1" |
| 14 | + "k8s.io/apimachinery/pkg/runtime" |
| 15 | + "k8s.io/apimachinery/pkg/types" |
| 16 | + |
| 17 | + fakeclientset "k8s.io/client-go/kubernetes/fake" |
| 18 | + ctrl "sigs.k8s.io/controller-runtime" |
| 19 | + fakeclient "sigs.k8s.io/controller-runtime/pkg/client/fake" |
| 20 | +) |
| 21 | + |
| 22 | +var ( |
| 23 | + kubeconfigBytes = []byte(` |
| 24 | +kind: Config |
| 25 | +apiVersion: v1 |
| 26 | +clusters: |
| 27 | +- cluster: |
| 28 | + api-version: v1 |
| 29 | + server: https://test:443 |
| 30 | + certificate-authority: test.crt |
| 31 | + name: kubeconfig-cluster |
| 32 | +users: |
| 33 | +- name: kubeconfig-user |
| 34 | + user: |
| 35 | + client-certificate-data: LS0tLS1CRUdJTiBDRVJUSUZJQ0FURS0tLS0tCk1JSUJrakNDQVRlZ0F3SUJBZ0lJT2FQRzhMc21MNWd3Q2dZSUtvWkl6ajBFQXdJd0l6RWhNQjhHQTFVRUF3d1kKYXpOekxXTnNhV1Z1ZEMxallVQXhOekE0TURBNE1qRXpNQjRYRFRJME1ESXhOVEUwTkRNek0xb1hEVEkxTURJeApOREUwTkRNek0xb3dNREVYTUJVR0ExVUVDaE1PYzNsemRHVnRPbTFoYzNSbGNuTXhGVEFUQmdOVkJBTVRESE41CmMzUmxiVHBoWkcxcGJqQlpNQk1HQnlxR1NNNDlBZ0VHQ0NxR1NNNDlBd0VIQTBJQUJDbysyRzRzQ0pjaTVZTlMKMkp6VTd5ZnEzSUR0dE1tcnU2bGtGV2NMR2FJSVRTVDZPbFdzaDdaYkJRb3FrTkk5c3dTOStCWHptV2FOQ1FzRgp1Q0ZaL0F1alNEQkdNQTRHQTFVZER3RUIvd1FFQXdJRm9EQVRCZ05WSFNVRUREQUtCZ2dyQmdFRkJRY0RBakFmCkJnTlZIU01FR0RBV2dCUyt0MG1hMFR2ZHN5d2RuVGpYd0ExWis0eFZJakFLQmdncWhrak9QUVFEQWdOSkFEQkcKQWlFQThjZXNlcWhjOFpGU0Z3TERzdDJYUS9lU0xiVWFuNnNYenhFeHFtSlNEbXNDSVFEcDdJWmRJd3FaVmY2WQpQMWRaOWwzeE9JTDFRL2Y5VXdNVC9aOFRaZEZJa2c9PQotLS0tLUVORCBDRVJUSUZJQ0FURS0tLS0tCi0tLS0tQkVHSU4gQ0VSVElGSUNBVEUtLS0tLQpNSUlCZGpDQ0FSMmdBd0lCQWdJQkFEQUtCZ2dxaGtqT1BRUURBakFqTVNFd0h3WURWUVFEREJock0zTXRZMnhwClpXNTBMV05oUURFM01EZ3dNRGd5TVRNd0hoY05NalF3TWpFMU1UUTBNek16V2hjTk16UXdNakV5TVRRME16TXoKV2pBak1TRXdId1lEVlFRRERCaHJNM010WTJ4cFpXNTBMV05oUURFM01EZ3dNRGd5TVRNd1dUQVRCZ2NxaGtqTwpQUUlCQmdncWhrak9QUU1CQndOQ0FBVCtZbTVnL0o4TzIwQ0llSFB4Z2hRWTBXajl3QVZzc0QxdHRzS0VnMFFRCjA3UDNLZEttV3AzS3BvV3FkdkN4dTNFMkp4ZDBGVDh5eG1IOVJiamVXRW90bzBJd1FEQU9CZ05WSFE4QkFmOEUKQkFNQ0FxUXdEd1lEVlIwVEFRSC9CQVV3QXdFQi96QWRCZ05WSFE0RUZnUVV2cmRKbXRFNzNiTXNIWjA0MThBTgpXZnVNVlNJd0NnWUlLb1pJemowRUF3SURSd0F3UkFJZ1VldS9yVnBmc1NoUUZmSjIyb05CMVhwY1djUWFPY2FBCnF4ZGg0dzhGdHBRQ0lIdmVTRE00clN2V3ZGZktROXRWTDRFZkpUdDc2cWliMFMyY2FBdDQwUHNGCi0tLS0tRU5EIENFUlRJRklDQVRFLS0tLS0K |
| 36 | + client-key-data: LS0tLS1CRUdJTiBFQyBQUklWQVRFIEtFWS0tLS0tCk1IY0NBUUVFSVBWS2JlQzJua2JaZ1UxZUNaS2NxUHpnSXd0MWxtOGcxZFNRaENoaHRURWVvQW9HQ0NxR1NNNDkKQXdFSG9VUURRZ0FFS2o3WWJpd0lseUxsZzFMWW5OVHZKK3JjZ08yMHlhdTdxV1FWWndzWm9naE5KUG82VmF5SAp0bHNGQ2lxUTBqMnpCTDM0RmZPWlpvMEpDd1c0SVZuOEN3PT0KLS0tLS1FTkQgRUMgUFJJVkFURSBLRVktLS0tLQo= |
| 37 | +`) |
| 38 | +) |
| 39 | + |
| 40 | +func TestRunControllersTests(t *testing.T) { |
| 41 | + gomega.RegisterFailHandler(ginkgo.Fail) |
| 42 | + ginkgo.RunSpecs(t, "controllers suite") |
| 43 | +} |
| 44 | + |
| 45 | +var _ = ginkgo.Describe("Vcluster Controller test", func() { |
| 46 | + ginkgo.Context("Reconcile", func() { |
| 47 | + var ( |
| 48 | + reconciler *controllers.VClusterReconciler |
| 49 | + ctx context.Context |
| 50 | + scheme *runtime.Scheme |
| 51 | + hemlClient *MockHelmClient |
| 52 | + ) |
| 53 | + |
| 54 | + ginkgo.BeforeEach(func() { |
| 55 | + scheme = runtime.NewScheme() |
| 56 | + err := v1alpha1.AddToScheme(scheme) |
| 57 | + gomega.Expect(err).NotTo(gomega.HaveOccurred()) |
| 58 | + |
| 59 | + err = corev1.AddToScheme(scheme) |
| 60 | + gomega.Expect(err).NotTo(gomega.HaveOccurred()) |
| 61 | + |
| 62 | + ctx = context.Background() |
| 63 | + hemlClient = &MockHelmClient{} |
| 64 | + }) |
| 65 | + |
| 66 | + ginkgo.It("reconcile successfully", func() { |
| 67 | + vCluster := &v1alpha1.VCluster{ |
| 68 | + ObjectMeta: metav1.ObjectMeta{ |
| 69 | + Name: "test-vcluster", |
| 70 | + Namespace: "default", |
| 71 | + }, |
| 72 | + Spec: v1alpha1.VClusterSpec{}, |
| 73 | + } |
| 74 | + secret := &corev1.Secret{ |
| 75 | + ObjectMeta: metav1.ObjectMeta{ |
| 76 | + Namespace: vCluster.Namespace, |
| 77 | + Name: "vc-test-vcluster", |
| 78 | + }, |
| 79 | + Data: map[string][]byte{ |
| 80 | + "config": kubeconfigBytes, |
| 81 | + }, |
| 82 | + } |
| 83 | + hemlClient.On("Upgrade").Return(nil) |
| 84 | + f := fakeclientset.NewSimpleClientset() |
| 85 | + |
| 86 | + _, err := f.CoreV1().ServiceAccounts("default").Create(context.Background(), &corev1.ServiceAccount{ |
| 87 | + ObjectMeta: metav1.ObjectMeta{ |
| 88 | + Name: "default", |
| 89 | + Namespace: "default", |
| 90 | + }, |
| 91 | + }, metav1.CreateOptions{}) |
| 92 | + gomega.Expect(err).NotTo(gomega.HaveOccurred()) |
| 93 | + |
| 94 | + reconciler = &controllers.VClusterReconciler{ |
| 95 | + Client: fakeclient.NewClientBuilder().WithScheme(scheme).WithObjects(vCluster, secret).Build(), |
| 96 | + HelmClient: hemlClient, |
| 97 | + Scheme: scheme, |
| 98 | + ClientConfigGetter: &fakeConfigGetter{ |
| 99 | + fake: f, |
| 100 | + }, |
| 101 | + HTTPClientGetter: &fakeHTTPClientGetter{}, |
| 102 | + } |
| 103 | + req := ctrl.Request{ |
| 104 | + NamespacedName: types.NamespacedName{ |
| 105 | + Name: vCluster.Name, |
| 106 | + Namespace: vCluster.Namespace, |
| 107 | + }, |
| 108 | + } |
| 109 | + result, err := reconciler.Reconcile(ctx, req) |
| 110 | + gomega.Expect(err).NotTo(gomega.HaveOccurred()) |
| 111 | + gomega.Expect(result.RequeueAfter).Should(gomega.Equal(time.Minute)) |
| 112 | + }) |
| 113 | + }) |
| 114 | + |
| 115 | +}) |
0 commit comments