Rebase tweaks

mobile-dev-inc · Dec 31, 2024 · 2dff691 · 2dff691
1 parent a2aab8e
commit 2dff691
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/maestro-client/src/main/java/maestro/drivers/IOSDriver.kt b/maestro-client/src/main/java/maestro/drivers/IOSDriver.kt
@@ -432,8 +432,8 @@ class IOSDriver(
     }
 
     override fun setOrientation(orientation: DeviceOrientation) {
-        runDeviceCall {
-            iosDevice.setOrientation(orientation.camelCaseName)
+        metrics.measured("operation", mapOf("command" to "setOrientation")) {
+            runDeviceCall("setOrientation") { iosDevice.setOrientation(orientation.camelCaseName) }
         }
     }
 

diff --git a/maestro-orchestra/src/main/java/maestro/orchestra/yaml/YamlFluentCommand.kt b/maestro-orchestra/src/main/java/maestro/orchestra/yaml/YamlFluentCommand.kt
@@ -57,6 +57,7 @@ import maestro.orchestra.ScrollCommand
 import maestro.orchestra.ScrollUntilVisibleCommand
 import maestro.orchestra.SetAirplaneModeCommand
 import maestro.orchestra.SetLocationCommand
+import maestro.orchestra.SetOrientationCommand
 import maestro.orchestra.StartRecordingCommand
 import maestro.orchestra.StopAppCommand
 import maestro.orchestra.StopRecordingCommand