Merge pull request #638 from uber/options_to_create_mocks

Add option to select which mocks to generate. This helps to specify which mocks to create when running postGenhooks (Client mock, service Mock, workflow Mock)

codegen/module_system.go

@@ -230,20 +230,30 @@ type ClientTestFixture struct {
 // module system (clients, endpoints, services) with a post build hook to generate client and service mocks
 func NewDefaultModuleSystemWithMockHook(
 	h *PackageHelper,
+	clientsMock bool,
+	workflowMock bool,
+	serviceMock bool,
 	hooks ...PostGenHook,
 ) (*ModuleSystem, error) {
 	t, err := NewDefaultTemplate()
 	if err != nil {
 		return nil, err
 	}
 
-	clientMockGenHook, err := ClientMockGenHook(h, t)
-	if err != nil {
-		return nil, errors.Wrap(err, "error creating client mock gen hook")
+	var clientMockGenHook, workflowMockGenHook, serviceMockGenHook PostGenHook
+	if clientsMock {
+		clientMockGenHook, err = ClientMockGenHook(h, t)
+		if err != nil {
+			return nil, errors.Wrap(err, "error creating client mock gen hook")
+		}
 	}
 
-	workflowMockGenHook := WorkflowMockGenHook(h, t)
-	serviceMockGenHook := ServiceMockGenHook(h, t)
+	if workflowMock {
+		workflowMockGenHook = WorkflowMockGenHook(h, t)
+	}
+	if serviceMock {
+		serviceMockGenHook = ServiceMockGenHook(h, t)
+	}
 
 	allHooks := append([]PostGenHook{
 		clientMockGenHook,

codegen/runner/runner.go

@@ -142,7 +142,7 @@ func main() {
 	}
 	var moduleSystem *codegen.ModuleSystem
 	if genMock {
-		moduleSystem, err = codegen.NewDefaultModuleSystemWithMockHook(packageHelper)
+		moduleSystem, err = codegen.NewDefaultModuleSystemWithMockHook(packageHelper, true, true, true)
 	} else {
 		moduleSystem, err = codegen.NewDefaultModuleSystem(packageHelper)
 	}