Implement total ordering for PKISubsystem and PKIInstance

In Python 3 subclasses no longer implement automatic ordering. To
provide ordering for sort() and custom comparison, __eq__ and __lt__ are
required.

https://fedorahosted.org/pki/ticket/2216

1 files changed, 61 insertions, 0 deletions

diff --git a/tests/python/server/test_server.py b/tests/python/server/test_server.py
new file mode 100644
index 000000000..646e6826f
--- /dev/null
+++ b/tests/python/server/test_server.py
@@ -0,0 +1,61 @@
+# -*- coding: utf-8 -*-
+# Authors:
+#     Christian Heimes <cheimes@redhat.com>
+#
+# This program is free software; you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation; version 2 of the License.
+#
+# This program is distributed in the hope that it will be useful,
+# but WITHOUT ANY WARRANTY; without even the implied warranty of
+# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+# GNU General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License along
+# with this program; if not, write to the Free Software Foundation, Inc.,
+# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
+#
+# Copyright (C) 2016 Red Hat, Inc.
+# All rights reserved.
+#
+
+import unittest
+
+from pki.server import PKISubsystem, PKIInstance
+
+
+class PKIServerTests(unittest.TestCase):
+    def test_instance_ordering(self):
+        ca = PKIInstance('ca')
+        ca9 = PKIInstance('ca', 9)
+        kra = PKIInstance('kra')
+        instances = [kra, ca, ca9]
+        self.assertEqual(sorted(instances), [ca9, ca, kra])
+        self.assertEqual(sorted(instances, reverse=True), [kra, ca, ca9])
+        self.assertTrue(ca == ca)
+        self.assertFalse(ca != ca)
+        self.assertFalse(ca == ca9)
+        self.assertTrue(ca != ca9)
+        with self.assertRaises(TypeError):
+            hash(ca)
+
+    def test_subsystem(self):
+        ca = PKIInstance('ca')
+        kra = PKIInstance('kra')
+        casub = PKISubsystem(ca, 'ca sub')
+        krasub = PKISubsystem(kra, 'kra sub')
+        subs = [casub, krasub]
+        self.assertEqual(sorted(subs), [casub, krasub])
+        self.assertEqual(sorted(subs, reverse=True), [krasub, casub])
+        self.assertTrue(casub == casub)
+        self.assertFalse(casub != casub)
+        self.assertFalse(casub == krasub)
+        self.assertTrue(casub != krasub)
+        self.assertFalse(ca == casub)
+        self.assertTrue(ca != casub)
+        with self.assertRaises(TypeError):
+            hash(casub)
+
+
+if __name__ == '__main__':
+    unittest.main()