unit tests for activate_session

2017-09-14 22:59:41 -04:00 · 2017-09-14 22:59:41 -04:00 · 9bd9e07b42
commit 9bd9e07b42
parent 3233dc760f
1 changed files with 124 additions and 0 deletions
--- a/tests/test_cli/test_activate_session.py
+++ b/tests/test_cli/test_activate_session.py
@ -0,0 +1,124 @@
+import unittest
+import mock
+
+from koji_cli.lib import activate_session
+
+
+class TestActivateSession(unittest.TestCase):
+
+    def setUp(self):
+        self.isfile = mock.patch('os.path.isfile').start()
+        self.warn = mock.patch('koji_cli.lib.warn').start()
+        self.error = mock.patch('koji_cli.lib.error').start()
+        # self.ensure_connection = mock.patch('koji_cli.lib.warn.ensure_connection').start()
+        self.has_krb_creds = mock.patch('koji_cli.lib.has_krb_creds').start()
+        self.isfile.return_value = False
+        self.has_krb_creds.return_value = False
+
+    def tearDown(self):
+        mock.patch.stopall()
+
+    def test_activate_session_noauth(self):
+        session = mock.MagicMock()
+        session.logged_in = False
+        options = {'authtype': 'noauth', 'debug': False}
+        activate_session(session, options)
+        options = {'authtype': None, 'noauth': True, 'debug': False}
+        activate_session(session, options)
+        session.login.assert_not_called()
+        session.ssl_login.assert_not_called()
+        session.krb_login.assert_not_called()
+
+    def test_activate_session_ssl(self):
+        session = mock.MagicMock()
+        session.logged_in = True
+        options = {'authtype': 'ssl',
+                'debug': False,
+                'cert': 'CERT',
+                'serverca': 'SERVERCA'}
+        activate_session(session, options)
+        session.ssl_login.assert_called_once_with('CERT', None, 'SERVERCA',
+                    proxyuser=None)
+        session.login.assert_not_called()
+        session.krb_login.assert_not_called()
+
+    def test_activate_session_ssl_implicit(self):
+        session = mock.MagicMock()
+        session.logged_in = True
+        options = {'authtype': None,
+                'debug': False,
+                'cert': 'CERT',
+                'serverca': 'SERVERCA'}
+        self.isfile.return_value = True
+        activate_session(session, options)
+        session.ssl_login.assert_called_once_with('CERT', None, 'SERVERCA',
+                    proxyuser=None)
+        session.login.assert_not_called()
+        session.krb_login.assert_not_called()
+
+    def test_activate_session_pw(self):
+        session = mock.MagicMock()
+        session.logged_in = True
+        self.isfile.return_value = True
+        options = {'authtype': 'password', 'debug': False, 'cert': None}
+        activate_session(session, options)
+        session.login.assert_called_once_with()
+        session.ssl_login.assert_not_called()
+        session.krb_login.assert_not_called()
+
+    def test_activate_session_pw_implicit(self):
+        session = mock.MagicMock()
+        session.logged_in = True
+        self.isfile.return_value = False
+        options = {'authtype': None, 'debug': False, 'cert': None,
+                    'user': 'USER'}
+        activate_session(session, options)
+        session.login.assert_called_once_with()
+        session.ssl_login.assert_not_called()
+        session.krb_login.assert_not_called()
+
+    def test_activate_session_krb(self):
+        session = mock.MagicMock()
+        session.logged_in = True
+        self.isfile.return_value = True
+        options = {'authtype': 'kerberos', 'debug': False, 'cert': None,
+                'keytab': None, 'principal': None}
+        activate_session(session, options)
+        session.login.assert_not_called()
+        session.ssl_login.assert_not_called()
+        session.krb_login.assert_called_once_with(proxyuser=None)
+
+    def test_activate_session_krb_implicit(self):
+        session = mock.MagicMock()
+        session.logged_in = True
+        self.isfile.return_value = False
+        options = {'authtype': None, 'debug': False, 'cert': None,
+                'keytab': None, 'principal': None}
+        self.has_krb_creds.return_value = True
+        activate_session(session, options)
+        session.login.assert_not_called()
+        session.ssl_login.assert_not_called()
+        session.krb_login.assert_called_once_with(proxyuser=None)
+
+    def test_activate_session_krb_keytab(self):
+        session = mock.MagicMock()
+        session.logged_in = True
+        self.isfile.return_value = True
+        options = {'authtype': 'kerberos', 'debug': False, 'cert': None,
+                'keytab': 'KEYTAB', 'principal': 'PRINCIPAL'}
+        activate_session(session, options)
+        session.login.assert_not_called()
+        session.ssl_login.assert_not_called()
+        session.krb_login.assert_called_once_with(principal='PRINCIPAL',
+                    keytab='KEYTAB', proxyuser=None)
+
+    def test_activate_session_no_method(self):
+        session = mock.MagicMock()
+        session.logged_in = False
+        options = {'authtype': None, 'debug': False, 'cert': None}
+        self.has_krb_creds.return_value = False
+        activate_session(session, options)
+        session.login.assert_not_called()
+        session.ssl_login.assert_not_called()
+        session.krb_login.assert_not_called()
+        self.error.assert_called_once()