Merge pull request #23020 from jayeshka/raet_publish-unit-test

adding raet_publish unit test case

tests/unit/modules/raet_publish_test.py

@@ -0,0 +1,81 @@
+# -*- coding: utf-8 -*-
+'''
+    :codeauthor: :email:`Jayesh Kariya <jayeshk@saltstack.com>`
+'''
+
+# Import Python libs
+from __future__ import absolute_import
+
+# Import Salt Testing Libs
+from salttesting import skipIf, TestCase
+from salttesting.mock import (
+    NO_MOCK,
+    NO_MOCK_REASON,
+    MagicMock,
+    patch)
+
+from salttesting.helpers import ensure_in_syspath
+
+ensure_in_syspath('../../')
+
+# Import Salt Libs
+from salt.modules import raet_publish
+import salt.transport
+from salt.exceptions import SaltReqTimeoutError
+
+# Globals
+raet_publish.__opts__ = {}
+
+
+@skipIf(NO_MOCK, NO_MOCK_REASON)
+class RaetPublishTestCase(TestCase):
+    '''
+    Test cases for salt.modules.raet_publish
+    '''
+    def test_publish(self):
+        '''
+        Test for publish a command from the minion out to other minions.
+        '''
+        with patch.object(raet_publish, '_publish', return_value='A'):
+            self.assertEqual(raet_publish.publish('tgt', 'fun'), 'A')
+
+    def test_full_data(self):
+        '''
+        Test for return the full data about the publication,
+         this is invoked in the same way as the publish function
+        '''
+        with patch.object(raet_publish, '_publish', return_value='A'):
+            self.assertEqual(raet_publish.full_data('tgt', 'fun'), 'A')
+
+    def test_runner(self):
+        '''
+        Test for execute a runner on the master and return
+         the data from the runner function
+        '''
+        with patch.dict(raet_publish.__opts__, {'id': 'id'}):
+            with patch.object(salt.transport.Channel, 'factory', MagicMock()):
+                self.assertTrue(raet_publish.runner('fun'))
+
+        class MockFactory(object):
+            '''
+            Mock factory class
+            '''
+            load = ''
+
+            def send(self, load):
+                '''
+                mock send method
+                '''
+                self.load = load
+                raise SaltReqTimeoutError(load)
+
+        with patch.dict(raet_publish.__opts__, {'id': 'id'}):
+            with patch.object(salt.transport.Channel, 'factory',
+                              MagicMock(return_value=MockFactory())):
+                self.assertEqual(raet_publish.runner(1),
+                                 '1 runner publish timed out')
+
+
+if __name__ == '__main__':
+    from integration import run_tests
+    run_tests(RaetPublishTestCase, needs_daemon=False)