new: add --scheduler-name to build and buildrun

- test: add go test and bats tests

Signed-off-by: rxinui <[email protected]>

Author: rxinui

docs/shp_build_create.md

@@ -30,6 +30,7 @@ shp build create <name> [flags]
       --retention-succeeded-limit uint           number of succeeded BuildRuns to be kept (default 65535)
       --retention-ttl-after-failed duration      duration to delete a failed BuildRun after completion
       --retention-ttl-after-succeeded duration   duration to delete a succeeded BuildRun after completion
+      --scheduler-name string                    specify the scheduler to be used to dispatch the Pod
       --source-context-dir string                use a inner directory as context directory
       --source-git-clone-secret string           name of the secret with credentials to access the git source, e.g. git credentials
       --source-git-revision string               git repository source revision

docs/shp_build_run.md

@@ -32,6 +32,7 @@ shp build run <name> [flags]
       --retention-ttl-after-failed duration      duration to delete the BuildRun after it failed
       --retention-ttl-after-succeeded duration   duration to delete the BuildRun after it succeeded
       --sa-name string                           Kubernetes service-account name
+      --scheduler-name string                    specify the scheduler to be used to dispatch the Pod
       --timeout duration                         build process timeout
 ```
 

docs/shp_build_upload.md

@@ -42,6 +42,7 @@ shp build upload <build-name> [path/to/source|.] [flags]
       --retention-ttl-after-failed duration      duration to delete the BuildRun after it failed
       --retention-ttl-after-succeeded duration   duration to delete the BuildRun after it succeeded
       --sa-name string                           Kubernetes service-account name
+      --scheduler-name string                    specify the scheduler to be used to dispatch the Pod
       --timeout duration                         build process timeout
 ```
 

docs/shp_buildrun_create.md

@@ -31,6 +31,7 @@ shp buildrun create <name> [flags]
       --retention-ttl-after-failed duration      duration to delete the BuildRun after it failed
       --retention-ttl-after-succeeded duration   duration to delete the BuildRun after it succeeded
       --sa-name string                           Kubernetes service-account name
+      --scheduler-name string                    specify the scheduler to be used to dispatch the Pod
       --timeout duration                         build process timeout
 ```
 

pkg/shp/flags/build.go

@@ -44,7 +44,8 @@ func BuildSpecFromFlags(flags *pflag.FlagSet) (*buildv1beta1.BuildSpec, *string,
 			TTLAfterFailed:    &metav1.Duration{},
 			TTLAfterSucceeded: &metav1.Duration{},
 		},
-		NodeSelector: map[string]string{},
+		NodeSelector:  map[string]string{},
+		SchedulerName: ptr.To(""),
 	}
 
 	sourceFlags(flags, spec.Source)
@@ -57,6 +58,7 @@ func BuildSpecFromFlags(flags *pflag.FlagSet) (*buildv1beta1.BuildSpec, *string,
 	imageAnnotationsFlags(flags, spec.Output.Annotations)
 	buildRetentionFlags(flags, spec.Retention)
 	buildNodeSelectorFlags(flags, spec.NodeSelector)
+	buildSchedulerNameFlag(flags, spec.SchedulerName)
 	var dockerfile, builderImage string
 	dockerfileFlags(flags, &dockerfile)
 	builderImageFlag(flags, &builderImage)

pkg/shp/flags/build_test.go

@@ -55,7 +55,8 @@ func TestBuildSpecFromFlags(t *testing.T) {
 				Duration: 30 * time.Minute,
 			},
 		},
-		NodeSelector: map[string]string{"kubernetes.io/hostname": "worker-1"},
+		NodeSelector:  map[string]string{"kubernetes.io/hostname": "worker-1"},
+		SchedulerName: ptr.To("dolphinscheduler"),
 	}
 
 	cmd := &cobra.Command{}
@@ -123,6 +124,13 @@ func TestBuildSpecFromFlags(t *testing.T) {
 		g.Expect(expected.NodeSelector).To(o.Equal(spec.NodeSelector), ".spec.nodeSelector")
 	})
 
+	t.Run(".spec.schedulerName", func(_ *testing.T) {
+		err := flags.Set(SchedulerNameFlag, *expected.SchedulerName)
+		g.Expect(err).To(o.BeNil())
+
+		g.Expect(expected.SchedulerName).To(o.Equal(spec.SchedulerName), "spec.schedulerName")
+	})
+
 	t.Run(".spec.timeout", func(_ *testing.T) {
 		err := flags.Set(TimeoutFlag, expected.Timeout.Duration.String())
 		g.Expect(err).To(o.BeNil())

pkg/shp/flags/buildrun.go

@@ -28,7 +28,8 @@ func BuildRunSpecFromFlags(flags *pflag.FlagSet) *buildv1beta1.BuildRunSpec {
 			TTLAfterFailed:    &metav1.Duration{},
 			TTLAfterSucceeded: &metav1.Duration{},
 		},
-		NodeSelector: map[string]string{},
+		NodeSelector:  map[string]string{},
+		SchedulerName: ptr.To(""),
 	}
 
 	buildRefFlags(flags, &spec.Build)
@@ -41,6 +42,7 @@ func BuildRunSpecFromFlags(flags *pflag.FlagSet) *buildv1beta1.BuildRunSpec {
 	imageAnnotationsFlags(flags, spec.Output.Annotations)
 	buildRunRetentionFlags(flags, spec.Retention)
 	buildNodeSelectorFlags(flags, spec.NodeSelector)
+	buildSchedulerNameFlag(flags, spec.SchedulerName)
 	return spec
 }
 

pkg/shp/flags/buildrun_test.go

@@ -38,7 +38,8 @@ func TestBuildRunSpecFromFlags(t *testing.T) {
 				Duration: 30 * time.Minute,
 			},
 		},
-		NodeSelector: map[string]string{"kubernetes.io/hostname": "worker-1"},
+		NodeSelector:  map[string]string{"kubernetes.io/hostname": "worker-1"},
+		SchedulerName: ptr.To("dolphinscheduler"),
 	}
 
 	cmd := &cobra.Command{}
@@ -83,6 +84,13 @@ func TestBuildRunSpecFromFlags(t *testing.T) {
 		g.Expect(expected.NodeSelector).To(o.Equal(spec.NodeSelector), ".spec.nodeSelector")
 	})
 
+	t.Run(".spec.schedulerName", func(_ *testing.T) {
+		err := flags.Set(SchedulerNameFlag, *expected.SchedulerName)
+		g.Expect(err).To(o.BeNil())
+
+		g.Expect(expected.SchedulerName).To(o.Equal(spec.SchedulerName), "spec.schedulerName")
+	})
+
 	t.Run(".spec.retention.ttlAfterFailed", func(_ *testing.T) {
 		err := flags.Set(RetentionTTLAfterFailedFlag, expected.Retention.TTLAfterFailed.Duration.String())
 		g.Expect(err).To(o.BeNil())

pkg/shp/flags/flags.go

@@ -66,6 +66,8 @@ const (
 	RetentionTTLAfterSucceededFlag = "retention-ttl-after-succeeded"
 	// NodeSelectorFlag command-line flag.
 	NodeSelectorFlag = "node-selector"
+	// SchedulerNameFlag command-line flag.
+	SchedulerNameFlag = "scheduler-name"
 )
 
 // sourceFlags flags for ".spec.source"
@@ -266,6 +268,16 @@ func buildNodeSelectorFlags(flags *pflag.FlagSet, nodeSelectorLabels map[string]
 	flags.Var(NewMapValue(nodeSelectorLabels), NodeSelectorFlag, "set of key-value pairs that correspond to labels of a node to match")
 }
 
+// buildSchedulerNameFlag registers flags for adding BuildSpec.SchedulerName
+func buildSchedulerNameFlag(flags *pflag.FlagSet, schedulerName *string) {
+	flags.StringVar(
+		schedulerName,
+		SchedulerNameFlag,
+		"",
+		"specify the scheduler to be used to dispatch the Pod",
+	)
+}
+
 // envFlags registers flags for adding corev1.EnvVars.
 func envFlags(flags *pflag.FlagSet, envs *[]corev1.EnvVar) {
 	flags.VarP(

test/e2e/node-selector.bats

@@ -13,6 +13,8 @@ teardown() {
 	run kubectl delete buildruns.shipwright.io --all
 }
 
+scheduler_name="dolphinscheduler"
+
 @test "shp build create --node-selector single label" {
     # generate random names for our build
 	build_name=$(random_name)
@@ -50,6 +52,24 @@ teardown() {
     assert_output --partial '"kubernetes.io/os":"linux"'
 }
 
+@test "shp build create --scheduler-name" {
+    # generate random names for our build
+	build_name=$(random_name)
+
+    # create a Build with node selector 
+    run shp build create ${build_name} --source-git-url=https://github.com/shipwright-io/sample-go --output-image=my-fake-image --scheduler-name=${scheduler_name}
+    assert_success
+
+    # ensure that the build was successfully created
+	assert_output --partial "Created build \"${build_name}\""
+
+    # get the jsonpath of Build object .spec.nodeSelector
+	run kubectl get builds.shipwright.io/${build_name} -ojsonpath="{.spec.schedulerName}"
+	assert_success
+
+    assert_output "${scheduler_name}"
+}
+
 @test "shp buildrun create --node-selector single label" {
     # generate random names for our buildrun
 	buildrun_name=$(random_name)
@@ -89,6 +109,25 @@ teardown() {
     assert_output --partial '"kubernetes.io/os":"linux"'
 }
 
+@test "shp buildrun create --scheduler-name" {
+    # generate random names for our buildrun
+	buildrun_name=$(random_name)
+	build_name=$(random_name)
+
+    # create a Build with node selector 
+    run shp buildrun create ${buildrun_name} --buildref-name=${build_name} --scheduler-name=${scheduler_name}
+    assert_success
+
+    # ensure that the build was successfully created
+	assert_output --partial "BuildRun created \"${buildrun_name}\" for Build \"${build_name}\""
+
+    # get the jsonpath of Build object .spec.nodeSelector
+	run kubectl get buildruns.shipwright.io/${buildrun_name} -ojsonpath="{.spec.schedulerName}"
+	assert_success
+
+    assert_output "${scheduler_name}"
+}
+
 
 @test "shp build run --node-selector set" {
     # generate random names for our build
@@ -111,4 +150,27 @@ teardown() {
 	run kubectl get buildruns.shipwright.io -ojsonpath='{.items[*].spec.nodeSelector}' 
 	assert_success
     assert_output --partial '"kubernetes.io/hostname":"node-1"'
+}
+
+@test "shp build run --scheduler-name" {
+    # generate random names for our build
+	build_name=$(random_name)
+
+    # create a Build with node selector 
+    run shp build create ${build_name} --source-git-url=https://github.com/shipwright-io/sample-go --output-image=my-fake-image
+    assert_success
+
+    # ensure that the build was successfully created
+	assert_output --partial "Created build \"${build_name}\""
+
+    # get the build object
+	run kubectl get builds.shipwright.io/${build_name}
+	assert_success
+
+    run shp build run ${build_name} --scheduler-name=${scheduler_name}
+
+    # get the jsonpath of Build object .spec.nodeSelector
+	run kubectl get buildruns.shipwright.io -ojsonpath='{.items[*].spec.schedulerName}' 
+	assert_success
+    assert_output --partial "${scheduler_name}"
 }