TG-62 upload file (template) => test

backend/tests/tools/libPdf/PyPDF2/testFusionFichiers.py

@@ -0,0 +1,19 @@
+import os
+import unittest
+from builtins import print
+from pathlib import Path
+
+from app.tools.FusionPdf import fusion_fichiers, get_pdf_from_directory
+
+
+class TestFusionTestCase(unittest.TestCase):
+    def setUp(self):
+        self.datadir = os.path.join(os.path.dirname(__file__))
+        self.list_pdf = [self.datadir + "/page1.pdf", self.datadir + "/page2.pdf"]
+
+    def test_fusion(self):
+        fusion_fichiers(self.datadir, "testFusion.pdf", self.list_pdf)
+        print(self.datadir)
+        self.assertTrue(Path(self.datadir + "/testFusion.pdf").is_file(), "Pdf fusionne inexistant")
+        self.assertTrue(len(get_pdf_from_directory(self.datadir)) > 0, "pdf non trouve")
+        os.remove(self.datadir + "/testFusion.pdf")

backend/tests/tools/libPdf/UploadPDF/testUploadPdf.py

@@ -0,0 +1,20 @@
+import os
+import unittest
+
+from werkzeug.datastructures import FileStorage
+
+from app.tools.LibPdf import upload_file
+
+
+class TestFusionTestCase(unittest.TestCase):
+    def setUp(self):
+        self.datadir = os.path.join(os.path.dirname(__file__))
+
+    def test_fusion(self):
+        with open("page1.pdf", 'rb') as fp:
+            file = FileStorage(fp)
+            upload_file(file, "upload")
+
+            # self.assertTrue(Path(self.datadir + "/testFusion.pdf").is_file(), "Pdf fusionne inexistant")
+            # self.assertTrue(len(get_pdf_from_directory(self.datadir)) > 0, "pdf non trouve")
+            # os.remove(self.datadir + "/testFusion.pdf")

backend/tests/tools/libPdf/pdfjinja/testInsertTemplate.py

@@ -0,0 +1,50 @@
+import os
+import unittest
+from io import BytesIO
+from pathlib import Path
+
+from app.tools.InsertTemplate import remplir_template
+from pdfjinja import PdfJinja
+
+
+class InsertTemplateTestCase(unittest.TestCase):
+    datadir = os.path.join(os.path.dirname(__file__))
+
+    def setUp(self):
+        pdffile = os.path.join(self.datadir, "sample.pdf")
+        self.data = {
+            'firstName': 'Renan',
+            'lastName': 'Husson',
+            'address': {
+                'street': '24 rue de la pommes',
+                'apt': 'C317',
+                'city': 'TOULOUSE',
+                'zipcode': 31000
+            },
+            'universite': 'Jean Jaures',
+            'spirit': 'Panda',
+            'evil': True,
+            'language': {
+                'french': True,
+                'esperento': True
+            }
+        }
+        self.pdfjinja = PdfJinja(pdffile)
+
+    def tearDown(self):
+        del self.data
+        del self.pdfjinja
+
+    def test_render(self):
+        remplir_template(self.datadir, "sample.pdf", self.datadir, "output.pdf", self.data)
+        output = self.pdfjinja(self.data)
+        outfile = BytesIO()
+        output.write(outfile)
+        outfile.seek(0)
+        self.assertTrue(len(outfile.read()) > 0, "Output PDF is not empty.")
+        self.assertTrue(Path(self.datadir + "/output.pdf").is_file(), "Pdf généré inexistant")
+        os.remove(self.datadir + "/output.pdf")
+
+
+if __name__ == '__main__':
+    unittest.main()