test: add go test and bats tests

- test: OK

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_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_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())

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}"
 }