Skip to content

Commit

Permalink
mk only
Browse files Browse the repository at this point in the history
  • Loading branch information
mszadkow committed Jan 3, 2025
1 parent dc225b3 commit e9a02cc
Show file tree
Hide file tree
Showing 2 changed files with 14 additions and 4 deletions.
4 changes: 2 additions & 2 deletions Makefile-test.mk
Original file line number Diff line number Diff line change
Expand Up @@ -39,10 +39,10 @@ ENVTEST_K8S_VERSION ?= 1.31
# to run without parallelism.
INTEGRATION_NPROCS ?= 4
# Folder where the integration tests are located.
INTEGRATION_TARGET ?= ./test/integration/...
INTEGRATION_TARGET ?= ./test/integration/multikueue/...
# Verbosity level for apiserver logging.
# The logging is disabled if 0.
INTEGRATION_API_LOG_LEVEL ?= 5
INTEGRATION_API_LOG_LEVEL ?= 1
# Integration filters
INTEGRATION_RUN_ALL?=true
ifneq ($(INTEGRATION_RUN_ALL),true)
Expand Down
14 changes: 12 additions & 2 deletions test/integration/multikueue/suite_test.go
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,7 @@ import (
workloadtfjob "sigs.k8s.io/kueue/pkg/controller/jobs/kubeflow/jobs/tfjob"
workloadxgboostjob "sigs.k8s.io/kueue/pkg/controller/jobs/kubeflow/jobs/xgboostjob"
workloadmpijob "sigs.k8s.io/kueue/pkg/controller/jobs/mpijob"
"sigs.k8s.io/kueue/pkg/controller/jobs/rayjob"
"sigs.k8s.io/kueue/pkg/queue"
"sigs.k8s.io/kueue/pkg/util/kubeversion"
utiltesting "sigs.k8s.io/kueue/pkg/util/testing"
Expand Down Expand Up @@ -210,9 +211,18 @@ func managerSetup(ctx context.Context, mgr manager.Manager) {
err = workloadmpijob.SetupMPIJobWebhook(mgr, jobframework.WithCache(cCache), jobframework.WithQueues(queues))
gomega.Expect(err).NotTo(gomega.HaveOccurred())

reconciler := rayjob.NewReconciler(
mgr.GetClient(),
mgr.GetEventRecorderFor(constants.JobControllerName))
err = rayjob.SetupIndexes(ctx, mgr.GetFieldIndexer())
gomega.Expect(err).NotTo(gomega.HaveOccurred())
err = reconciler.SetupWithManager(mgr)
gomega.Expect(err).NotTo(gomega.HaveOccurred())
err = rayjob.SetupRayJobWebhook(mgr)
gomega.Expect(err).NotTo(gomega.HaveOccurred())

names := jobframework.GetIntegrationsList()
enabledNames := jobframework.GetEnabledIntegrationsList()
fmt.Println("KROWA", names, "KROWA2", enabledNames)
fmt.Println("KROWA", names)
}

func managerAndMultiKueueSetup(ctx context.Context, mgr manager.Manager, gcInterval time.Duration, enabledIntegrations sets.Set[string]) {
Expand Down

0 comments on commit e9a02cc

Please sign in to comment.