Refactor

Flank · Jun 8, 2021 · d4fea83 · d4fea83
1 parent ff41e13
commit d4fea83
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/test_runner/src/main/kotlin/ftl/run/model/AndroidTestContext.kt b/test_runner/src/main/kotlin/ftl/run/model/AndroidTestContext.kt
@@ -6,11 +6,7 @@ import ftl.args.AndroidArgs
 import ftl.args.IgnoredTestCases
 import ftl.shard.Chunk
 
-interface WithArgs {
-    val args: AndroidArgs
-}
-
-sealed class AndroidTestContext : WithArgs
+sealed class AndroidTestContext(open val args: AndroidArgs)
 
 data class InstrumentationTestContext(
     val app: FileReference,
@@ -20,22 +16,22 @@ data class InstrumentationTestContext(
     val environmentVariables: Map<String, String> = emptyMap(),
     val testTargetsForShard: ShardChunks = emptyList(),
     override val args: AndroidArgs
-) : AndroidTestContext()
+) : AndroidTestContext(args)
 
 data class RoboTestContext(
     val app: FileReference,
     val roboScript: FileReference,
     override val args: AndroidArgs
-) : AndroidTestContext()
+) : AndroidTestContext(args)
 
 data class GameLoopContext(
     val app: FileReference,
     val scenarioLabels: List<String>,
     val scenarioNumbers: List<String>,
     override val args: AndroidArgs
-) : AndroidTestContext()
+) : AndroidTestContext(args)
 
 data class SanityRoboTestContext(
     val app: FileReference,
     override val args: AndroidArgs
-) : AndroidTestContext()
+) : AndroidTestContext(args)