Allow test WM initialization with custom Config.

Test: N/A
Change-Id: Id4a4e86a27b86ba1f79388451ec9dfa46dafeef5

1 files changed, 15 insertions, 2 deletions

diff --git a/work/workmanager-test/src/main/java/androidx/work/test/WorkManagerTestInitHelper.java b/work/workmanager-test/src/main/java/androidx/work/test/WorkManagerTestInitHelper.java
index e1b1bf2a010..a349865f442 100644
--- a/work/workmanager-test/src/main/java/androidx/work/test/WorkManagerTestInitHelper.java
+++ b/work/workmanager-test/src/main/java/androidx/work/test/WorkManagerTestInitHelper.java
@@ -34,17 +34,30 @@ import java.util.UUID;
  */
 public final class WorkManagerTestInitHelper {
     /**
-     * Initializes {@link androidx.work.WorkManager} with a {@link SynchronousExecutor}.
+     * Initializes a test {@link androidx.work.WorkManager} with a {@link SynchronousExecutor}.
      *
      * @param context The application {@link Context}
      */
     public static void initializeTestWorkManager(@NonNull Context context) {
-        setupSynchronousTaskExecutor();
         SynchronousExecutor synchronousExecutor = new SynchronousExecutor();
         Configuration configuration = new Configuration.Builder()
                 .setExecutor(synchronousExecutor)
                 .build();
+        initializeTestWorkManager(context, configuration);
+    }
+
+    /**
+     * Initializes a test {@link androidx.work.WorkManager} with a user-specified
+     * {@link androidx.work.Configuration}.
+     *
+     * @param context The application {@link Context}
+     * @param configuration The {@link androidx.work.Configuration}
+     */
+    public static void initializeTestWorkManager(
+            @NonNull Context context,
+            @NonNull Configuration configuration) {
 
+        setupSynchronousTaskExecutor();
         final TestScheduler scheduler = new TestScheduler();
         WorkManagerImpl workManager = new TestWorkManagerImpl(context, configuration) {
             @NonNull