1 files changed, 91 insertions, 0 deletions
diff --git a/testing/mozbase/mozrunner/mozrunner/devices/emulator_screen.py b/testing/mozbase/mozrunner/mozrunner/devices/emulator_screen.py
new file mode 100644
index 0000000000..8f261c3610
--- /dev/null
+++ b/testing/mozbase/mozrunner/mozrunner/devices/emulator_screen.py
@@ -0,0 +1,91 @@
+# This Source Code Form is subject to the terms of the Mozilla Public
+# License, v. 2.0. If a copy of the MPL was not distributed with this
+# file, You can obtain one at http://mozilla.org/MPL/2.0/.
+
+
+class EmulatorScreen(object):
+    """Class for screen related emulator commands."""
+
+    SO_PORTRAIT_PRIMARY = "portrait-primary"
+    SO_PORTRAIT_SECONDARY = "portrait-secondary"
+    SO_LANDSCAPE_PRIMARY = "landscape-primary"
+    SO_LANDSCAPE_SECONDARY = "landscape-secondary"
+
+    def __init__(self, emulator):
+        self.emulator = emulator
+
+    def initialize(self):
+        self.orientation = self.SO_PORTRAIT_PRIMARY
+
+    def _get_raw_orientation(self):
+        """Get the raw value of the current device orientation."""
+        response = self.emulator._run_telnet("sensor get orientation")
+
+        return response[0].split("=")[1].strip()
+
+    def _set_raw_orientation(self, data):
+        """Set the raw value of the specified device orientation."""
+        self.emulator._run_telnet("sensor set orientation %s" % data)
+
+    def get_orientation(self):
+        """Get the current device orientation.
+
+        Returns;
+            orientation -- Orientation of the device. One of:
+                            SO_PORTRAIT_PRIMARY - system buttons at the bottom
+                            SO_PORTRIAT_SECONDARY - system buttons at the top
+                            SO_LANDSCAPE_PRIMARY - system buttons at the right
+                            SO_LANDSCAPE_SECONDARY - system buttons at the left
+
+        """
+        data = self._get_raw_orientation()
+
+        if data == "0:-90:0":
+            orientation = self.SO_PORTRAIT_PRIMARY
+        elif data == "0:90:0":
+            orientation = self.SO_PORTRAIT_SECONDARY
+        elif data == "0:0:90":
+            orientation = self.SO_LANDSCAPE_PRIMARY
+        elif data == "0:0:-90":
+            orientation = self.SO_LANDSCAPE_SECONDARY
+        else:
+            raise ValueError("Unknown orientation sensor value: %s." % data)
+
+        return orientation
+
+    def set_orientation(self, orientation):
+        """Set the specified device orientation.
+
+        Args
+            orientation -- Orientation of the device. One of:
+                            SO_PORTRAIT_PRIMARY - system buttons at the bottom
+                            SO_PORTRIAT_SECONDARY - system buttons at the top
+                            SO_LANDSCAPE_PRIMARY - system buttons at the right
+                            SO_LANDSCAPE_SECONDARY - system buttons at the left
+        """
+        orientation = SCREEN_ORIENTATIONS[orientation]
+
+        if orientation == self.SO_PORTRAIT_PRIMARY:
+            data = "0:-90:0"
+        elif orientation == self.SO_PORTRAIT_SECONDARY:
+            data = "0:90:0"
+        elif orientation == self.SO_LANDSCAPE_PRIMARY:
+            data = "0:0:90"
+        elif orientation == self.SO_LANDSCAPE_SECONDARY:
+            data = "0:0:-90"
+        else:
+            raise ValueError("Invalid orientation: %s" % orientation)
+
+        self._set_raw_orientation(data)
+
+    orientation = property(get_orientation, set_orientation)
+
+
+SCREEN_ORIENTATIONS = {
+    "portrait": EmulatorScreen.SO_PORTRAIT_PRIMARY,
+    "landscape": EmulatorScreen.SO_LANDSCAPE_PRIMARY,
+    "portrait-primary": EmulatorScreen.SO_PORTRAIT_PRIMARY,
+    "landscape-primary": EmulatorScreen.SO_LANDSCAPE_PRIMARY,
+    "portrait-secondary": EmulatorScreen.SO_PORTRAIT_SECONDARY,
+    "landscape-secondary": EmulatorScreen.SO_LANDSCAPE_SECONDARY,
+}