@@ -0,0 +1,52 @@
1+# -*- Mode: Python; coding: utf-8; indent-tabs-mode: nil; tab-width: 4 -*-
2+#
3+# Unity Autopilot Test Suite
4+# Copyright (C) 2015 Canonical
5+#
6+# This program is free software: you can redistribute it and/or modify
7+# it under the terms of the GNU General Public License as published by
8+# the Free Software Foundation, either version 3 of the License, or
9+# (at your option) any later version.
10+#
11+# This program is distributed in the hope that it will be useful,
12+# but WITHOUT ANY WARRANTY; without even the implied warranty of
13+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14+# GNU General Public License for more details.
15+#
16+# You should have received a copy of the GNU General Public License
17+# along with this program. If not, see <http://www.gnu.org/licenses/>.
18+#
19+
20+"""Set up and clean up fixtures for the Unity acceptance tests."""
21+
22+import os
23+import subprocess
24+import sysconfig
25+import dbus
26+
27+import fixtures
28+
29+import unity8
30+
31+class IndicatorDisplayRotationLock(fixtures.Fixture):
32+
33+ def __init__(self, enable):
34+ super().__init__()
35+ self.enable = enable
36+
37+ def setUp(self):
38+ super().setUp()
39+ original_state = self._is_rotation_lock_enabled()
40+ if self.enable != original_state:
41+ self.addCleanup(self._set_rotation_lock, original_state)
42+ self._set_rotation_lock(self.enable)
43+
44+ def _is_rotation_lock_enabled(self):
45+ session_bus = dbus.SessionBus()
46+ rotation_lock = session_bus.get_object('com.canonical.indicator.rotation_lock', '/com/canonical/indicator/rotation_lock')
47+ # TODO: get the real lock value.
48+ return False
49+
50+ def _set_rotation_lock(self, value):
51+ session_bus = dbus.SessionBus()
52+ rotation_lock = session_bus.get_object('com.canonical.indicator.rotation_lock', '/com/canonical/indicator/rotation_lock')