Add unit tests notifications, list-channels

tests/test_cli/test_version.py

@@ -0,0 +1,38 @@
+from __future__ import absolute_import
+
+import mock
+from six.moves import StringIO
+
+import koji
+from koji_cli.commands import handle_version
+from . import utils
+
+
+class TestVersion(utils.CliTestCase):
+    def setUp(self):
+        self.options = mock.MagicMock()
+        self.options.debug = False
+        self.session = mock.MagicMock()
+        self.session.getAPIVersion.return_value = koji.API_VERSION
+
+    @mock.patch('sys.stdout', new_callable=StringIO)
+    def test_version_valid(self, stdout):
+        expected = """Client: 1.26.0
+Hub:    1.26.0
+"""
+        self.session.getKojiVersion.return_value = '1.26.0'
+        rv = handle_version(self.options, self.session, [])
+        self.assertEqual(rv, None)
+        self.assert_console_message(stdout, expected)
+        self.session.getKojiVersion.assert_called_once_with()
+
+    @mock.patch('sys.stdout', new_callable=StringIO)
+    def test_version_invalid(self, stdout):
+        expected = """Client: 1.26.0
+Hub:    Can't determine (older than 1.23)
+"""
+        self.session.getKojiVersion.side_effect = koji.GenericError()
+        rv = handle_version(self.options, self.session, [])
+        self.assertEqual(rv, None)
+        self.assert_console_message(stdout, expected)
+        self.session.getKojiVersion.assert_called_once_with()