Add test for toolbar_pool.toolbars sort order

django-cms · Feb 27, 2014 · d21515d · d21515d
1 parent 7f06aea
commit d21515d
Showing 1 changed file with 13 additions and 2 deletions.
diff --git a/cms/tests/toolbar_pool.py b/cms/tests/toolbar_pool.py
@@ -1,5 +1,6 @@
 from django.core.exceptions import ImproperlyConfigured
 from django.test import TestCase
+from django.utils.datastructures import SortedDict
 
 from cms.exceptions import ToolbarAlreadyRegistered, ToolbarNotRegistered
 from cms.toolbar_base import CMSToolbar
@@ -23,16 +24,26 @@ def test_register(self):
             'cms.tests.toolbar_pool.TestToolbar': TestToolbar})
 
         self.assertRaises(ToolbarAlreadyRegistered,
-                self.pool.register, TestToolbar)
+                          self.pool.register, TestToolbar)
 
     def test_register_type(self):
         self.assertRaises(ImproperlyConfigured, self.pool.register, str)
         self.assertRaises(ImproperlyConfigured, self.pool.register, object)
 
+    def test_register_order(self):
+        print(self.pool.toolbars)
+        self.pool.register(TestToolbar)
+        self.pool.register(CMSToolbar)
+
+        test_toolbar = SortedDict()
+        test_toolbar['cms.tests.toolbar_pool.TestToolbar'] = TestToolbar
+        test_toolbar['cms.toolbar_base.CMSToolbar'] = CMSToolbar
+        self.assertEqual(list(test_toolbar.keys()), list(self.pool.toolbars.keys()))
+
     def test_unregister(self):
         self.pool.register(TestToolbar)
         self.pool.unregister(TestToolbar)
         self.assertEqual(self.pool.toolbars, {})
 
         self.assertRaises(ToolbarNotRegistered,
-                self.pool.unregister, TestToolbar)
+                          self.pool.unregister, TestToolbar)