Refactor

Flank · Apr 8, 2021 · d7437c4 · d7437c4
1 parent 05288e9
commit d7437c4
Showing 1 changed file with 6 additions and 9 deletions.
diff --git a/test_runner/src/main/kotlin/ftl/ios/xctest/XcTestData.kt b/test_runner/src/main/kotlin/ftl/ios/xctest/XcTestData.kt
@@ -64,17 +64,14 @@ private fun IosArgs.useCustomShardingV1(dictionary: NSDictionary) =
 private fun IosArgs.useCustomShardingV2(dictionary: NSDictionary) =
     customShardingJson.isNotBlank() && dictionary.getXcTestRunVersion() == V2
 
-private fun IosArgs.shardsFromV1() =
-    createCustomSharding<List<XctestrunMethods>>(commonArgs.customShardingJson)
+private fun IosArgs.shardsFromV1() = mapOf(
+    "" to createCustomSharding<List<XctestrunMethods>>(commonArgs.customShardingJson)
         .run {
-            mapOf(
-                "" to run {
-                    map { xcMethods ->
-                        Chunk(xcMethods.values.flatMap { it.map(::TestMethod) })
-                    } to this
-                }
-            )
+            map { xcMethods ->
+                Chunk(xcMethods.values.flatMap { it.map(::TestMethod) })
+            } to this
         }
+)
 
 private fun IosArgs.shardsFromV2() =
     createCustomSharding<Map<String, List<XctestrunMethods>>>(commonArgs.customShardingJson)