Add workflow test for Contacts.saveContatcs

examples/example-gateway/endpoints/contacts/save_contacts_test.go

@@ -2,10 +2,12 @@ package contacts_test
 
 import (
 	"bytes"
+	"context"
 	"testing"
 
 	"github.com/stretchr/testify/assert"
 
+	"github.com/uber/zanzibar/examples/example-gateway/build/endpoints/contacts/mock-workflow"
 	endpointContacts "github.com/uber/zanzibar/examples/example-gateway/build/gen-code/endpoints/contacts/contacts"
 	ms "github.com/uber/zanzibar/examples/example-gateway/build/services/example-gateway/mock-service"
 )
@@ -31,3 +33,22 @@ func TestSaveContactsCall(t *testing.T) {
 
 	assert.Equal(t, "202 Accepted", res.Status)
 }
+
+func TestSaveContactsCallWorkflow(t *testing.T) {
+	mh, mc := mockcontactsworkflow.NewContactsSaveContactsWorkflowMock(t)
+
+	mc.Contacts.ExpectSaveContacts().Success()
+
+	endpointReqeust := &endpointContacts.SaveContactsRequest{
+		UserUUID: "foo",
+		Contacts: []*endpointContacts.Contact{},
+	}
+
+	res, resHeaders, err := mh.Handle(context.Background(), nil, endpointReqeust)
+
+	if !assert.NoError(t, err, "got error") {
+		return
+	}
+	assert.Nil(t, resHeaders)
+	assert.Equal(t, &endpointContacts.SaveContactsResponse{}, res)
+}