@@ -19,27 +19,30 @@
1919
2020"""Tests for Notifications"""
2121
22−from unity8 import shell
23−from unity8.process_helpers import unlock_unity
24−from unity8.shell.tests import UnityTestCase, _get_device_emulation_scenarios
25−
26−from testtools.matchers import Equals, NotEquals
27−from autopilot.matchers import Eventually
28−
29−from gi.repository import Notify
3022import time
3123import os
3224import logging
3325import signal
3426import subprocess
3527
28+from autopilot.matchers import Eventually
29+from gi.repository import Notify
30+from testtools.matchers import Equals, NotEquals
31+from ubuntuuitoolkit import ubuntu_scenarios
32+
33+from unity8 import shell
34+from unity8.process_helpers import unlock_unity
35+from unity8.shell.tests import UnityTestCase
36+
37+
3638logger = logging.getLogger(__name__)
3739
3840
3941class NotificationsBase(UnityTestCase):
4042 """Base class for all notification tests that provides helper methods."""
4143
42− scenarios = _get_device_emulation_scenarios('Nexus4')
44+ scenarios = ubuntu_scenarios.get_device_simulation_scenarios(
45+ ubuntu_scenarios.NEXUS4_DEVICE)
4346
4447 def _get_icon_path(self, icon_name):
4548 """Given an icons file name returns the full path (either system or