test

Author: mamico

src/iosanita/contenttypes/adapters/extractor.py

@@ -0,0 +1,19 @@
+from iosanita.contenttypes.interfaces import IoSanitaViewExtraData
+from iosanita.contenttypes.restapi.services.view_extra_data.extractor import (
+    ViewExtraDataExtractor,
+)
+from irst.policy.content.at import IAT
+from zope.component import adapter
+from zope.interface import implementer
+from zope.interface import Interface
+
+
+@implementer(IoSanitaViewExtraData)
+@adapter(IAT, Interface)
+class ViewExtraDataExtractorAT(ViewExtraDataExtractor):
+    def __call__(self):
+        bando_view = self.context.restrictedTraverse("bando_view")
+        return {
+            # "approfondimenti": self.get_approfondimenti(),
+            "stato_bando": bando_view.getBandoState(),
+        }

src/iosanita/contenttypes/browser/export_view.py

@@ -2,28 +2,30 @@
 from io import BytesIO
 from io import StringIO
 from iosanita.contenttypes import _
+from PIL import Image
 from plone import api
+from plone.memoize import forever
 from Products.Five.browser import BrowserView
 from weasyprint import HTML
 from zExceptions import NotFound
 from zope.interface import implementer
 from zope.publisher.interfaces import IPublishTraverse
-from plone.memoize import forever
+
+import base64
 import csv
+import imghdr
 import importlib.resources
 import logging
 import re
-from PIL import Image
-import base64
-import imghdr
+
 
 logger = logging.getLogger(__name__)
 
 fontools_logger = logging.getLogger("fontTools.subset")
 fontools_logger.setLevel(logging.WARNING)
 
 
-# @forever.memoize
+@forever.memoize
 def image_to_html(input_string):
     """
     Convert image data to a base64 string formatted for HTML.

src/iosanita/contenttypes/browser/searchblock.py

@@ -138,7 +138,6 @@ def get_data(self):
         # 4. fare la ricerca
         # 5. fare export in csv/pdf a seconda del formato
         """
-
         # 2. Get columns, base filters and sorting
         columns = self.block_data.get("columns", [])
 
@@ -195,13 +194,14 @@ def get_data(self):
         # XXX: consideriamo però che senza usare il serializzatore un utente potrebbe
         #      chiedere qualsiasi atttributo degli oggetti, senza un controllo fine
         #      sullo schema
-        fullobjects = True
-        self.request.form["b_size"] = 9999
-        results = getMultiAdapter((results, self.request), ISerializeToJson)(
-            fullobjects=fullobjects
-        )
-        for obj in results["items"]:
-            yield [obj["title"]] + [obj.get(c["field"]) for c in columns]
+        if results:
+            fullobjects = True
+            self.request.form["b_size"] = 9999
+            results = getMultiAdapter((results, self.request), ISerializeToJson)(
+                fullobjects=fullobjects
+            )
+            for obj in results["items"]:
+                yield [obj["title"]] + [obj.get(c["field"]) for c in columns]
 
     def get_columns(self, data):
         # Il titolo va aggiunto di default come prima colonna ?

src/iosanita/contenttypes/tests/test_pdf_csv_export.py

@@ -0,0 +1,124 @@
+from ..testing import INTEGRATION_TESTING
+from plone import api
+from plone.app.testing import login
+from plone.app.testing import logout
+from plone.app.testing import setRoles
+from plone.app.testing import TEST_USER_ID
+from plone.app.testing import TEST_USER_NAME
+from plone.testing.z2 import Browser
+from transaction import commit
+from zope.component import getMultiAdapter
+
+import csv
+import unittest
+import uuid
+
+
+class TestExport(unittest.TestCase):
+    """Test PDF export functionality"""
+
+    layer = INTEGRATION_TESTING
+
+    def setUp(self):
+        self.app = self.layer["app"]
+        self.portal = self.layer["portal"]
+        self.request = self.layer["request"]
+        self.portal_url = self.portal.absolute_url()
+        setRoles(self.portal, TEST_USER_ID, ["Manager"])
+        login(self.portal, TEST_USER_NAME)
+
+        self.doc = api.content.create(
+            container=self.portal,
+            type="Document",
+            id="test-document",
+            title="Test Document",
+            description="This is a test document",
+        )
+        api.content.transition(self.doc, to_state="published")
+        commit()
+
+        self.browser = Browser(self.app)
+
+    def tearDown(self):
+        api.content.delete(self.doc)
+        commit()
+        logout()
+
+    def test_pdf_export_view_exists(self):
+        """Test that the PDF export view is registered and accessible
+
+        TODO: view must be accessible only for IExportViewDownload context
+        """
+        view = getMultiAdapter((self.doc, self.request), name="export_pdf")
+        self.assertTrue(view is not None)
+
+    def test_searchblock_pdf(self):
+        block_id = uuid.uuid4().hex
+        self.doc.blocks = {
+            block_id: {
+                "@type": "search",
+                "query": {
+                    "query": [
+                        {
+                            "i": "portal_type",
+                            "o": "plone.app.querystring.operation.selection.is",
+                            "v": "Document",
+                        }
+                    ]
+                },
+            }
+        }
+        self.doc.blocks_layout = {"items": [block_id]}
+        commit()
+
+        self.browser.open(
+            f"{self.doc.absolute_url()}/searchblock/@@download/{block_id}.pdf"
+        )
+
+        # Check response headers
+        self.assertEqual(self.browser.headers["Content-Type"], "application/pdf")
+        self.assertIn(
+            "attachment;filename=", self.browser.headers["Content-Disposition"]
+        )
+        # Check that the response is not empty and looks like a PDF
+        self.assertTrue(b"%PDF-" in self.browser.contents[:10])
+
+        # TODO
+        # api.portal.get_tool("portal_transforms").convertTo("plain/text", self.browser.contents, mimetype="application/pdf")
+        # oppure con pypdf
+
+    def test_searchblock_csv(self):
+        block_id = uuid.uuid4().hex
+        self.doc.blocks = {
+            block_id: {
+                "@type": "search",
+                "query": {
+                    "query": [
+                        {
+                            "i": "portal_type",
+                            "o": "plone.app.querystring.operation.selection.is",
+                            "v": "Document",
+                        }
+                    ]
+                },
+            }
+        }
+        self.doc.blocks_layout = {"items": [block_id]}
+        commit()
+
+        self.browser.open(
+            f"{self.doc.absolute_url()}/searchblock/@@download/{block_id}.csv"
+        )
+
+        # Check response headers
+        self.assertEqual(
+            self.browser.headers["Content-Type"], "text/csv; charset=utf-8-sig"
+        )
+        self.assertIn(
+            "attachment;filename=", self.browser.headers["Content-Disposition"]
+        )
+
+        # Check that the response is a CSV with expected data
+        reader = csv.DictReader(self.browser.contents)
+        self.assertEqual(reader.fieldnames, ["Titolo"])
+        self.assertEqual([row for row in reader], [{"Titolo": "Test Document"}])