Test that PostGIS query can be passed as both unicode and string literal

See https://issues.qgis.org/issues/16833
qgis · Jul 19, 2017 · 9efd666 · 9efd666
1 parent 90d4768
commit 9efd666
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/python/plugins/db_manager/db_plugins/postgis/plugin_test.py b/python/plugins/db_manager/db_plugins/postgis/plugin_test.py
@@ -132,11 +132,16 @@ def test_unicodeInQuery(self):
         obj = QObject() # needs to be kept alive
         database = PGDatabase(obj, QgsDataSourceUri())
         self.assertIsInstance(database, PGDatabase)
-        sql = "SELECT 'é'::text"
-        res = database.sqlResultModel(sql, obj)
+        # SQL as string literal
+        res = database.sqlResultModel("SELECT 'é'::text", obj)
         self.assertIsInstance(res, PGSqlResultModel)
         dat = res.getData(0, 0)
-        self.assertEqual(dat, "é")
+        self.assertEqual(dat, u"é")
+        # SQL as unicode literal
+        res = database.sqlResultModel(u"SELECT 'é'::text", obj)
+        self.assertIsInstance(res, PGSqlResultModel)
+        dat = res.getData(0, 0)
+        self.assertEqual(dat, u"é")
 
 
 if __name__ == '__main__':