github.com/1aal/kubeblocks@v0.0.0-20231107070852-e1c03e598921/pkg/controller/builder/builder_job_test.go (about)

     1  /*
     2  Copyright (C) 2022-2023 ApeCloud Co., Ltd
     3  
     4  This file is part of KubeBlocks project
     5  
     6  This program is free software: you can redistribute it and/or modify
     7  it under the terms of the GNU Affero General Public License as published by
     8  the Free Software Foundation, either version 3 of the License, or
     9  (at your option) any later version.
    10  
    11  This program is distributed in the hope that it will be useful
    12  but WITHOUT ANY WARRANTY; without even the implied warranty of
    13  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    14  GNU Affero General Public License for more details.
    15  
    16  You should have received a copy of the GNU Affero General Public License
    17  along with this program.  If not, see <http://www.gnu.org/licenses/>.
    18  */
    19  
    20  package builder
    21  
    22  import (
    23  	. "github.com/onsi/ginkgo/v2"
    24  	. "github.com/onsi/gomega"
    25  
    26  	corev1 "k8s.io/api/core/v1"
    27  )
    28  
    29  var _ = Describe("job builder", func() {
    30  	It("should work well", func() {
    31  		const (
    32  			name = "foo"
    33  			ns   = "default"
    34  			port = int32(12345)
    35  		)
    36  		pod := NewPodBuilder(ns, "foo").
    37  			AddContainer(corev1.Container{
    38  				Name:  "foo",
    39  				Image: "bar",
    40  				Ports: []corev1.ContainerPort{
    41  					{
    42  						Name:          "foo",
    43  						Protocol:      corev1.ProtocolTCP,
    44  						ContainerPort: port,
    45  					},
    46  				},
    47  			}).GetObject()
    48  		template := corev1.PodTemplateSpec{
    49  			ObjectMeta: pod.ObjectMeta,
    50  			Spec:       pod.Spec,
    51  		}
    52  		selectorKey, selectorValue := "foo", "bar"
    53  		suspend := true
    54  		limit := int32(5)
    55  		ttl := int32(12)
    56  		job := NewJobBuilder(ns, name).
    57  			SetPodTemplateSpec(template).
    58  			AddSelector(selectorKey, selectorValue).
    59  			SetSuspend(suspend).
    60  			SetBackoffLimit(limit).
    61  			SetTTLSecondsAfterFinished(ttl).
    62  			GetObject()
    63  
    64  		Expect(job.Name).Should(Equal(name))
    65  		Expect(job.Namespace).Should(Equal(ns))
    66  		Expect(job.Spec.Template).Should(Equal(template))
    67  		Expect(job.Spec.Selector).ShouldNot(BeNil())
    68  		Expect(job.Spec.Selector.MatchLabels).ShouldNot(BeNil())
    69  		Expect(job.Spec.Selector.MatchLabels[selectorKey]).Should(Equal(selectorValue))
    70  		Expect(job.Spec.Suspend).ShouldNot(BeNil())
    71  		Expect(*job.Spec.Suspend).Should(Equal(suspend))
    72  		Expect(job.Spec.BackoffLimit).ShouldNot(BeNil())
    73  		Expect(*job.Spec.BackoffLimit).Should(Equal(limit))
    74  		Expect(job.Spec.TTLSecondsAfterFinished).ShouldNot(BeNil())
    75  		Expect(*job.Spec.TTLSecondsAfterFinished).Should(Equal(ttl))
    76  	})
    77  })