Make MaestroCommand toString more concise (#2011)

mobile-dev-inc · Sep 3, 2024 · 022840f · 022840f
1 parent c95fe9d
commit 022840f
Show file tree

Hide file tree

Showing 2 changed files with 31 additions and 0 deletions.
diff --git a/maestro-orchestra-models/src/main/java/maestro/orchestra/MaestroCommand.kt b/maestro-orchestra-models/src/main/java/maestro/orchestra/MaestroCommand.kt
@@ -163,4 +163,10 @@ data class MaestroCommand(
     fun description(): String {
         return asCommand()?.description() ?: "No op"
     }
+
+    override fun toString(): String =
+        asCommand()?.let { command ->
+            val argName = command::class.simpleName?.replaceFirstChar(Char::lowercaseChar) ?: "command"
+            "MaestroCommand($argName=$command)"
+        } ?: "MaestroCommand()"
 }
diff --git a/maestro-orchestra/src/test/java/maestro/orchestra/MaestroCommandTest.kt b/maestro-orchestra/src/test/java/maestro/orchestra/MaestroCommandTest.kt
@@ -55,4 +55,29 @@ internal class MaestroCommandTest {
         assertThat(description)
             .isEqualTo("Set location with negative coordinates")
     }
+
+    @Test
+    fun `toString (no commands)`() {
+        // given
+        val maestroCommand = MaestroCommand(null)
+
+        // when
+        val toString = maestroCommand.toString()
+
+        // then
+        assertThat(toString).isEqualTo("MaestroCommand()")
+    }
+
+    @Test
+    fun `toString (at least one command)`() {
+        // given
+        val command = BackPressCommand()
+        val maestroCommand = MaestroCommand(command)
+
+        // when
+        val toString = maestroCommand.toString()
+
+        // then
+        assertThat(toString).isEqualTo("MaestroCommand(backPressCommand=$command)")
+    }
 }