Use new test libs

mcpanayides · Mar 7, 2017 · c5a7b68 · c5a7b68
1 parent be2c2fa
commit c5a7b68
Showing 1 changed file with 4 additions and 8 deletions.
diff --git a/tests/unit/beacons/test_telegram_bot_msg_beacon.py b/tests/unit/beacons/test_telegram_bot_msg_beacon.py
@@ -4,14 +4,13 @@
 from __future__ import absolute_import
 import datetime
 
-# Salt testing libs
-from salttesting import TestCase, skipIf
-from salttesting.helpers import ensure_in_syspath
-from salttesting.mock import NO_MOCK, NO_MOCK_REASON, MagicMock, patch
-
 # Salt libs
 from salt.beacons import telegram_bot_msg
 
+# Salt testing libs
+from tests.support.unit import TestCase, skipIf
+from tests.support.mock import NO_MOCK, NO_MOCK_REASON, MagicMock, patch
+
 # Third-party libs
 try:
     import telegram
@@ -20,9 +19,6 @@
     HAS_TELEGRAM = False
 
 
-ensure_in_syspath('../../')
-
-
 @skipIf(not HAS_TELEGRAM, 'telegram is not available')
 @skipIf(NO_MOCK, NO_MOCK_REASON)
 class TelegramBotMsgBeaconTestCase(TestCase):