Toolbar tests

tests/test_toolbar.py

@@ -0,0 +1,27 @@
+# -*- coding: utf-8 -*-
+from cms.toolbar.items import ModalItem
+from django.core.urlresolvers import reverse
+from djangocms_blog.models import BLOG_CURRENT_POST_IDENTIFIER
+
+
+from . import BaseTest
+
+
+class ToolbarTest(BaseTest):
+
+    def test_toolbar_with_items(self):
+        """
+        Test that Blog toolbar is present and contains all items
+        """
+        from cms.toolbar.toolbar import CMSToolbar
+        post = self._get_post(self.data['en'][0])
+        page1, page2 = self.get_pages()
+        request = self.get_page_request(page1, self.user, r'/en/blog/', edit=True)
+        setattr(request, BLOG_CURRENT_POST_IDENTIFIER, post)
+        toolbar = CMSToolbar(request)
+        toolbar.get_left_items()
+        blog_menu = toolbar.menus['djangocms_blog']
+        self.assertEqual(len(blog_menu.find_items(ModalItem, url=reverse('admin:djangocms_blog_post_changelist'))), 1)
+        self.assertEqual(len(blog_menu.find_items(ModalItem, url=reverse('admin:djangocms_blog_post_add'))), 1)
+        self.assertEqual(len(blog_menu.find_items(ModalItem, url=reverse('admin:djangocms_blog_post_change', args=(post.pk,)))), 1)
+