Processing - Tests - Add Ogr2Ogr test with GeoPackage support

rldhont · mdouchin · commit 781d0e11217c · 2018-10-24T15:43:44.000+02:00
diff --git a/python/plugins/processing/tests/GdalAlgorithmsTest.py b/python/plugins/processing/tests/GdalAlgorithmsTest.py
@@ -27,15 +27,18 @@
 
 import AlgorithmsTestBase
 from processing.algs.gdal.ogr2ogrtopostgis import Ogr2OgrToPostGis
+from processing.algs.gdal.ogr2ogr import Ogr2Ogr
 
 import nose2
 import shutil
-
+import os
 from qgis.testing import (
     start_app,
     unittest
 )
 
+testDataPath = os.path.join(os.path.dirname(__file__), 'testdata')
+
 
 class TestGdalAlgorithms(unittest.TestCase, AlgorithmsTestBase.AlgorithmsTest):
 
@@ -54,6 +57,29 @@ def tearDownClass(cls):
     def test_definition_file(self):
         return 'gdal_algorithm_tests.yaml'
 
+    def testOgr2Ogr(self):
+        source = os.path.join(testDataPath, 'polys.gml')
+
+        alg = Ogr2Ogr()
+        alg.setParameterValue('INPUT_LAYER', source)
+        alg.setParameterValue('FORMAT', 0)
+        alg.setOutputValue('OUTPUT_LAYER', 'd:/temp/check.shp')
+        self.assertEqual(
+            alg.getConsoleCommands(),
+            ['ogr2ogr',
+             '-f "ESRI Shapefile" "d:/temp/check.shp" ' +
+             source + ' polys2'])
+
+        alg = Ogr2Ogr()
+        alg.setParameterValue('INPUT_LAYER', source)
+        alg.setParameterValue('FORMAT', 22)
+        alg.setOutputValue('OUTPUT_LAYER', 'd:/temp/check.gpkg')
+        self.assertEqual(
+            alg.getConsoleCommands(),
+            ['ogr2ogr',
+             '-f GPKG "d:/temp/check.gpkg" ' +
+             source + ' polys2'])
+
 
 class TestGdalOgr2OgrToPostgis(unittest.TestCase):
 
