mirror of
https://github.com/qgis/QGIS.git
synced 2025-10-21 00:07:33 -04:00
73 lines
1.8 KiB
Python
73 lines
1.8 KiB
Python
# coding=utf-8
|
|
""""Test for postgres layer metadata provider
|
|
|
|
.. 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__ = 'elpaso@itopen.it'
|
|
__date__ = '2022-08-19'
|
|
__copyright__ = 'Copyright 2022, ItOpen'
|
|
|
|
import os
|
|
|
|
from qgis.core import (
|
|
QgsVectorLayer,
|
|
QgsProviderRegistry,
|
|
)
|
|
|
|
from qgis.testing import unittest
|
|
from qgslayermetadataprovidertestbase import LayerMetadataProviderTestBase
|
|
|
|
|
|
class TestPostgresLayerMetadataProvider(unittest.TestCase, LayerMetadataProviderTestBase):
|
|
|
|
def getMetadataProviderId(self) -> str:
|
|
|
|
return 'postgres'
|
|
|
|
def getLayer(self):
|
|
|
|
return QgsVectorLayer('{} type=Point table="qgis_test"."someData" (geom) sql='.format(self.getConnectionUri()), "someData", 'postgres')
|
|
|
|
def getConnectionUri(self) -> str:
|
|
|
|
dbconn = 'service=qgis_test'
|
|
|
|
if 'QGIS_PGTEST_DB' in os.environ:
|
|
dbconn = os.environ['QGIS_PGTEST_DB']
|
|
|
|
return dbconn
|
|
|
|
def clearMetadataTable(self):
|
|
|
|
self.conn.execSql('DROP TABLE IF EXISTS qgis_test.qgis_layer_metadata')
|
|
|
|
def setUp(self):
|
|
|
|
super().setUp()
|
|
|
|
dbconn = 'service=qgis_test'
|
|
|
|
if 'QGIS_PGTEST_DB' in os.environ:
|
|
dbconn = os.environ['QGIS_PGTEST_DB']
|
|
|
|
md = QgsProviderRegistry.instance().providerMetadata('postgres')
|
|
conn = md.createConnection(self.getConnectionUri(), {})
|
|
conn.setConfiguration({'metadataInDatabase': True})
|
|
conn.store('PG Metadata Enabled Connection')
|
|
self.conn = conn
|
|
self.clearMetadataTable()
|
|
|
|
def tearDown(self):
|
|
|
|
super().tearDown()
|
|
self.clearMetadataTable()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|