test improved

document_page_approval/tests/__init__.py

@@ -1,2 +1,3 @@
 """Module test initialisation."""
 # -*- coding: utf-8 -*-
+from . import test_document_page_approval

document_page_approval/tests/test_document_page_approval.py

@@ -1,16 +1,35 @@
 """DocumentPageApproval test."""
 # -*- coding: utf-8 -*-
-from openerp.test.common import TransactionCase
+from openerp.tests import common
 
 
-class TestDocumentPageApproval(TransactionCase):
+class TestDocumentPageApproval(common.TransactionCase):
     """Test document page approval model."""
 
-    def setUp(self):
-        """Test setUp."""
-        super()
-        pass
+    def test_get_display_content(self):
+        """Test page display content."""
+        # Check content of a category
+        category = self.env['document.page'].search(
+                [('name', '=', 'OpenERP Features')])
 
-    def __init__(self, arg):
-        """Test initialisation."""
-        pass
+        self.assertIsNotNone(category.display_content, 'a category')
+
+        # Check content of a page
+        pages = self.env['document.page'].search(
+                [('parent_id', '=', category.id)])
+        page = pages[0]
+        self.assertIsNotNone(page.display_content, 'Page content')
+
+        # Check if approval is required
+        self.assertFalse(page.is_approval_required(page))
+
+        # Check content of an approval page
+        page.approval_required = True
+
+        self.assertIsNotNone(page.display_content, 'Page content')
+
+        # Check if approval is required
+        self.assertTrue(page.is_approval_required(page))
+
+        # Check if parent approval is required
+        self.assertFalse(page.is_parent_approval_required)