fixing broken unit.modules.test_telegram test.

tests/unit/modules/test_telegram.py

@@ -8,11 +8,14 @@ Tests for the Telegram execution module.
 # Import Python Libs
 from __future__ import absolute_import
 
+import logging
+
 # Import Salt Testing Libs
 from tests.support.mixins import LoaderModuleMockMixin
 from tests.support.unit import TestCase, skipIf
 from tests.support.mock import (
     patch,
+    Mock,
     MagicMock,
     NO_MOCK,
     NO_MOCK_REASON
@@ -22,6 +25,49 @@ from tests.support.mock import (
 import salt.modules.telegram as telegram
 
 
+log = logging.getLogger(__name__)
+
+class RequestMock(Mock):
+    '''
+    Request Mock
+    '''
+
+    def get(self, *args, **kwargs):
+        return RequestResponseMock()
+
+    def put(self, *args, **kwargs):
+        self.args = args
+        self.kwargs = kwargs
+        return RequestPutResponseMock()
+
+    def delete(self, *args, **kwargs):
+        self.args = args
+        self.kwargs = kwargs
+        return RequestResponseMock()
+
+
+class RequestResponseMock(Mock):
+    '''
+    Request Response Mock
+    '''
+    def json(self):
+        return [{'url': 'http://example.org',
+                 '_id': 1234}, ]
+
+
+class RequestPutResponseMock(Mock):
+    '''
+    Request Put Response Mock
+    '''
+    ok = True
+
+    def json(self):
+        return {'_id': 4321}
+
+
+REQUEST_MOCK = RequestMock()
+
+
 @skipIf(NO_MOCK, NO_MOCK_REASON)
 class TelegramModuleTest(TestCase, LoaderModuleMockMixin):
     '''
@@ -29,35 +75,22 @@ class TelegramModuleTest(TestCase, LoaderModuleMockMixin):
     '''
     def setup_loader_modules(self):
         module_globals = {
-            telegram: {
-                '__salt__': {
-                    'config.get': MagicMock(return_value={
-                        'telegram': {
-                            'chat_id': '123456789',
-                            'token': '000000000:xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'
-                        }
-                    })
-                }
-            }
+            '__salt__': {
+                'config.get': MagicMock(return_value={
+                    'telegram': {
+                        'chat_id': '123456789',
+                        'token': '000000000:xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx'
+                    }
+                }),
+                'requests.put': Mock(),
+            },
+            'requests': REQUEST_MOCK
         }
-        if telegram.HAS_REQUESTS is False:
-            module_globals['sys.modules'] = {'requests': MagicMock()}
-        return module_globals
+        return {telegram: module_globals}
 
     def test_post_message(self):
         '''
         Test the post_message function.
         '''
         message = 'Hello World!'
-
-        class MockRequests(object):
-            """
-            Mock of requests response.
-            """
-            def json(self):
-                return {'ok': True}
-
-        with patch('salt.modules.telegram.requests.post',
-                   MagicMock(return_value=MockRequests())):
-
-            self.assertTrue(telegram.post_message(message))
+        self.assertTrue(telegram.post_message(message))