Tests

qgis · Aug 17, 2019 · c23b1c6 · c23b1c6
1 parent 65b4496
commit c23b1c6
Show file tree

Hide file tree

Showing 3 changed files with 180 additions and 0 deletions.
diff --git a/src/core/qgsabstractgeopdfexporter.h b/src/core/qgsabstractgeopdfexporter.h
@@ -194,6 +194,7 @@ class CORE_EXPORT QgsAbstractGeoPdfExporter
     QString createCompositionXml( const QList< QgsAbstractGeoPdfExporter::ComponentLayerDetail > &components );
 
     friend class TestQgsLayoutGeoPdfExport;
+    friend class TestQgsGeoPdfExport;
 };
 
 #endif //QGSABSTRACTGEOPDFEXPORTER_H

diff --git a/tests/src/core/CMakeLists.txt b/tests/src/core/CMakeLists.txt
@@ -117,6 +117,7 @@ SET(TESTS
  testqgsgdalutils.cpp
  testqgsvectorfilewriter.cpp
  testqgsfontmarker.cpp
+ testqgsgeopdfexport.cpp
  testqgsgeometryimport.cpp
  testqgsgeometry.cpp
  testqgsgeometryutils.cpp

diff --git a/tests/src/core/testqgsgeopdfexport.cpp b/tests/src/core/testqgsgeopdfexport.cpp
@@ -0,0 +1,178 @@
+/***************************************************************************
+                         testqgsgeopdfexport.cpp
+                         ----------------------
+    begin                : August 2019
+    copyright            : (C) 2019 by Nyall Dawson
+    email                : nyall dot dawson at gmail dot com
+ ***************************************************************************/
+
+/***************************************************************************
+ *                                                                         *
+ *   This program is free software; you can redistribute it and/or modify  *
+ *   it under the terms of the GNU General Public License as published by  *
+ *   the Free Software Foundation; either version 2 of the License, or     *
+ *   (at your option) any later version.                                   *
+ *                                                                         *
+ ***************************************************************************/
+
+#include "qgsapplication.h"
+#include "qgslayout.h"
+#include "qgsmultirenderchecker.h"
+#include "qgslayoutitemmap.h"
+#include "qgsvectorlayer.h"
+#include "qgsproject.h"
+#include "qgsabstractgeopdfexporter.h"
+#include <QObject>
+#include "qgstest.h"
+
+class TestGeoPdfExporter : public QgsAbstractGeoPdfExporter
+{
+
+  private:
+
+    VectorComponentDetail componentDetailForLayerId( const QString &layerId ) override
+    {
+      Q_UNUSED( layerId );
+      VectorComponentDetail detail;
+      detail.mapLayerId = layerId;
+      detail.name = QStringLiteral( "name %1" ).arg( layerId );
+      detail.displayAttribute = QStringLiteral( "attr %1" ).arg( layerId );
+      return detail;
+    }
+
+};
+
+class TestQgsGeoPdfExport : public QObject
+{
+    Q_OBJECT
+
+  public:
+    TestQgsGeoPdfExport() = default;
+
+  private slots:
+    void initTestCase();// will be called before the first testfunction is executed.
+    void cleanupTestCase();// will be called after the last testfunction was executed.
+    void init();// will be called before each testfunction is executed.
+    void cleanup();// will be called after every testfunction.
+    void testCollectingFeatures();
+
+  private:
+
+    QString mReport;
+};
+
+void TestQgsGeoPdfExport::initTestCase()
+{
+  QgsApplication::init();
+  QgsApplication::initQgis();
+
+  mReport = QStringLiteral( "<h1>GeoPDF Export Tests</h1>\n" );
+}
+
+void TestQgsGeoPdfExport::cleanupTestCase()
+{
+  QString myReportFile = QDir::tempPath() + "/qgistest.html";
+  QFile myFile( myReportFile );
+  if ( myFile.open( QIODevice::WriteOnly | QIODevice::Append ) )
+  {
+    QTextStream myQTextStream( &myFile );
+    myQTextStream << mReport;
+    myFile.close();
+  }
+
+  QgsApplication::exitQgis();
+}
+
+void TestQgsGeoPdfExport::init()
+{
+}
+
+void TestQgsGeoPdfExport::cleanup()
+{
+}
+
+void TestQgsGeoPdfExport::testCollectingFeatures()
+{
+  if ( !QgsAbstractGeoPdfExporter::geoPDFCreationAvailable() )
+  {
+    QSKIP( "This test requires GeoPDF creation abilities", SkipSingle );
+  }
+
+  TestGeoPdfExporter geoPdfExporter;
+  // no features, no crash
+  QVERIFY( geoPdfExporter.saveTemporaryLayers() );
+  QCOMPARE( geoPdfExporter.mVectorComponents.count(), 0 );
+
+  QgsFields fields;
+  fields.append( QgsField( QStringLiteral( "a1" ), QVariant::Int ) );
+  fields.append( QgsField( QStringLiteral( "a2" ), QVariant::Int ) );
+  QgsFeature f( fields );
+
+  f.setAttributes( QgsAttributes() << 1 << 2 );
+  f.setGeometry( QgsGeometry( new QgsPoint( 1, 2 ) ) );
+  QgsGeometry renderedBounds( QgsGeometry::fromRect( QgsRectangle( 1, 10, 6, 20 ) ) );
+  geoPdfExporter.pushRenderedFeature( QStringLiteral( "layer1" ), QgsAbstractGeoPdfExporter::RenderedFeature( f, renderedBounds ) );
+  f.setAttributes( QgsAttributes() << 11 << 22 );
+  f.setGeometry( QgsGeometry( new QgsPoint( 3, 4 ) ) );
+  renderedBounds = QgsGeometry::fromRect( QgsRectangle( 2, 10, 7, 20 ) );
+  geoPdfExporter.pushRenderedFeature( QStringLiteral( "layer1" ), QgsAbstractGeoPdfExporter::RenderedFeature( f, renderedBounds ) );
+  f.setAttributes( QgsAttributes() << 31 << 32 );
+  f.setGeometry( QgsGeometry( new QgsPoint( 4, 5 ) ) );
+  renderedBounds = QgsGeometry::fromWkt( QStringLiteral( "LineString(1 1, 2 2)" ) );
+  geoPdfExporter.pushRenderedFeature( QStringLiteral( "layer2" ), QgsAbstractGeoPdfExporter::RenderedFeature( f, renderedBounds ) );
+
+  QVERIFY( geoPdfExporter.saveTemporaryLayers() );
+  QCOMPARE( geoPdfExporter.mVectorComponents.count(), 2 );
+  QgsAbstractGeoPdfExporter::VectorComponentDetail component;
+  for ( const auto &it : qgis::as_const( geoPdfExporter.mVectorComponents ) )
+  {
+    if ( it.mapLayerId == QStringLiteral( "layer1" ) )
+    {
+      component = it;
+      break;
+    }
+  }
+  QCOMPARE( component.mapLayerId, QStringLiteral( "layer1" ) );
+  QCOMPARE( component.name, QStringLiteral( "name layer1" ) );
+  // check that temporary layers were correctly written
+  std::unique_ptr< QgsVectorLayer > layer = qgis::make_unique< QgsVectorLayer >( QStringLiteral( "%1|layerName=%2" ).arg( component.sourceVectorPath, component.sourceVectorLayer ), QStringLiteral( "layer" ), QStringLiteral( "ogr" ) );
+  QVERIFY( layer->isValid() );
+  QCOMPARE( layer->featureCount(), 2 );
+  QCOMPARE( layer->wkbType(), QgsWkbTypes::Polygon );
+  QCOMPARE( layer->fields().at( 1 ).name(), QStringLiteral( "a1" ) );
+  QCOMPARE( layer->fields().at( 2 ).name(), QStringLiteral( "a2" ) );
+  QgsFeatureIterator it = layer->getFeatures();
+  QVERIFY( it.nextFeature( f ) );
+  QCOMPARE( f.attributes().at( 1 ).toInt(), 1 );
+  QCOMPARE( f.attributes().at( 2 ).toInt(), 2 );
+  QCOMPARE( f.geometry().asWkt(), QStringLiteral( "Polygon ((1 10, 6 10, 6 20, 1 20, 1 10))" ) );
+  QVERIFY( it.nextFeature( f ) );
+  QCOMPARE( f.attributes().at( 1 ).toInt(), 11 );
+  QCOMPARE( f.attributes().at( 2 ).toInt(), 22 );
+  QCOMPARE( f.geometry().asWkt(), QStringLiteral( "Polygon ((2 10, 7 10, 7 20, 2 20, 2 10))" ) );
+
+  for ( const auto &it : qgis::as_const( geoPdfExporter.mVectorComponents ) )
+  {
+    if ( it.mapLayerId == QStringLiteral( "layer2" ) )
+    {
+      component = it;
+      break;
+    }
+  }
+  QCOMPARE( component.mapLayerId, QStringLiteral( "layer2" ) );
+  QCOMPARE( component.name, QStringLiteral( "name layer2" ) );
+  layer = qgis::make_unique< QgsVectorLayer >( QStringLiteral( "%1|layerName=%2" ).arg( component.sourceVectorPath, component.sourceVectorLayer ), QStringLiteral( "layer" ), QStringLiteral( "ogr" ) );
+  QVERIFY( layer->isValid() );
+  QCOMPARE( layer->featureCount(), 1 );
+  QCOMPARE( layer->wkbType(), QgsWkbTypes::LineString );
+  QCOMPARE( layer->fields().at( 1 ).name(), QStringLiteral( "a1" ) );
+  QCOMPARE( layer->fields().at( 2 ).name(), QStringLiteral( "a2" ) );
+  it = layer->getFeatures();
+  QVERIFY( it.nextFeature( f ) );
+  QCOMPARE( f.attributes().at( 1 ).toInt(), 31 );
+  QCOMPARE( f.attributes().at( 2 ).toInt(), 32 );
+  QCOMPARE( f.geometry().asWkt(), QStringLiteral( "LineString (1 1, 2 2)" ) );
+}
+
+QGSTEST_MAIN( TestQgsGeoPdfExport )
+#include "testqgsgeopdfexport.moc"