Enable/disable channel

Fixes: https://pagure.io/koji/issue/1851

tests/test_cli/test_disable_channel.py

@@ -0,0 +1,115 @@
+from __future__ import absolute_import
+
+import unittest
+
+import mock
+import six
+
+import koji
+from koji_cli.commands import handle_disable_channel
+from . import utils
+
+
+class TestDisableChannel(utils.CliTestCase):
+    # Show long diffs in error output...
+    maxDiff = None
+
+    def setUp(self):
+        self.error_format = """Usage: %s disable-channel [options] <channelname> [<channelname> ...]
+(Specify the --help global option for a list of other help options)
+
+%s: error: {message}
+""" % (self.progname, self.progname)
+        self.channelinfo = [
+            {'comment': None, 'description': None, 'enabled': False, 'id': 1,
+             'name': 'test-channel'}
+        ]
+
+    def __vm(self, result):
+        m = koji.VirtualCall('mcall_method', [], {})
+        if isinstance(result, dict) and result.get('faultCode'):
+            m._result = result
+        else:
+            m._result = (result,)
+        return m
+
+    @mock.patch('sys.stderr', new_callable=six.StringIO)
+    @mock.patch('sys.stdout', new_callable=six.StringIO)
+    @mock.patch('koji_cli.commands.activate_session')
+    def test_handle_disable_channel(self, activate_session_mock, stdout, stderr):
+        """Test disable-channel function"""
+        options = mock.MagicMock()
+        session = mock.MagicMock()
+
+        mcall = session.multicall.return_value.__enter__.return_value
+
+        mcall.getChannel.return_value = self.__vm(None)
+
+        arguments = ['test-channel']
+        with self.assertRaises(SystemExit) as ex:
+            handle_disable_channel(options, session, arguments)
+        self.assertExitCode(ex, 1)
+        activate_session_mock.assert_called_once()
+        session.multicall.assert_called_once()
+        session.disableChannel.assert_not_called()
+        expect = ''
+        for host in arguments:
+            expect += "No such channel: %s\n" % host
+        stderr_exp = "No changes made. Please correct the command line.\n"
+        self.assert_console_message(stdout, expect)
+        self.assert_console_message(stderr, stderr_exp)
+
+        # reset session mocks
+        activate_session_mock.reset_mock()
+        session.disableChannel.reset_mock()
+        session.multicall.reset_mock()
+        mcall = session.multicall.return_value.__enter__.return_value
+
+        mcall.getChannel.return_value = self.__vm(self.channelinfo)
+
+        arguments = ['test-channel', '--comment', 'enable channel test']
+        handle_disable_channel(options, session, arguments)
+        activate_session_mock.assert_called_once()
+        self.assertEqual(2, session.multicall.call_count)
+        self.assert_console_message(stdout, '')
+
+    @mock.patch('sys.stdout', new_callable=six.StringIO)
+    @mock.patch('koji_cli.commands.activate_session')
+    def test_handle_disable_host_no_argument(self, activate_session_mock, stdout):
+        """Test disable-channel function without arguments"""
+        options = mock.MagicMock()
+        session = mock.MagicMock()
+
+        session.getChannel.return_value = None
+        session.multicall.return_value = [[None]]
+        session.disableChannel.return_value = True
+
+        expected = self.format_error_message("At least one channel must be specified")
+        self.assert_system_exit(
+            handle_disable_channel,
+            options,
+            session,
+            [],
+            stderr=expected,
+            activate_session=None)
+
+        activate_session_mock.assert_not_called()
+        session.getChannel.assert_not_called()
+        session.multicall.assert_not_called()
+        session.disableChannel.assert_not_called()
+
+    def test_handle_disable_channel_help(self):
+        """Test disable-channel help message"""
+        self.assert_help(
+            handle_disable_channel,
+            """Usage: %s disable-channel [options] <channelname> [<channelname> ...]
+(Specify the --help global option for a list of other help options)
+
+Options:
+  -h, --help         show this help message and exit
+  --comment=COMMENT  Comment indicating why the channel(s) are being disabled
+""" % self.progname)
+
+
+if __name__ == '__main__':
+    unittest.main()