Added css class tests

collective · May 31, 2017 · c60d49f · c60d49f
1 parent 7a87ca0
commit c60d49f
Showing 1 changed file with 30 additions and 0 deletions.
diff --git a/collective/portlet/embed/tests/test_portlet.py b/collective/portlet/embed/tests/test_portlet.py
@@ -71,6 +71,36 @@ def test_obtain_renderer(self):
             (context, request, view, manager, assignment), IPortletRenderer)
         self.failUnless(isinstance(renderer, portlet.Renderer))
 
+    def test_correct_css_class_if_not_set(self):
+        context = self.folder
+        request = self.folder.REQUEST
+        view = self.folder.restrictedTraverse('@@plone')
+        manager = getUtility(IPortletManager, name='plone.rightcolumn',
+                             context=self.portal)
+
+        assignment = portlet.Assignment(header='Header')
+
+        renderer = getMultiAdapter(
+            (context, request, view, manager, assignment), IPortletRenderer)
+        self.assertEqual(renderer.css_class(), 'portlet-embed-header')
+
+    def test_correct_css_class_if_set(self):
+        context = self.folder
+        request = self.folder.REQUEST
+        view = self.folder.restrictedTraverse('@@plone')
+        manager = getUtility(IPortletManager, name='plone.rightcolumn',
+                             context=self.portal)
+
+        assignment = portlet.Assignment(
+            header='Header',
+            portlet_class='test-class',
+        )
+
+        renderer = getMultiAdapter(
+            (context, request, view, manager, assignment), IPortletRenderer)
+        self.assertEqual(renderer.css_class(),
+                         'portlet-embed-header test-class')
+
 
 class TestRenderer(IntegrationTestCase):