/tests/src/python/test_qgsdatabaseschemamodel.py

https://github.com/ricardogsilva/Quantum-GIS · Python · 211 lines · 174 code · 23 blank · 14 comment · 12 complexity · da94294fcc22a730cb7902a21108c5b1 MD5 · raw file 

    

  1. # -*- coding: utf-8 -*-
    2. 

  2. """QGIS Unit tests for QgsDatabaseSchemaModel
    3. 

    4. 

  4. .. note:: This program is free software; you can redistribute it and/or modify
    5. 

  5. it under the terms of the GNU General Public License as published by
    6. 

  6. the Free Software Foundation; either version 2 of the License, or
    7. 

  7. (at your option) any later version.
    8. 

  8. 

  9. """
    10. 

  10. __author__ = 'Nyall Dawson'
    11. 

  11. __date__ = '07/03/2020'
    12. 

  12. __copyright__ = 'Copyright 2020, The QGIS Project'
    13. 

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

  14. __revision__ = '$Format:%H$'
    15. 

  15. 

  16. import os
    17. 

  17. from qgis.core import (
    18. 

  18.     QgsDatabaseSchemaModel,
    19. 

  19.     QgsProviderRegistry,
    20. 

  20. )
    21. 

  21. from qgis.PyQt.QtCore import (
    22. 

  22.     QCoreApplication,
    23. 

  23.     QModelIndex,
    24. 

  24.     Qt
    25. 

  25. )
    26. 

  26. from qgis.testing import unittest, start_app
    27. 

  27. 

  28. 

  29. class TestPyQgsDatabaseSchemaModel(unittest.TestCase):
    30. 

  30. 

  31.     # Provider test cases must define the string URI for the test
    32. 

  32.     uri = ''
    33. 

  33.     # Provider test cases must define the provider name (e.g. "postgres" or "ogr")
    34. 

  34.     providerKey = 'postgres'
    35. 

  35. 

  36.     @classmethod
    37. 

  37.     def setUpClass(cls):
    38. 

  38.         """Run before all tests"""
    39. 

  39. 

  40.         QCoreApplication.setOrganizationName("QGIS_Test")
    41. 

  41.         QCoreApplication.setOrganizationDomain(cls.__name__)
    42. 

  42.         QCoreApplication.setApplicationName(cls.__name__)
    43. 

  43.         start_app()
    44. 

  44.         cls.postgres_conn = "service='qgis_test'"
    45. 

  45.         if 'QGIS_PGTEST_DB' in os.environ:
    46. 

  46.             cls.postgres_conn = os.environ['QGIS_PGTEST_DB']
    47. 

  47.         cls.uri = cls.postgres_conn + ' sslmode=disable'
    48. 

  48. 

  49.     def testModel(self):
    50. 

  50.         conn = QgsProviderRegistry.instance().providerMetadata('postgres').createConnection(self.uri, {})
    51. 

  51.         self.assertTrue(conn)
    52. 

  52.         model = QgsDatabaseSchemaModel(conn)
    53. 

  53.         self.assertGreaterEqual(model.rowCount(), 3)
    54. 

  54.         old_count = model.rowCount()
    55. 

  55.         self.assertEqual(model.columnCount(), 1)
    56. 

  56.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    57. 

  57.         self.assertIn('public', schemas)
    58. 

  58.         self.assertIn('CamelCaseSchema', schemas)
    59. 

  59.         self.assertIn('qgis_test', schemas)
    60. 

  60.         self.assertEqual(model.data(model.index(schemas.index('qgis_test'), 0, QModelIndex()), Qt.ToolTipRole), 'qgis_test')
    61. 

  61.         self.assertIsNone(model.data(model.index(model.rowCount(), 0, QModelIndex()), Qt.DisplayRole))
    62. 

  62. 

  63.         model.refresh()
    64. 

  64.         self.assertEqual(model.rowCount(), old_count)
    65. 

  65. 

  66.         conn.createSchema('myNewSchema')
    67. 

  67.         self.assertEqual(model.rowCount(), old_count)
    68. 

  68.         model.refresh()
    69. 

  69.         self.assertEqual(model.rowCount(), old_count + 1)
    70. 

  70.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    71. 

  71.         self.assertIn('public', schemas)
    72. 

  72.         self.assertIn('CamelCaseSchema', schemas)
    73. 

  73.         self.assertIn('qgis_test', schemas)
    74. 

  74.         self.assertIn('myNewSchema', schemas)
    75. 

  75. 

  76.         conn.createSchema('myNewSchema2')
    77. 

  77.         conn.createSchema('myNewSchema3')
    78. 

  78.         model.refresh()
    79. 

  79.         self.assertEqual(model.rowCount(), old_count + 3)
    80. 

  80.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    81. 

  81.         self.assertIn('public', schemas)
    82. 

  82.         self.assertIn('CamelCaseSchema', schemas)
    83. 

  83.         self.assertIn('qgis_test', schemas)
    84. 

  84.         self.assertIn('myNewSchema', schemas)
    85. 

  85.         self.assertIn('myNewSchema2', schemas)
    86. 

  86.         self.assertIn('myNewSchema3', schemas)
    87. 

  87. 

  88.         conn.createSchema('myNewSchema4')
    89. 

  89.         conn.dropSchema('myNewSchema2')
    90. 

  90.         conn.dropSchema('myNewSchema')
    91. 

  91.         model.refresh()
    92. 

  92.         self.assertEqual(model.rowCount(), old_count + 2)
    93. 

  93.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    94. 

  94.         self.assertIn('public', schemas)
    95. 

  95.         self.assertIn('CamelCaseSchema', schemas)
    96. 

  96.         self.assertIn('qgis_test', schemas)
    97. 

  97.         self.assertNotIn('myNewSchema', schemas)
    98. 

  98.         self.assertNotIn('myNewSchema2', schemas)
    99. 

  99.         self.assertIn('myNewSchema3', schemas)
    100. 

  100.         self.assertIn('myNewSchema4', schemas)
    101. 

  101. 

  102.         conn.dropSchema('myNewSchema3')
    103. 

  103.         conn.dropSchema('myNewSchema4')
    104. 

  104.         model.refresh()
    105. 

  105.         self.assertEqual(model.rowCount(), old_count)
    106. 

  106.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    107. 

  107.         self.assertIn('public', schemas)
    108. 

  108.         self.assertIn('CamelCaseSchema', schemas)
    109. 

  109.         self.assertIn('qgis_test', schemas)
    110. 

  110.         self.assertNotIn('myNewSchema3', schemas)
    111. 

  111.         self.assertNotIn('myNewSchema4', schemas)
    112. 

  112. 

  113.     def test_model_allow_empty(self):
    114. 

  114.         """Test model with empty entry"""
    115. 

  115.         conn = QgsProviderRegistry.instance().providerMetadata('postgres').createConnection(self.uri, {})
    116. 

  116.         self.assertTrue(conn)
    117. 

  117.         model = QgsDatabaseSchemaModel(conn)
    118. 

  118.         self.assertGreaterEqual(model.rowCount(), 3)
    119. 

  119.         old_count = model.rowCount()
    120. 

  120.         model.setAllowEmptySchema(True)
    121. 

  121.         self.assertEqual(model.rowCount(), old_count + 1)
    122. 

  122. 

  123.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    124. 

  124.         self.assertIn('public', schemas)
    125. 

  125.         self.assertIn('CamelCaseSchema', schemas)
    126. 

  126.         self.assertIn('qgis_test', schemas)
    127. 

  127.         self.assertFalse(model.data(model.index(0, 0, QModelIndex()), Qt.DisplayRole))
    128. 

  128.         self.assertTrue(model.data(model.index(0, 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    129. 

  129.         self.assertFalse(model.data(model.index(schemas.index('qgis_test'), 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    130. 

  130.         self.assertIsNone(model.data(model.index(model.rowCount(), 0, QModelIndex()), Qt.DisplayRole))
    131. 

  131. 

  132.         model.refresh()
    133. 

  133.         self.assertEqual(model.rowCount(), old_count + 1)
    134. 

  134. 

  135.         conn.createSchema('myNewSchema')
    136. 

  136.         self.assertEqual(model.rowCount(), old_count + 1)
    137. 

  137.         model.refresh()
    138. 

  138.         self.assertEqual(model.rowCount(), old_count + 2)
    139. 

  139.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    140. 

  140.         self.assertIn('public', schemas)
    141. 

  141.         self.assertIn('CamelCaseSchema', schemas)
    142. 

  142.         self.assertIn('qgis_test', schemas)
    143. 

  143.         self.assertIn('myNewSchema', schemas)
    144. 

  144.         self.assertFalse(model.data(model.index(0, 0, QModelIndex()), Qt.DisplayRole))
    145. 

  145.         self.assertTrue(model.data(model.index(0, 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    146. 

  146.         self.assertFalse(model.data(model.index(schemas.index('qgis_test'), 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    147. 

  147. 

  148.         model.setAllowEmptySchema(False)
    149. 

  149.         self.assertEqual(model.rowCount(), old_count + 1)
    150. 

  150.         self.assertTrue(model.data(model.index(0, 0, QModelIndex()), Qt.DisplayRole))
    151. 

  151.         self.assertFalse(model.data(model.index(0, 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    152. 

  152.         model.setAllowEmptySchema(True)
    153. 

  153.         self.assertEqual(model.rowCount(), old_count + 2)
    154. 

  154.         self.assertFalse(model.data(model.index(0, 0, QModelIndex()), Qt.DisplayRole))
    155. 

  155.         self.assertTrue(model.data(model.index(0, 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    156. 

  156.         self.assertFalse(model.data(model.index(schemas.index('qgis_test'), 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    157. 

  157. 

  158.         conn.createSchema('myNewSchema2')
    159. 

  159.         conn.createSchema('myNewSchema3')
    160. 

  160.         model.refresh()
    161. 

  161.         self.assertEqual(model.rowCount(), old_count + 4)
    162. 

  162.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    163. 

  163.         self.assertIn('public', schemas)
    164. 

  164.         self.assertIn('CamelCaseSchema', schemas)
    165. 

  165.         self.assertIn('qgis_test', schemas)
    166. 

  166.         self.assertIn('myNewSchema', schemas)
    167. 

  167.         self.assertIn('myNewSchema2', schemas)
    168. 

  168.         self.assertIn('myNewSchema3', schemas)
    169. 

  169.         self.assertFalse(model.data(model.index(0, 0, QModelIndex()), Qt.DisplayRole))
    170. 

  170.         self.assertTrue(model.data(model.index(0, 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    171. 

  171.         self.assertFalse(model.data(model.index(schemas.index('qgis_test'), 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    172. 

  172. 

  173.         conn.createSchema('myNewSchema4')
    174. 

  174.         conn.dropSchema('myNewSchema2')
    175. 

  175.         conn.dropSchema('myNewSchema')
    176. 

  176.         model.refresh()
    177. 

  177.         self.assertEqual(model.rowCount(), old_count + 3)
    178. 

  178.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    179. 

  179.         self.assertIn('public', schemas)
    180. 

  180.         self.assertIn('CamelCaseSchema', schemas)
    181. 

  181.         self.assertIn('qgis_test', schemas)
    182. 

  182.         self.assertNotIn('myNewSchema', schemas)
    183. 

  183.         self.assertNotIn('myNewSchema2', schemas)
    184. 

  184.         self.assertIn('myNewSchema3', schemas)
    185. 

  185.         self.assertIn('myNewSchema4', schemas)
    186. 

  186.         self.assertFalse(model.data(model.index(0, 0, QModelIndex()), Qt.DisplayRole))
    187. 

  187.         self.assertTrue(model.data(model.index(0, 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    188. 

  188.         self.assertFalse(model.data(model.index(schemas.index('qgis_test'), 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    189. 

  189. 

  190.         conn.dropSchema('myNewSchema3')
    191. 

  191.         conn.dropSchema('myNewSchema4')
    192. 

  192.         model.refresh()
    193. 

  193.         self.assertEqual(model.rowCount(), old_count + 1)
    194. 

  194.         schemas = [model.data(model.index(r, 0, QModelIndex()), Qt.DisplayRole) for r in range(model.rowCount())]
    195. 

  195.         self.assertIn('public', schemas)
    196. 

  196.         self.assertIn('CamelCaseSchema', schemas)
    197. 

  197.         self.assertIn('qgis_test', schemas)
    198. 

  198.         self.assertNotIn('myNewSchema3', schemas)
    199. 

  199.         self.assertNotIn('myNewSchema4', schemas)
    200. 

  200.         self.assertFalse(model.data(model.index(0, 0, QModelIndex()), Qt.DisplayRole))
    201. 

  201.         self.assertTrue(model.data(model.index(0, 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    202. 

  202.         self.assertFalse(model.data(model.index(schemas.index('qgis_test'), 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    203. 

  203. 

  204.         model.setAllowEmptySchema(False)
    205. 

  205.         self.assertEqual(model.rowCount(), old_count)
    206. 

  206.         self.assertTrue(model.data(model.index(0, 0, QModelIndex()), Qt.DisplayRole))
    207. 

  207.         self.assertFalse(model.data(model.index(0, 0, QModelIndex()), QgsDatabaseSchemaModel.RoleEmpty))
    208. 

  208. 

  209. 

  210. if __name__ == '__main__':
    211. 

  211.     unittest.main()
    212.