Added tests for Token comparison

bjmb · bjmb · commit 5138cf5c993c · 2017-03-01T18:27:31.000-05:00
diff --git a/tests/unit/test_types.py b/tests/unit/test_types.py
@@ -36,6 +36,7 @@
 from cassandra.pool import Host
 from cassandra.policies import SimpleConvictionPolicy, ConvictionPolicy
 from cassandra.util import Date, Time
+from cassandra.metadata import Token
 
 
 class TypeTests(unittest.TestCase):
@@ -371,3 +372,18 @@ def test_timer_order(self):
         self._check_sequence_consistency(time_from_string_equal, True)
 
         self._check_sequence_consistency(self._shuffle_lists(time_from_int, time_from_datetime, time_from_string))
+
+    def test_token_order(self):
+        """
+        Test Token class is ordered consistently
+
+        @since 3.9
+        @jira_ticket PYTHON-714
+        @expected_result the tokens are ordered correctly
+
+        @test_category data_types
+        """
+        tokens = [Token(1), Token(2), Token(3), Token(4)]
+        tokens_equal = [Token(1), Token(1)]
+        self._check_sequence_consistency(tokens)
+        self._check_sequence_consistency(tokens_equal, True)
