fixup! addressed comments.

apache · Jan 24, 2017 · fb54273 · fb54273
1 parent 974975e
commit fb54273
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/sdks/java/core/src/main/java/org/apache/beam/sdk/io/FileSystems.java b/sdks/java/core/src/main/java/org/apache/beam/sdk/io/FileSystems.java
@@ -92,10 +92,12 @@ public static void setDefaultConfigInWorkers(PipelineOptions options) {
    */
   @VisibleForTesting
   static FileSystem getFileSystemInternal(URI uri) {
+    checkNotNull(
+        defaultConfig,
+        "Expect the runner have called setDefaultConfigInWorkers().");
     String lowerCaseScheme = (uri.getScheme() != null
         ? uri.getScheme().toLowerCase() : LocalFileSystemRegistrar.LOCAL_FILE_SCHEME);
-    return getRegistrarInternal(lowerCaseScheme)
-        .fromOptions(checkNotNull(defaultConfig, "defaultConfig"));
+    return getRegistrarInternal(lowerCaseScheme).fromOptions(defaultConfig);
   }
 
   /**