Add a unit test for exporting reports

qgis · Jan 5, 2018 · 57628fa · 57628fa
1 parent 767075a
commit 57628fa
Show file tree

Hide file tree

Showing 3 changed files with 52 additions and 1 deletion.
diff --git a/tests/src/python/test_qgslayoutexporter.py b/tests/src/python/test_qgslayoutexporter.py
@@ -37,7 +37,8 @@
                        QgsVectorLayer,
                        QgsCoordinateReferenceSystem,
                        QgsPrintLayout,
-                       QgsSingleSymbolRenderer)
+                       QgsSingleSymbolRenderer,
+                       QgsReport)
 from qgis.PyQt.QtCore import QSize, QSizeF, QDir, QRectF, Qt
 from qgis.PyQt.QtGui import QImage, QPainter
 from qgis.PyQt.QtSvg import QSvgRenderer, QSvgGenerator
@@ -726,6 +727,56 @@ def testIteratorToPdf(self):
         pdfToPng(pdf_path, rendered_page_4, dpi=80, page=4)
         self.assertTrue(os.path.exists(rendered_page_4))
 
+    def testExportReport(self):
+        p = QgsProject()
+        r = QgsReport()
+
+        # add a header
+        r.setHeaderEnabled(True)
+        report_header = QgsLayout(p)
+        report_header.initializeDefaults()
+        item1 = QgsLayoutItemShape(report_header)
+        item1.attemptSetSceneRect(QRectF(10, 20, 100, 150))
+        fill = QgsSimpleFillSymbolLayer()
+        fill_symbol = QgsFillSymbol()
+        fill_symbol.changeSymbolLayer(0, fill)
+        fill.setColor(Qt.green)
+        fill.setStrokeStyle(Qt.NoPen)
+        item1.setSymbol(fill_symbol)
+        report_header.addItem(item1)
+
+        r.setHeader(report_header)
+
+        # add a footer
+        r.setFooterEnabled(True)
+        report_footer = QgsLayout(p)
+        report_footer.initializeDefaults()
+        item2 = QgsLayoutItemShape(report_footer)
+        item2.attemptSetSceneRect(QRectF(10, 20, 100, 150))
+        item2.attemptMove(QgsLayoutPoint(10, 20))
+        fill = QgsSimpleFillSymbolLayer()
+        fill_symbol = QgsFillSymbol()
+        fill_symbol.changeSymbolLayer(0, fill)
+        fill.setColor(Qt.cyan)
+        fill.setStrokeStyle(Qt.NoPen)
+        item2.setSymbol(fill_symbol)
+        report_footer.addItem(item2)
+
+        r.setFooter(report_footer)
+
+        # setup settings
+        settings = QgsLayoutExporter.ImageExportSettings()
+        settings.dpi = 80
+
+        report_path = os.path.join(self.basetestpath, 'test_report')
+        result, error = QgsLayoutExporter.exportToImage(r, report_path, 'png', settings)
+        self.assertEqual(result, QgsLayoutExporter.Success, error)
+
+        page1_path = os.path.join(self.basetestpath, 'test_report_0001.png')
+        self.assertTrue(self.checkImage('report_page1', 'report_page1', page1_path))
+        page2_path = os.path.join(self.basetestpath, 'test_report_0002.png')
+        self.assertTrue(self.checkImage('report_page2', 'report_page2', page2_path))
+
 
 if __name__ == '__main__':
     unittest.main()
diff --git a/...r/expected_layoutexporter_report_page1/expected_layoutexporter_report_page1.png b/...r/expected_layoutexporter_report_page1/expected_layoutexporter_report_page1.png
diff --git a/...r/expected_layoutexporter_report_page2/expected_layoutexporter_report_page2.png b/...r/expected_layoutexporter_report_page2/expected_layoutexporter_report_page2.png