Fix and add example for iwftest

iwftest/example/example.go

@@ -0,0 +1,24 @@
+package example
+
+import "github.com/indeedeng/iwf-golang-sdk/iwf"
+
+func NewInitState() iwf.WorkflowState {
+	return initState{}
+}
+
+type initState struct {
+	iwf.WorkflowStateDefaults
+}
+
+const keyCustomer = "customer"
+
+func (b initState) WaitUntil(ctx iwf.WorkflowContext, input iwf.Object, persistence iwf.Persistence, communication iwf.Communication) (*iwf.CommandRequest, error) {
+	var customer string
+	input.Get(&customer)
+	persistence.SetDataAttribute(keyCustomer, customer)
+	return iwf.EmptyCommandRequest(), nil
+}
+
+func (b initState) Execute(ctx iwf.WorkflowContext, input iwf.Object, commandResults iwf.CommandResults, persistence iwf.Persistence, communication iwf.Communication) (*iwf.StateDecision, error) {
+	return iwf.GracefulCompletingWorkflow, nil
+}

iwftest/example/example_test.go

@@ -0,0 +1,46 @@
+package example
+
+import (
+	"github.com/golang/mock/gomock"
+	"github.com/indeedeng/iwf-golang-sdk/iwf"
+	"github.com/indeedeng/iwf-golang-sdk/iwftest"
+	"github.com/stretchr/testify/assert"
+	"testing"
+)
+
+var mockWfCtx *iwftest.MockWorkflowContext
+var mockPersistence *iwftest.MockPersistence
+var mockCommunication *iwftest.MockCommunication
+var emptyCmdResults = iwf.CommandResults{}
+var testCustomer = "customer1"
+var emptyObj = iwftest.NewTestObject(testCustomer)
+
+func beforeEach(t *testing.T) {
+	ctrl := gomock.NewController(t)
+
+	mockWfCtx = iwftest.NewMockWorkflowContext(ctrl)
+	mockPersistence = iwftest.NewMockPersistence(ctrl)
+	mockCommunication = iwftest.NewMockCommunication(ctrl)
+}
+
+func TestInitState_WaitUntil(t *testing.T) {
+	beforeEach(t)
+
+	state := NewInitState()
+
+	mockPersistence.EXPECT().SetDataAttribute(keyCustomer, testCustomer)
+	cmdReq, err := state.WaitUntil(mockWfCtx, emptyObj, mockPersistence, mockCommunication)
+	assert.Nil(t, err)
+	assert.Equal(t, iwf.EmptyCommandRequest(), cmdReq)
+}
+
+func TestInitState_Execute(t *testing.T) {
+	beforeEach(t)
+
+	state := NewInitState()
+	input := iwftest.NewTestObject(testCustomer)
+
+	decision, err := state.Execute(mockWfCtx, input, emptyCmdResults, mockPersistence, mockCommunication)
+	assert.Nil(t, err)
+	assert.Equal(t, iwf.GracefulCompletingWorkflow, decision)
+}