From 070d39ca86b7a69ec3db3f399a4c52813c50ac5d Mon Sep 17 00:00:00 2001
From: Davit Lomauridze <lomauridze@hmg.software>
Date: Tue, 13 Aug 2024 15:13:27 +0400
Subject: [PATCH] [FIX]doument_page_portal: added test cases

---
 .../tests/test_document_controller.py         | 54 ++++++++++---------
 1 file changed, 28 insertions(+), 26 deletions(-)

diff --git a/document_page_portal/tests/test_document_controller.py b/document_page_portal/tests/test_document_controller.py
index 4a3a8042..87068161 100644
--- a/document_page_portal/tests/test_document_controller.py
+++ b/document_page_portal/tests/test_document_controller.py
@@ -1,36 +1,36 @@
 # Path: addons/document_page_portal/tests/test_portal.py
 
-from odoo.tests.common import HttpCase, tagged
-from odoo.exceptions import AccessError, MissingError
 from odoo import fields as odoo_fields
+from odoo.exceptions import AccessError, MissingError
+from odoo.tests.common import HttpCase, tagged
 
-@tagged('post_install', '-at_install')
+
+@tagged("post_install", "-at_install")
 class TestDocumentPageController(HttpCase):
-
     def setUp(self):
-        super(TestDocumentPagePortal, self).setUp()
-        self.portal_user = self.env.ref('base.public_user')
-        self.document_page_model = self.env['document.page']
+        super(TestDocumentPageController, self).setUp()
+        self.portal_user = self.env.ref("base.public_user")
+        self.document_page_model = self.env["document.page"]
 
     def test_prepare_portal_layout_values(self):
         """Test _prepare_portal_layout_values method."""
         with self.env.cr.savepoint():
             self.env = self.env(user=self.portal_user)
-            result = self.env['ir.http']._dispatch()
-            values = self.env['portal.customer.portal']._prepare_portal_layout_values()
+            self.env["ir.http"]._dispatch()
+            values = self.env["portal.customer.portal"]._prepare_portal_layout_values()
             document_page_count = self.document_page_model.search_count(
-                [('type', '=', 'content')]
+                [("type", "=", "content")]
             )
-            self.assertEqual(values.get('document_page_count'), document_page_count)
+            self.assertEqual(values.get("document_page_count"), document_page_count)
 
     def test_get_archive_groups(self):
         """Test _get_archive_groups method."""
         with self.env.cr.savepoint():
             self.env = self.env(user=self.portal_user)
-            result = self.env['ir.http']._dispatch()
-            domain = [('type', '=', 'content')]
-            groups = self.env['portal.customer.portal']._get_archive_groups(
-                'document.page', domain
+            self.env["ir.http"]._dispatch()
+            domain = [("type", "=", "content")]
+            groups = self.env["portal.customer.portal"]._get_archive_groups(
+                "document.page", domain
             )
             self.assertTrue(isinstance(groups, list))
 
@@ -38,17 +38,19 @@ class TestDocumentPageController(HttpCase):
         """Test portal_my_knowledge_document_pages route."""
         with self.env.cr.savepoint():
             self.env = self.env(user=self.portal_user)
-            result = self.url_open('/my/knowledge/documents')
+            result = self.url_open("/my/knowledge/documents")
             self.assertEqual(result.status_code, 200)
 
             # Test with search query
-            result = self.url_open('/my/knowledge/documents?search=test')
+            result = self.url_open("/my/knowledge/documents?search=test")
             self.assertEqual(result.status_code, 200)
 
             # Test with date filter
             date_begin = odoo_fields.Date.to_string(odoo_fields.Date.today())
             date_end = odoo_fields.Date.to_string(odoo_fields.Date.today())
-            result = self.url_open(f'/my/knowledge/documents?date_begin={date_begin}&date_end={date_end}')
+            result = self.url_open(
+                f"/my/knowledge/documents?date_begin={date_begin}&date_end={date_end}"
+            )
             self.assertEqual(result.status_code, 200)
 
     def test_document_pages_followup(self):
@@ -56,18 +58,18 @@ class TestDocumentPageController(HttpCase):
         with self.env.cr.savepoint():
             self.env = self.env(user=self.portal_user)
             # Assuming there is at least one document.page record
-            document_page = self.document_page_model.create({
-                'name': 'Test Document',
-                'type': 'content'
-            })
-            result = self.url_open(f'/knowledge/document/{document_page.id}')
+            document_page = self.document_page_model.create(
+                {"name": "Test Document", "type": "content"}
+            )
+            result = self.url_open(f"/knowledge/document/{document_page.id}")
             self.assertEqual(result.status_code, 200)
 
             # Test with an invalid document_page_id
             with self.assertRaises(MissingError):
-                self.url_open('/knowledge/document/999999')
+                self.url_open("/knowledge/document/999999")
 
             # Test access with a non-public user
             with self.assertRaises(AccessError):
-                self.url_open(f'/knowledge/document/{document_page.id}?token=invalid_token')
-
+                self.url_open(
+                    f"/knowledge/document/{document_page.id}?token=invalid_token"
+                )