From 50d74cf44c7f666ca6c50a0072127439be5c7a81 Mon Sep 17 00:00:00 2001
From: Adam <adam.filipowicz92@gmail.com>
Date: Fri, 29 May 2020 13:37:06 +0200
Subject: [PATCH] Update AndroidArgsTest.kt

---
 .../test/kotlin/ftl/args/AndroidArgsTest.kt    | 18 +++++++++++++-----
 1 file changed, 13 insertions(+), 5 deletions(-)

diff --git a/test_runner/src/test/kotlin/ftl/args/AndroidArgsTest.kt b/test_runner/src/test/kotlin/ftl/args/AndroidArgsTest.kt
index 4091ce30e3..427517457b 100644
--- a/test_runner/src/test/kotlin/ftl/args/AndroidArgsTest.kt
+++ b/test_runner/src/test/kotlin/ftl/args/AndroidArgsTest.kt
@@ -1,16 +1,12 @@
 package ftl.args
 
-import com.google.api.services.testing.model.TestSpecification
 import com.google.common.truth.Truth.assertThat
 import ftl.args.yml.AppTestPair
-import ftl.args.yml.UploadedApks
 import ftl.cli.firebase.test.android.AndroidRunCommand
 import ftl.config.Device
 import ftl.config.FlankRoboDirective
 import ftl.config.FtlConstants.defaultAndroidModel
 import ftl.config.FtlConstants.defaultAndroidVersion
-import ftl.gc.android.setupAndroidTest
-import ftl.run.platform.android.createAndroidTestConfig
 import ftl.run.platform.runAndroidTests
 import ftl.run.platform.android.getAndroidShardChunks
 import ftl.run.status.OutputStyle
@@ -28,7 +24,6 @@ import org.junit.After
 import org.junit.Assert
 import org.junit.Assert.assertEquals
 import org.junit.Assert.assertFalse
-import org.junit.Assert.assertTrue
 import org.junit.Rule
 import org.junit.Test
 import org.junit.rules.ExpectedException
@@ -1401,6 +1396,19 @@ AndroidArgs
         val testSpecification = TestSpecification().setupAndroidTest(androidTestConfig)
         assertTrue(testSpecification.androidInstrumentationTest.testTargets.isNotEmpty())
     }
+
+    @Test
+    fun `if set max-test-shards to -1 should give maximum amount`() {
+        val yaml = """
+        gcloud:
+          app: $appApk
+          test: $testApk
+        flank:
+          max-test-shards: -1
+        """.trimIndent()
+        val args = AndroidArgs.load(yaml)
+        assertEquals(50, args.maxTestShards)
+    }
 }
 
 private fun AndroidArgs.Companion.load(yamlData: String, cli: AndroidRunCommand? = null): AndroidArgs = load(StringReader(yamlData), cli)