Add tests for representations.

Lukasa · Lukasa · commit 424e8b1b1631 · 2015-11-30T16:55:21.000Z
diff --git a/test/test_priority.py b/test/test_priority.py
@@ -5,6 +5,8 @@
 
 Tests for the Priority trees
 """
+import itertools
+
 from hypothesis import given
 from hypothesis.strategies import integers, lists, tuples
 
@@ -19,6 +21,15 @@
 )
 
 
+class TestStream(object):
+    def test_stream_repr(self):
+        """
+        The stream representation renders according to the README.
+        """
+        s = priority.Stream(stream_id=80, weight=16)
+        assert repr(s) == "Stream<id=80, weight=16>"
+
+
 class TestPriorityTree(object):
     def test_priority_tree_one_stream(self):
         """
@@ -65,6 +76,30 @@ def test_priorities_stream_weights(self, stream_data):
         for stream_id, weight in stream_data:
             assert weight == priorities.stream_weight(stream_id)
 
+    def test_priorities_repr(self, readme_tree):
+        """
+        The priorities representation renders according to the README.
+        """
+        priorities = readme_tree.priorities()
+        r = repr(priorities)
+
+        # Dictionaries hurt ordering, so just check all possible orderings.
+        stream_reprs = [
+            'Stream<id=1, weight=16>',
+            'Stream<id=3, weight=16>',
+            'Stream<id=7, weight=32>',
+        ]
+        possible_orderings = [
+            ', '.join(permutation)
+            for permutation in itertools.permutations(stream_reprs)
+        ]
+        base_repr = "Priorities<total_weight=64, streams=[%s]>"
+        all_reprs = [
+            base_repr % streams for streams in possible_orderings
+        ]
+
+        assert any(r == representation for representation in all_reprs)
+
     def test_drilling_down(self, readme_tree):
         """
         We can drill down each layer of the tree by stream ID.
