Skip to content

Commit

Permalink
[YUNIKORN-2247] Preemption e2e tests should run in separate namespaces (
Browse files Browse the repository at this point in the history
apache#753)

Closes: apache#753

Signed-off-by: Peter Bacsko <[email protected]>
  • Loading branch information
brandboat authored and pbacsko committed Dec 26, 2023
1 parent 24c699f commit 710f7bf
Showing 1 changed file with 14 additions and 15 deletions.
29 changes: 14 additions & 15 deletions test/e2e/preemption/preemption_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -40,7 +40,7 @@ import (
var kClient k8s.KubeCtl
var restClient yunikorn.RClient
var ns *v1.Namespace
var dev = "dev" + common.RandSeq(5)
var dev string
var oldConfigMap = new(v1.ConfigMap)
var annotation = "ann-" + common.RandSeq(10)

Expand All @@ -66,11 +66,6 @@ var _ = ginkgo.BeforeSuite(func() {
var err = kClient.PortForwardYkSchedulerPod()
Ω(err).NotTo(gomega.HaveOccurred())

ginkgo.By("create development namespace")
ns, err = kClient.CreateNamespace(dev, nil)
gomega.Ω(err).NotTo(gomega.HaveOccurred())
gomega.Ω(ns.Status.Phase).To(gomega.Equal(v1.NamespaceActive))

var nodes *v1.NodeList
nodes, err = kClient.GetNodes()
Ω(err).NotTo(gomega.HaveOccurred())
Expand Down Expand Up @@ -126,12 +121,18 @@ var _ = ginkgo.AfterSuite(func() {
checks, err := yunikorn.GetFailedHealthChecks()
Ω(err).NotTo(gomega.HaveOccurred())
Ω(checks).To(gomega.Equal(""), checks)
ginkgo.By("Tearing down namespace: " + ns.Name)
err = kClient.TearDownNamespace(ns.Name)
Ω(err).NotTo(gomega.HaveOccurred())
})

var _ = ginkgo.Describe("Preemption", func() {
ginkgo.BeforeEach(func() {
dev = "dev-" + common.RandSeq(5)
ginkgo.By("Creating development namespace: " + dev)
var err error
ns, err = kClient.CreateNamespace(dev, nil)
Ω(err).NotTo(HaveOccurred())
Ω(ns.Status.Phase).To(gomega.Equal(v1.NamespaceActive))
})

ginkgo.It("Verify_basic_preemption", func() {
ginkgo.By("A queue uses resource more than the guaranteed value even after removing one of the pods. The cluster doesn't have enough resource to deploy a pod in another queue which uses resource less than the guaranteed value.")
// update config
Expand Down Expand Up @@ -552,12 +553,10 @@ var _ = ginkgo.Describe("Preemption", func() {
tests.LogTestClusterInfoWrapper(testDescription.FailureMessage(), []string{ns.Name})
tests.LogYunikornContainer(testDescription.FailureMessage())
}
// Delete all sleep pods
ginkgo.By("Delete all sleep pods")
err := kClient.DeletePods(ns.Name)
if err != nil {
fmt.Fprintf(ginkgo.GinkgoWriter, "Failed to delete pods in namespace %s - reason is %s\n", ns.Name, err.Error())
}

ginkgo.By("Tear down namespace: " + dev)
err := kClient.TearDownNamespace(dev)
Ω(err).NotTo(HaveOccurred())

// reset config
ginkgo.By("Restoring YuniKorn configuration")
Expand Down

0 comments on commit 710f7bf

Please sign in to comment.