QGIS » QGIS Application

# -*- coding: utf-8 -*-

"""QGIS Unit tests for QgsVirtualLayerProvider

.. note:: 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.

"""

__author__ = 'Hugo Mercier'

__date__ = '26/01/2015'

__copyright__ = 'Copyright 2015, The QGIS Project'

# This will get replaced with a git SHA1 when you do a git archive

__revision__ = '$Format:%H$'

import qgis

from PyQt4.QtCore import *

from qgis.core import (QGis,

                       QgsVectorLayer,

                       QgsFeature,

                       QgsFeatureRequest,

                       QgsField,

                       QgsGeometry,

                       QgsPoint,

                       QgsMapLayerRegistry

                      )

from utilities import (getQgisTestApp,

                       TestCase,

                       unittest,

                       unitTestDataPath

                       )

QGISAPP, CANVAS, IFACE, PARENT = getQgisTestApp()

import os

import time

import pyspatialite.dbapi2 as db

class TestConnPool(TestCase):

    def test(self):

        # always use the same file

        fn = '/tmp/t.sqlite'

        for it in range(2):

            print it

            # make sure the file does not exist

            if os.path.exists(fn):

                os.unlink(fn)

            conn = db.connect(fn, isolation_level = None)

            # create a simple geometry table with 2 features

            cur = conn.cursor()

            cur.execute("BEGIN")

            cur.execute("SELECT InitSpatialMetadata()")

            cur.execute("CREATE TABLE t%d(id INTEGER PRIMARY KEY)" % it)

            cur.execute("SELECT AddGeometryColumn('t%d', 'geometry', 4326, 'POINT', 'XY')" % it)

            cur.execute("INSERT INTO t%d (id, geometry) VALUES(1, GeomFromText('POINT(0 0)', 4326))" % it)

            cur.execute("INSERT INTO t%d (id, geometry) VALUES(2, GeomFromText('POINT(1 1)', 4326))" % it)

            cur.execute("COMMIT")

            conn.close()

            l = QgsVectorLayer("dbname=%s table=t%d (geometry)" % (fn,it), "tt", "spatialite", False)

            assert l.isValid()

            QgsMapLayerRegistry.instance().addMapLayer(l)

            assert l.hasGeometryType()

            # iterate over features

            assert sum(f.id() for f in l.getFeatures()) == 3

            # => fails the second time it is executed : "no such table t1"

            QgsMapLayerRegistry.instance().removeMapLayer(l.id())

if __name__ == '__main__':

    unittest.main()