From 03d4b249a46597ad7a131081a421ea0260955345 Mon Sep 17 00:00:00 2001
From: Omar Ismail <omarismail@google.com>
Date: Wed, 26 Jun 2024 18:21:13 +0100
Subject: [PATCH] Update TestRunnerServiceImplTest.kt to use SmarSharding

---
 .../androidx/ci/testRunner/TestRunnerServiceImplTest.kt    | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/AndroidXCI/lib/src/test/kotlin/dev/androidx/ci/testRunner/TestRunnerServiceImplTest.kt b/AndroidXCI/lib/src/test/kotlin/dev/androidx/ci/testRunner/TestRunnerServiceImplTest.kt
index 2d75ed4..68298aa 100644
--- a/AndroidXCI/lib/src/test/kotlin/dev/androidx/ci/testRunner/TestRunnerServiceImplTest.kt
+++ b/AndroidXCI/lib/src/test/kotlin/dev/androidx/ci/testRunner/TestRunnerServiceImplTest.kt
@@ -10,11 +10,11 @@ import dev.androidx.ci.generated.ftl.EnvironmentVariable
 import dev.androidx.ci.generated.ftl.GoogleCloudStorage
 import dev.androidx.ci.generated.ftl.ResultStorage
 import dev.androidx.ci.generated.ftl.ShardingOption
+import dev.androidx.ci.generated.ftl.SmartSharding
 import dev.androidx.ci.generated.ftl.TestEnvironmentCatalog
 import dev.androidx.ci.generated.ftl.TestMatrix
 import dev.androidx.ci.generated.ftl.TestSpecification
 import dev.androidx.ci.generated.ftl.ToolResultsExecution
-import dev.androidx.ci.generated.ftl.UniformSharding
 import dev.androidx.ci.generated.testResults.FileReference
 import dev.androidx.ci.generated.testResults.StackTrace
 import dev.androidx.ci.generated.testResults.Step
@@ -244,9 +244,8 @@ class TestRunnerServiceImplTest {
         ).isEqualTo(clientInfo)
 
         // shard
-        val shardingOption = ShardingOption(
-            uniformSharding = UniformSharding(3)
-        )
+        val shardingOption = ShardingOption(smartSharding = SmartSharding(targetedShardDuration = "600s"))
+
         val shardedTest = subject.scheduleTests(
             testApk = upload1,
             appApk = null,