/tests/src/python/test_qgsbookmarkmodel.py

https://github.com/ricardogsilva/Quantum-GIS · Python · 248 lines · 204 code · 29 blank · 15 comment · 9 complexity · b6c14f86c432efaf618d12d5bfeca147 MD5 · raw file 

    

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

  2. """QGIS Unit tests for QgsBookmarkModel.
    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. 

  9. __author__ = '(C) 2019 by Nyall Dawson'
    10. 

  10. __date__ = '02/09/2019'
    11. 

  11. __copyright__ = 'Copyright 2019, The QGIS Project'
    12. 

  12. 

  13. import qgis  # NOQA
    14. 

  14. 

  15. from qgis.PyQt.QtCore import Qt, QCoreApplication, QLocale
    16. 

  16. 

  17. from qgis.core import (QgsBookmark,
    18. 

  18.                        QgsBookmarkManager,
    19. 

  19.                        QgsBookmarkManagerModel,
    20. 

  20.                        QgsProject,
    21. 

  21.                        QgsReferencedRectangle,
    22. 

  22.                        QgsRectangle,
    23. 

  23.                        QgsCoordinateReferenceSystem,
    24. 

  24.                        QgsSettings)
    25. 

  25. 

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

  27. from utilities import unitTestDataPath
    28. 

  28. 

  29. start_app()
    30. 

  30. TEST_DATA_DIR = unitTestDataPath()
    31. 

  31. 

  32. 

  33. class TestQgsBookmarkManagerModel(unittest.TestCase):
    34. 

  34. 

  35.     @classmethod
    36. 

  36.     def setUpClass(cls):
    37. 

  37.         """Run before all tests"""
    38. 

  38.         QCoreApplication.setOrganizationName("QGIS_Test")
    39. 

  39.         QCoreApplication.setOrganizationDomain("QGIS_TestQgsBookmarkManager.com")
    40. 

  40.         QCoreApplication.setApplicationName("QGIS_TestQgsBookmarkManager")
    41. 

  41.         QgsSettings().clear()
    42. 

  42.         QLocale.setDefault(QLocale(QLocale.English))
    43. 

  43.         start_app()
    44. 

  44. 

  45.     def testBookmarkModel(self):
    46. 

  46.         p = QgsProject()
    47. 

  47.         project_manager = QgsBookmarkManager.createProjectBasedManager(p)
    48. 

  48.         app_manager = QgsBookmarkManager()
    49. 

  49. 

  50.         # initially no bookmarks
    51. 

  51.         model = QgsBookmarkManagerModel(app_manager, project_manager)
    52. 

  52.         self.assertEqual(model.rowCount(), 0)
    53. 

  53.         self.assertEqual(model.columnCount(), 8)
    54. 

  54.         self.assertFalse(model.data(model.index(-1, 0)))
    55. 

  55.         self.assertFalse(model.data(model.index(1, 0)))
    56. 

  56.         self.assertFalse(model.data(model.index(0, 0)))
    57. 

  57.         self.assertFalse(model.data(model.index(0, 0), QgsBookmarkManagerModel.RoleName))
    58. 

  58. 

  59.         self.assertEqual(model.headerData(0, Qt.Horizontal), 'Name')
    60. 

  60.         self.assertEqual(model.headerData(9, Qt.Horizontal), 10)
    61. 

  61.         self.assertEqual(model.headerData(-1, Qt.Horizontal), 0)
    62. 

  62. 

  63.         self.assertFalse(model.setData(model.index(-1, 0), 4, Qt.EditRole))
    64. 

  64.         self.assertFalse(model.setData(model.index(0, 0), 4, Qt.EditRole))
    65. 

  65. 

  66.         self.assertFalse(int(model.flags(model.index(0, 0)) & Qt.ItemIsEnabled))
    67. 

  67.         self.assertFalse(int(model.flags(model.index(0, 0)) & Qt.ItemIsEditable))
    68. 

  68. 

  69.         # add some bookmarks
    70. 

  70.         b = QgsBookmark()
    71. 

  71.         b.setId('1')
    72. 

  72.         b.setGroup('group 1')
    73. 

  73.         b.setName('b1')
    74. 

  74.         b.setExtent(QgsReferencedRectangle(QgsRectangle(11, 21, 31, 41), QgsCoordinateReferenceSystem('EPSG:4326')))
    75. 

  75. 

  76.         b2 = QgsBookmark()
    77. 

  77.         b2.setId('2')
    78. 

  78.         b2.setGroup('group 2')
    79. 

  79.         b2.setName('b2')
    80. 

  80.         b2.setExtent(QgsReferencedRectangle(QgsRectangle(12, 22, 32, 42), QgsCoordinateReferenceSystem('EPSG:4326')))
    81. 

  81. 

  82.         app_manager.addBookmark(b)
    83. 

  83.         app_manager.addBookmark(b2)
    84. 

  84. 

  85.         self.assertEqual(model.rowCount(), 2)
    86. 

  86.         self.assertFalse(model.data(model.index(-1, 0)))
    87. 

  87.         self.assertEqual(model.data(model.index(0, 0)), 'b1')
    88. 

  88.         self.assertEqual(model.data(model.index(0, 1)), 'group 1')
    89. 

  89.         self.assertEqual(model.data(model.index(0, 2)), 11.0)
    90. 

  90.         self.assertEqual(model.data(model.index(0, 3)), 21.0)
    91. 

  91.         self.assertEqual(model.data(model.index(0, 4)), 31.0)
    92. 

  92.         self.assertEqual(model.data(model.index(0, 5)), 41.0)
    93. 

  93.         self.assertEqual(model.data(model.index(0, 6)), 'EPSG:4326')
    94. 

  94.         self.assertEqual(model.data(model.index(0, 7)), None)
    95. 

  95.         self.assertEqual(model.data(model.index(0, 7), Qt.CheckStateRole), Qt.Unchecked)
    96. 

  96.         self.assertEqual(model.data(model.index(0, 0), QgsBookmarkManagerModel.RoleName), 'b1')
    97. 

  97.         self.assertEqual(model.data(model.index(0, 0), QgsBookmarkManagerModel.RoleGroup), 'group 1')
    98. 

  98.         id = model.data(model.index(0, 0), QgsBookmarkManagerModel.RoleId)
    99. 

  99.         self.assertEqual(app_manager.bookmarkById(id).name(), 'b1')
    100. 

  100.         self.assertEqual(model.data(model.index(0, 0), QgsBookmarkManagerModel.RoleExtent), app_manager.bookmarkById(id).extent())
    101. 

  101. 

  102.         self.assertEqual(model.data(model.index(1, 0)), 'b2')
    103. 

  103.         self.assertEqual(model.data(model.index(1, 1)), 'group 2')
    104. 

  104.         self.assertEqual(model.data(model.index(1, 2)), 12.0)
    105. 

  105.         self.assertEqual(model.data(model.index(1, 3)), 22.0)
    106. 

  106.         self.assertEqual(model.data(model.index(1, 4)), 32.0)
    107. 

  107.         self.assertEqual(model.data(model.index(1, 5)), 42.0)
    108. 

  108.         self.assertEqual(model.data(model.index(1, 6)), 'EPSG:4326')
    109. 

  109.         self.assertEqual(model.data(model.index(1, 7)), None)
    110. 

  110.         self.assertEqual(model.data(model.index(1, 7), Qt.CheckStateRole), Qt.Unchecked)
    111. 

  111.         self.assertEqual(model.data(model.index(1, 0), QgsBookmarkManagerModel.RoleName), 'b2')
    112. 

  112.         self.assertEqual(model.data(model.index(1, 0), QgsBookmarkManagerModel.RoleGroup), 'group 2')
    113. 

  113.         id = model.data(model.index(1, 0), QgsBookmarkManagerModel.RoleId)
    114. 

  114.         self.assertEqual(app_manager.bookmarkById(id).name(), 'b2')
    115. 

  115.         self.assertEqual(model.data(model.index(1, 0), QgsBookmarkManagerModel.RoleExtent), app_manager.bookmarkById(id).extent())
    116. 

  116.         self.assertFalse(model.data(model.index(2, 0)))
    117. 

  117. 

  118.         self.assertFalse(model.setData(model.index(-1, 0), 4, Qt.EditRole))
    119. 

  119.         self.assertTrue(model.setData(model.index(0, 0), 'new name', Qt.EditRole))
    120. 

  120.         self.assertEqual(model.data(model.index(0, 0)), 'new name')
    121. 

  121.         self.assertEqual(app_manager.bookmarks()[0].name(), 'new name')
    122. 

  122.         self.assertTrue(model.setData(model.index(1, 1), 'new group', Qt.EditRole))
    123. 

  123.         self.assertEqual(model.data(model.index(1, 1)), 'new group')
    124. 

  124.         self.assertEqual(app_manager.bookmarks()[1].group(), 'new group')
    125. 

  125.         self.assertTrue(model.setData(model.index(0, 2), 1, Qt.EditRole))
    126. 

  126.         self.assertEqual(model.data(model.index(0, 2)), 1.0)
    127. 

  127.         self.assertEqual(app_manager.bookmarks()[0].extent().xMinimum(), 1.0)
    128. 

  128.         self.assertTrue(model.setData(model.index(0, 3), 2, Qt.EditRole))
    129. 

  129.         self.assertEqual(model.data(model.index(0, 3)), 2.0)
    130. 

  130.         self.assertEqual(app_manager.bookmarks()[0].extent().yMinimum(), 2.0)
    131. 

  131.         self.assertTrue(model.setData(model.index(0, 4), 3, Qt.EditRole))
    132. 

  132.         self.assertEqual(model.data(model.index(0, 4)), 3.0)
    133. 

  133.         self.assertEqual(app_manager.bookmarks()[0].extent().xMaximum(), 3.0)
    134. 

  134.         self.assertTrue(model.setData(model.index(0, 5), 4, Qt.EditRole))
    135. 

  135.         self.assertEqual(model.data(model.index(0, 5)), 4.0)
    136. 

  136.         self.assertEqual(app_manager.bookmarks()[0].extent().yMaximum(), 4.0)
    137. 

  137.         self.assertFalse(model.setData(model.index(2, 0), 4, Qt.EditRole))
    138. 

  138. 

  139.         self.assertTrue(int(model.flags(model.index(0, 0)) & Qt.ItemIsEnabled))
    140. 

  140.         self.assertTrue(int(model.flags(model.index(0, 0)) & Qt.ItemIsEditable))
    141. 

  141.         self.assertTrue(int(model.flags(model.index(0, 7)) & Qt.ItemIsUserCheckable))
    142. 

  142.         self.assertTrue(int(model.flags(model.index(1, 7)) & Qt.ItemIsUserCheckable))
    143. 

  143.         self.assertTrue(int(model.flags(model.index(1, 0)) & Qt.ItemIsEnabled))
    144. 

  144.         self.assertTrue(int(model.flags(model.index(1, 0)) & Qt.ItemIsEditable))
    145. 

  145.         self.assertFalse(int(model.flags(model.index(2, 0)) & Qt.ItemIsEnabled))
    146. 

  146.         self.assertFalse(int(model.flags(model.index(2, 0)) & Qt.ItemIsEditable))
    147. 

  147.         self.assertFalse(int(model.flags(model.index(2, 7)) & Qt.ItemIsUserCheckable))
    148. 

  148. 

  149.         # add bookmark to project manager
    150. 

  150.         b3 = QgsBookmark()
    151. 

  151.         b3.setId('3')
    152. 

  152.         b3.setName('b3')
    153. 

  153.         b3.setGroup('group 3')
    154. 

  154.         b3.setExtent(QgsReferencedRectangle(QgsRectangle(32, 32, 33, 43), QgsCoordinateReferenceSystem('EPSG:28355')))
    155. 

  155.         project_manager.addBookmark(b3)
    156. 

  156. 

  157.         self.assertEqual(model.rowCount(), 3)
    158. 

  158.         self.assertFalse(model.data(model.index(-1, 0)))
    159. 

  159.         self.assertEqual(model.data(model.index(0, 0)), 'new name')
    160. 

  160.         self.assertEqual(model.data(model.index(0, 1)), 'group 1')
    161. 

  161.         self.assertEqual(model.data(model.index(0, 2)), 1.0)
    162. 

  162.         self.assertEqual(model.data(model.index(0, 3)), 2.0)
    163. 

  163.         self.assertEqual(model.data(model.index(0, 4)), 3.0)
    164. 

  164.         self.assertEqual(model.data(model.index(0, 5)), 4.0)
    165. 

  165.         self.assertEqual(model.data(model.index(0, 6)), 'EPSG:4326')
    166. 

  166.         self.assertEqual(model.data(model.index(0, 7)), None)
    167. 

  167.         self.assertEqual(model.data(model.index(0, 7), Qt.CheckStateRole), Qt.Unchecked)
    168. 

  168.         self.assertEqual(model.data(model.index(1, 0)), 'b2')
    169. 

  169.         self.assertEqual(model.data(model.index(1, 1)), 'new group')
    170. 

  170.         self.assertEqual(model.data(model.index(1, 2)), 12.0)
    171. 

  171.         self.assertEqual(model.data(model.index(1, 3)), 22.0)
    172. 

  172.         self.assertEqual(model.data(model.index(1, 4)), 32.0)
    173. 

  173.         self.assertEqual(model.data(model.index(1, 5)), 42.0)
    174. 

  174.         self.assertEqual(model.data(model.index(1, 6)), 'EPSG:4326')
    175. 

  175.         self.assertEqual(model.data(model.index(1, 7)), None)
    176. 

  176.         self.assertEqual(model.data(model.index(1, 7), Qt.CheckStateRole), Qt.Unchecked)
    177. 

  177.         self.assertEqual(model.data(model.index(2, 0)), 'b3')
    178. 

  178.         self.assertEqual(model.data(model.index(2, 1)), 'group 3')
    179. 

  179.         self.assertEqual(model.data(model.index(2, 2)), 32.0)
    180. 

  180.         self.assertEqual(model.data(model.index(2, 3)), 32.0)
    181. 

  181.         self.assertEqual(model.data(model.index(2, 4)), 33.0)
    182. 

  182.         self.assertEqual(model.data(model.index(2, 5)), 43.0)
    183. 

  183.         self.assertEqual(model.data(model.index(2, 6)), 'EPSG:28355')
    184. 

  184.         self.assertEqual(model.data(model.index(2, 7)), None)
    185. 

  185.         self.assertEqual(model.data(model.index(2, 7), Qt.CheckStateRole), Qt.Checked)
    186. 

  186.         self.assertEqual(model.data(model.index(2, 0), QgsBookmarkManagerModel.RoleName), 'b3')
    187. 

  187.         self.assertEqual(model.data(model.index(2, 0), QgsBookmarkManagerModel.RoleGroup), 'group 3')
    188. 

  188.         id = model.data(model.index(2, 0), QgsBookmarkManagerModel.RoleId)
    189. 

  189.         self.assertEqual(project_manager.bookmarkById(id).name(), 'b3')
    190. 

  190.         self.assertEqual(model.data(model.index(2, 0), QgsBookmarkManagerModel.RoleExtent), project_manager.bookmarkById(id).extent())
    191. 

  191.         self.assertFalse(model.data(model.index(3, 0)))
    192. 

  192. 

  193.         self.assertTrue(model.setData(model.index(2, 0), 'new name 2', Qt.EditRole))
    194. 

  194.         self.assertEqual(model.data(model.index(2, 0)), 'new name 2')
    195. 

  195.         self.assertEqual(project_manager.bookmarks()[0].name(), 'new name 2')
    196. 

  196.         self.assertTrue(model.setData(model.index(2, 1), 'new group', Qt.EditRole))
    197. 

  197.         self.assertEqual(model.data(model.index(2, 1)), 'new group')
    198. 

  198.         self.assertEqual(project_manager.bookmarks()[0].group(), 'new group')
    199. 

  199.         self.assertTrue(model.setData(model.index(2, 2), 1, Qt.EditRole))
    200. 

  200.         self.assertEqual(model.data(model.index(2, 2)), 1.0)
    201. 

  201.         self.assertEqual(project_manager.bookmarks()[0].extent().xMinimum(), 1.0)
    202. 

  202.         self.assertTrue(model.setData(model.index(2, 3), 2, Qt.EditRole))
    203. 

  203.         self.assertEqual(model.data(model.index(2, 3)), 2.0)
    204. 

  204.         self.assertEqual(project_manager.bookmarks()[0].extent().yMinimum(), 2.0)
    205. 

  205.         self.assertTrue(model.setData(model.index(2, 4), 3, Qt.EditRole))
    206. 

  206.         self.assertEqual(model.data(model.index(2, 4)), 3.0)
    207. 

  207.         self.assertEqual(project_manager.bookmarks()[0].extent().xMaximum(), 3.0)
    208. 

  208.         self.assertTrue(model.setData(model.index(2, 5), 4, Qt.EditRole))
    209. 

  209.         self.assertEqual(model.data(model.index(2, 5)), 4.0)
    210. 

  210.         self.assertEqual(project_manager.bookmarks()[0].extent().yMaximum(), 4.0)
    211. 

  211.         self.assertFalse(model.setData(model.index(3, 0), 4, Qt.EditRole))
    212. 

  212. 

  213.         self.assertTrue(int(model.flags(model.index(0, 0)) & Qt.ItemIsEnabled))
    214. 

  214.         self.assertTrue(int(model.flags(model.index(0, 0)) & Qt.ItemIsEditable))
    215. 

  215.         self.assertTrue(int(model.flags(model.index(0, 7)) & Qt.ItemIsUserCheckable))
    216. 

  216.         self.assertTrue(int(model.flags(model.index(1, 7)) & Qt.ItemIsUserCheckable))
    217. 

  217.         self.assertTrue(int(model.flags(model.index(1, 0)) & Qt.ItemIsEnabled))
    218. 

  218.         self.assertTrue(int(model.flags(model.index(1, 0)) & Qt.ItemIsEditable))
    219. 

  219.         self.assertTrue(int(model.flags(model.index(2, 0)) & Qt.ItemIsEnabled))
    220. 

  220.         self.assertTrue(int(model.flags(model.index(2, 0)) & Qt.ItemIsEditable))
    221. 

  221.         self.assertTrue(int(model.flags(model.index(2, 7)) & Qt.ItemIsUserCheckable))
    222. 

  222.         self.assertFalse(int(model.flags(model.index(3, 0)) & Qt.ItemIsEnabled))
    223. 

  223.         self.assertFalse(int(model.flags(model.index(3, 0)) & Qt.ItemIsEditable))
    224. 

  224.         self.assertFalse(int(model.flags(model.index(3, 7)) & Qt.ItemIsUserCheckable))
    225. 

  225. 

  226.         # try transferring bookmark from app->project
    227. 

  227.         self.assertTrue(model.setData(model.index(1, 7), Qt.Checked, Qt.CheckStateRole))
    228. 

  228.         self.assertEqual([b.name() for b in project_manager.bookmarks()], ['new name 2', 'b2'])
    229. 

  229.         self.assertEqual([b.name() for b in app_manager.bookmarks()], ['new name'])
    230. 

  230.         self.assertFalse(model.setData(model.index(1, 7), Qt.Checked, Qt.CheckStateRole))
    231. 

  231. 

  232.         # try transferring bookmark from project->app
    233. 

  233.         self.assertTrue(model.setData(model.index(1, 7), Qt.Unchecked, Qt.CheckStateRole))
    234. 

  234.         self.assertEqual([b.name() for b in project_manager.bookmarks()], ['b2'])
    235. 

  235.         self.assertEqual([b.name() for b in app_manager.bookmarks()], ['new name', 'new name 2'])
    236. 

  236.         self.assertFalse(model.setData(model.index(1, 7), Qt.Unchecked, Qt.CheckStateRole))
    237. 

  237. 

  238.         # remove rows
    239. 

  239.         model.removeRows(0, 1)
    240. 

  240.         self.assertEqual([b.name() for b in project_manager.bookmarks()], ['b2'])
    241. 

  241.         self.assertEqual([b.name() for b in app_manager.bookmarks()], ['new name 2'])
    242. 

  242.         model.removeRows(0, 2)
    243. 

  243.         self.assertEqual([b.name() for b in project_manager.bookmarks()], [])
    244. 

  244.         self.assertEqual([b.name() for b in app_manager.bookmarks()], [])
    245. 

  245. 

  246. 

  247. if __name__ == '__main__':
    248. 

  248.     unittest.main()
    249.