Add register workflow methods in shadower (#603)

* Add register workflow methods in shadower

* update annotation

Author: yux0

src/main/java/com/uber/cadence/testing/WorkflowShadower.java

@@ -15,6 +15,7 @@
  */
 package com.uber.cadence.testing;
 
+import com.google.common.annotations.VisibleForTesting;
 import com.uber.cadence.internal.shadowing.ReplayWorkflowActivity;
 import com.uber.cadence.internal.shadowing.ReplayWorkflowActivityImpl;
 import com.uber.cadence.internal.shadowing.ReplayWorkflowActivityResult;
@@ -26,6 +27,8 @@
 import com.uber.cadence.serviceclient.IWorkflowService;
 import com.uber.cadence.shadower.Mode;
 import com.uber.cadence.worker.ShadowingOptions;
+import com.uber.cadence.worker.WorkflowImplementationOptions;
+import com.uber.cadence.workflow.Functions;
 import com.uber.m3.tally.NoopScope;
 import com.uber.m3.tally.Scope;
 import java.time.Duration;
@@ -53,6 +56,7 @@ public WorkflowShadower(
         new ReplayWorkflowActivityImpl(service, metricsScope, taskList));
   }
 
+  @VisibleForTesting
   public WorkflowShadower(
       ShadowingOptions options,
       ScanWorkflowActivity scanWorkflow,
@@ -114,6 +118,28 @@ public void run() throws Throwable {
     } while (nextPageToken != null && options.getShadowMode() == Mode.Normal);
   }
 
+  public void registerWorkflowImplementationTypes(Class<?>... workflowImplementationClasses) {
+    replayWorkflow.registerWorkflowImplementationTypes(workflowImplementationClasses);
+  }
+
+  public void registerWorkflowImplementationTypes(
+      WorkflowImplementationOptions options, Class<?>... workflowImplementationClasses) {
+    replayWorkflow.registerWorkflowImplementationTypesWithOptions(
+        options, workflowImplementationClasses);
+  }
+
+  public <R> void addWorkflowImplementationFactory(
+      WorkflowImplementationOptions options,
+      Class<R> workflowInterface,
+      Functions.Func<R> factory) {
+    replayWorkflow.addWorkflowImplementationFactoryWithOptions(options, workflowInterface, factory);
+  }
+
+  public <R> void addWorkflowImplementationFactory(
+      Class<R> workflowInterface, Functions.Func<R> factory) {
+    replayWorkflow.addWorkflowImplementationFactory(workflowInterface, factory);
+  }
+
   private ShadowingOptions validateShadowingOptions(ShadowingOptions options) {
     Objects.requireNonNull(options);