operator/pkg/workflow: test job
In this commit, we unit test the job workflow in the operator package on initializing the running data runtime interface and running tasks and subtasks recursively. Signed-off-by: Mohamed Awnallah <mohamedmohey2352@gmail.com>
This commit is contained in:
parent
8cc712f698
commit
11bd90ecf8
|
@ -0,0 +1,163 @@
|
||||||
|
/*
|
||||||
|
Copyright 2024 The Karmada Authors.
|
||||||
|
|
||||||
|
Licensed under the Apache License, Version 2.0 (the "License");
|
||||||
|
you may not use this file except in compliance with the License.
|
||||||
|
You may obtain a copy of the License at
|
||||||
|
|
||||||
|
http://www.apache.org/licenses/LICENSE-2.0
|
||||||
|
|
||||||
|
Unless required by applicable law or agreed to in writing, software
|
||||||
|
distributed under the License is distributed on an "AS IS" BASIS,
|
||||||
|
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||||
|
See the License for the specific language governing permissions and
|
||||||
|
limitations under the License.
|
||||||
|
*/
|
||||||
|
|
||||||
|
package workflow
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"reflect"
|
||||||
|
"strings"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
type TestRunData struct {
|
||||||
|
name string
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestInitData(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
job *Job
|
||||||
|
wantRunData RunData
|
||||||
|
wantErr bool
|
||||||
|
errMsg string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "InitData_NoRunDataIsInitialized_InitializeRunDataWithError",
|
||||||
|
job: &Job{
|
||||||
|
runDataInitializer: func() (RunData, error) {
|
||||||
|
return nil, fmt.Errorf("failed to initialize run data")
|
||||||
|
},
|
||||||
|
runData: nil,
|
||||||
|
},
|
||||||
|
wantRunData: nil,
|
||||||
|
wantErr: true,
|
||||||
|
errMsg: "failed to initialize run data",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "InitData_NoRunDataIsInitialized_InitializeRunData",
|
||||||
|
job: &Job{
|
||||||
|
runDataInitializer: func() (RunData, error) {
|
||||||
|
return &TestRunData{name: "test"}, nil
|
||||||
|
},
|
||||||
|
runData: nil,
|
||||||
|
},
|
||||||
|
wantRunData: &TestRunData{name: "test"},
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "InitData_RunDataIsAlreadyInitialized_InitializeRunData",
|
||||||
|
job: &Job{
|
||||||
|
runDataInitializer: func() (RunData, error) {
|
||||||
|
return &TestRunData{name: "test"}, nil
|
||||||
|
},
|
||||||
|
runData: &TestRunData{name: "already-exist"},
|
||||||
|
},
|
||||||
|
wantRunData: &TestRunData{name: "already-exist"},
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
|
runData, err := test.job.initData()
|
||||||
|
if err == nil && test.wantErr {
|
||||||
|
t.Errorf("expected an error, but got none")
|
||||||
|
}
|
||||||
|
if err != nil && !test.wantErr {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if err != nil && test.wantErr && !strings.Contains(err.Error(), test.errMsg) {
|
||||||
|
t.Errorf("expected error msg %s to be part of %s", test.errMsg, err.Error())
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(runData, test.wantRunData) {
|
||||||
|
t.Errorf("unmatched run data equality, expected %v but got %v", test.wantRunData, runData)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestRun(t *testing.T) {
|
||||||
|
tests := []struct {
|
||||||
|
name string
|
||||||
|
job *Job
|
||||||
|
wantRunData RunData
|
||||||
|
wantErr bool
|
||||||
|
errMsg string
|
||||||
|
}{
|
||||||
|
{
|
||||||
|
name: "Run_NilRunData_RunDataIsInitialized",
|
||||||
|
job: &Job{
|
||||||
|
runDataInitializer: func() (RunData, error) {
|
||||||
|
return nil, fmt.Errorf("failed to initialize run data")
|
||||||
|
},
|
||||||
|
runData: nil,
|
||||||
|
},
|
||||||
|
wantErr: true,
|
||||||
|
errMsg: "failed to initialize run data",
|
||||||
|
},
|
||||||
|
{
|
||||||
|
name: "Run_Task_TaskRunSuccessfully",
|
||||||
|
job: &Job{
|
||||||
|
runDataInitializer: func() (RunData, error) {
|
||||||
|
return &TestRunData{name: "test"}, nil
|
||||||
|
},
|
||||||
|
runData: nil,
|
||||||
|
Tasks: []Task{
|
||||||
|
{
|
||||||
|
Name: "SkipRunningTask",
|
||||||
|
Skip: func(RunData) (bool, error) {
|
||||||
|
return true, nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
{
|
||||||
|
Name: "RunSubTask",
|
||||||
|
Run: func(RunData) error {
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
Tasks: []Task{
|
||||||
|
{
|
||||||
|
Name: "RunSubTask_2",
|
||||||
|
Run: func(RunData) error {
|
||||||
|
return nil
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
RunSubTasks: true,
|
||||||
|
},
|
||||||
|
},
|
||||||
|
},
|
||||||
|
wantRunData: &TestRunData{name: "test"},
|
||||||
|
wantErr: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
for _, test := range tests {
|
||||||
|
t.Run(test.name, func(t *testing.T) {
|
||||||
|
err := test.job.Run()
|
||||||
|
if err == nil && test.wantErr {
|
||||||
|
t.Errorf("expected an error, but got none")
|
||||||
|
}
|
||||||
|
if err != nil && !test.wantErr {
|
||||||
|
t.Errorf("unexpected error: %v", err)
|
||||||
|
}
|
||||||
|
if err != nil && test.wantErr && !strings.Contains(err.Error(), test.errMsg) {
|
||||||
|
t.Errorf("expected error msg %s to be part of %s", test.errMsg, err.Error())
|
||||||
|
}
|
||||||
|
if !reflect.DeepEqual(test.job.runData, test.wantRunData) {
|
||||||
|
t.Errorf("unmatched run data equality, expected %v but got %v", test.wantRunData, test.job.runData)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}
|
||||||
|
}
|
Loading…
Reference in New Issue