Add test for iib_api

Author: woblerr

modules/iib_api.py

@@ -19,13 +19,19 @@ def run_iib_command(**kwargs):
         command = iib_command.format(broker)
     else:
         command = iib_command
+    output = execute_command(command=command)
+    return output
+
+
+def execute_command(command):
+    """Executes in shell."""
     proc = subprocess.Popen(command,
                             shell=True,
                             stdout=subprocess.PIPE,
                             stderr=subprocess.STDOUT,
                             universal_newlines=True)
-    output = proc.communicate()[0]
-    return output
+    result = proc.communicate()[0]
+    return result
 
 
 def get_status(status):

tests/test_iib_api.py

@@ -3,10 +3,33 @@
 import os
 import sys
 import unittest
-from modules.iib_api import get_status
+try:
+    from unittest.mock import patch
+except ImportError:
+    from mock import patch
+from modules.iib_api import (
+    run_iib_command,
+    get_status)
 sys.path.append(os.getcwd())
 
 
+def mock_execute_command(command):
+    """Mock for `execute_command` function."""
+    return command
+
+
+class TestRunMqCommand(unittest.TestCase):
+    @patch('modules.iib_api.execute_command', side_effect=mock_execute_command)
+    def test_run_iib_command(self, execute_command):
+        """Tests for `run_iib_command` function."""
+        self.assertEqual(
+            run_iib_command(task='get_brokers_status'),
+            'mqsilist | grep Broker')
+        self.assertEqual(
+            run_iib_command(task='get_broker_objects', broker_name='TEST'),
+            'mqsilist TEST -r')
+
+
 class TestGetStatus(unittest.TestCase):
     def test_get_status(self):
         """Test for `get_status` function."""