From 53d315788de5a02d378611ace406ec241428e8ac Mon Sep 17 00:00:00 2001
From: Mike McLean <mikem@redhat.com>
Date: Nov 22 2016 21:41:31 +0000
Subject: add test for ClientSession.serverPrincipal


---

diff --git a/tests/test_client_session.py b/tests/test_client_session.py
new file mode 100644
index 0000000..dc226f2
--- /dev/null
+++ b/tests/test_client_session.py
@@ -0,0 +1,35 @@
+import mock
+import unittest
+
+import koji
+
+
+class TestClientSession(unittest.TestCase):
+
+    @mock.patch('socket.getfqdn')
+    def test_server_principal_rdns(self, getfqdn):
+        opts = {'krb_rdns': True}
+        session = koji.ClientSession('http://koji.example.com/kojihub', opts)
+        cprinc = mock.MagicMock()
+        cprinc.realm = "REALM"
+        getfqdn.return_value = 'koji02.example.com'
+
+        princ = session._serverPrincipal(cprinc)
+        self.assertEqual(princ, 'host/koji02.example.com@REALM')
+        getfqdn.assert_called_with('koji.example.com')
+
+    @mock.patch('socket.getfqdn')
+    def test_server_principal_no_rdns(self, getfqdn):
+        opts = {'krb_rdns': False}
+        session = koji.ClientSession('http://koji.example.com/kojihub', opts)
+        cprinc = mock.MagicMock()
+        cprinc.realm = "REALM"
+        getfqdn.return_value = 'koji02.example.com'
+
+        princ = session._serverPrincipal(cprinc)
+        self.assertEqual(princ, 'host/koji.example.com@REALM')
+        getfqdn.assert_not_called()
+
+
+
+