84 lines
1.4 KiB
Go
84 lines
1.4 KiB
Go
package common
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/stretchr/testify/assert"
|
|
)
|
|
|
|
func TestNewWorkflow(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
// empty
|
|
emptyWorkflow := NewPipelineExecutor()
|
|
assert.Nil(emptyWorkflow())
|
|
|
|
// error case
|
|
errorWorkflow := NewErrorExecutor(fmt.Errorf("test error"))
|
|
assert.NotNil(errorWorkflow())
|
|
|
|
// multiple success case
|
|
runcount := 0
|
|
successWorkflow := NewPipelineExecutor(
|
|
func() error {
|
|
runcount ++
|
|
return nil
|
|
},
|
|
func() error {
|
|
runcount ++
|
|
return nil
|
|
})
|
|
assert.Nil(successWorkflow())
|
|
assert.Equal(2, runcount)
|
|
}
|
|
|
|
func TestNewConditionalExecutor(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
trueCount := 0
|
|
falseCount := 0
|
|
|
|
err := NewConditionalExecutor(func() bool {
|
|
return false
|
|
}, func() error {
|
|
trueCount++
|
|
return nil
|
|
}, func() error {
|
|
falseCount++
|
|
return nil
|
|
})()
|
|
|
|
assert.Nil(err)
|
|
assert.Equal(0, trueCount)
|
|
assert.Equal(1, falseCount)
|
|
|
|
err = NewConditionalExecutor(func() bool {
|
|
return true
|
|
}, func() error {
|
|
trueCount++
|
|
return nil
|
|
}, func() error {
|
|
falseCount++
|
|
return nil
|
|
})()
|
|
|
|
assert.Nil(err)
|
|
assert.Equal(1, trueCount)
|
|
assert.Equal(1, falseCount)
|
|
}
|
|
|
|
func TestNewParallelExecutor(t *testing.T) {
|
|
assert := assert.New(t)
|
|
|
|
count := 0
|
|
emptyWorkflow := NewPipelineExecutor(func() error {
|
|
count++
|
|
return nil
|
|
})
|
|
|
|
err := NewParallelExecutor(emptyWorkflow, emptyWorkflow)()
|
|
assert.Equal(2, count)
|
|
|
|
assert.Nil(err)
|
|
}
|