From cc63b0f28dd8c42f19baf77748b20c10e6c07a6d Mon Sep 17 00:00:00 2001
From: pawelpasterz <32893017+pawelpasterz@users.noreply.github.com>
Date: Fri, 9 Apr 2021 13:06:21 +0200
Subject: [PATCH] Update DumpShardsIT (#1790)

---
 integration_tests/src/test/kotlin/integration/DumpShardsIT.kt | 2 +-
 integration_tests/src/test/kotlin/utils/MatrixTestShards.kt   | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/integration_tests/src/test/kotlin/integration/DumpShardsIT.kt b/integration_tests/src/test/kotlin/integration/DumpShardsIT.kt
index 86cc20b460..cd08ca9235 100644
--- a/integration_tests/src/test/kotlin/integration/DumpShardsIT.kt
+++ b/integration_tests/src/test/kotlin/integration/DumpShardsIT.kt
@@ -87,7 +87,7 @@ class DumpShardsIT {
 
         assertThat(shards.count()).isEqualTo(2)
 
-        shards.first().let { firstShard ->
+        shards.first().values.flatten().let { firstShard ->
             assertThat(firstShard.count()).isEqualTo(8)
             assertThat(firstShard)
                 .contains("EarlGreyExampleSwiftTests/testWithCustomFailureHandler")
diff --git a/integration_tests/src/test/kotlin/utils/MatrixTestShards.kt b/integration_tests/src/test/kotlin/utils/MatrixTestShards.kt
index fafd443ce1..8e9c6a6ac3 100644
--- a/integration_tests/src/test/kotlin/utils/MatrixTestShards.kt
+++ b/integration_tests/src/test/kotlin/utils/MatrixTestShards.kt
@@ -12,4 +12,4 @@ data class AndroidTestShards(
     val junitIgnored: List<String> = emptyList()
 )
 
-typealias IosMatrixTestShards = List<List<String>>
+typealias IosMatrixTestShards = List<Map<String, List<String>>>