instascale_controller_test.go

package controllers

import (
	"context"

	. "github.com/onsi/ginkgo/v2"
	. "github.com/onsi/gomega"

	mfc "github.com/manifestival/controller-runtime-client"
	mf "github.com/manifestival/manifestival"

	codeflarev1alpha1 "github.com/project-codeflare/codeflare-operator/api/v1alpha1"
)

const (
	instascaleCRCase1             = "./testdata/instascale_test_cases/case_1.yaml"
	instascaleCRCase2             = "./testdata/instascale_test_cases/case_2.yaml"
	instascaleCRCase3             = "./testdata/instascale_test_cases/case_3.yaml"
	instascaleConfigMap1          = "./testdata/instascale_test_results/case_1/configmap.yaml"
	instascaleServiceAccount1     = "./testdata/instascale_test_results/case_1/serviceaccount.yaml"
	instascaleClusterRole1        = "./testdata/instascale_test_results/case_1/clusterrole.yaml"
	instascaleClusterRoleBinding1 = "./testdata/instascale_test_results/case_1/clusterrolebinding.yaml"
	instascaleDeployment1         = "./testdata/instascale_test_results/case_1/deployment.yaml"
	instascaleDeployment2         = "./testdata/instascale_test_results/case_2/deployment.yaml"
	instascaleDeployment3         = "./testdata/instascale_test_results/case_3/deployment.yaml"
)

func deployInstaScale(ctx context.Context, path string, opts mf.Option) {
	instascale := &codeflarev1alpha1.InstaScale{}
	err := convertToStructuredResource(path, instascale, opts)
	Expect(err).NotTo(HaveOccurred())
	Expect(k8sClient.Create(ctx, instascale)).Should(Succeed())
}

var _ = Describe("The Instascale Controller", func() {
	client := mfc.NewClient(k8sClient)
	opts := mf.UseClient(client)
	ctx := context.Background()

	Context("In a namespace, when a blank InstaScale Custom Resource is deployed", func() {

		It("It should deploy InstaScale with default settings", func() {
			deployInstaScale(ctx, instascaleCRCase1, opts)
			compareConfigMaps(instascaleConfigMap1, opts)
			compareServiceAccounts(instascaleServiceAccount1, opts)
			compareDeployments(instascaleDeployment1, opts)
			compareClusterRoles(instascaleClusterRole1, opts)
			compareClusterRoleBindings(instascaleClusterRoleBinding1, opts)
		})
	})

	Context("In a namespace, InstaScale ControllerResources is given", func() {

		It("It should deploy InstaScale with the given ControllerResources", func() {
			deployInstaScale(ctx, instascaleCRCase2, opts)
			compareDeployments(instascaleDeployment2, opts)
		})
	})

	Context("When an InstaScale resource with a custom image is given", func() {

		It("It should deploy InstaScale with the given controller image", func() {
			deployInstaScale(ctx, instascaleCRCase3, opts)
			compareDeployments(instascaleDeployment3, opts)
		})
	})
})