def testGettersSetters(self):
""" test model getters/setters """
m = QgsMapLayerProxyModel()
l1 = create_layer('l1')
QgsProject.instance().addMapLayer(l1)
l2 = create_layer('l2')
QgsProject.instance().addMapLayer(l2)
m.setFilters(QgsMapLayerProxyModel.LineLayer
| QgsMapLayerProxyModel.WritableLayer)
self.assertEqual(
m.filters(), QgsMapLayerProxyModel.LineLayer
| QgsMapLayerProxyModel.WritableLayer)
m.setExceptedLayerIds([l2.id()])
self.assertEqual(m.exceptedLayerIds(), [l2.id()])
m.setExceptedLayerList([l2])
self.assertEqual(m.exceptedLayerList(), [l2])
m.setLayerAllowlist([l2])
self.assertEqual(m.layerAllowlist(), [l2])
m.setExcludedProviders(['a', 'b'])
self.assertEqual(m.excludedProviders(), ['a', 'b'])
m.setFilterString('c')
self.assertEqual(m.filterString(), 'c')
def testFilterByLayer(self):
""" test filtering by layer"""
QgsProject.instance().clear()
m = QgsMapLayerProxyModel()
l1 = QgsVectorLayer(
"Point?crs=EPSG:3111&field=fldtxt:string&field=fldint:integer",
'layer 1', "memory")
QgsProject.instance().addMapLayer(l1)
l2 = QgsVectorLayer(
"Polygon?crs=EPSG:3111&field=fldtxt:string&field=fldint:integer",
'lAyEr 2', "memory")
QgsProject.instance().addMapLayer(l2)
l3 = QgsVectorLayer("None?field=fldtxt:string&field=fldint:integer",
'another', "memory")
QgsProject.instance().addMapLayer(l3)
l4 = QgsVectorLayer(
"LineString?crs=EPSG:3111&field=fldtxt:string&field=fldint:integer",
'final layer', "memory")
QgsProject.instance().addMapLayer(l4)
self.assertEqual(m.rowCount(), 4)
self.assertEqual(m.data(m.index(0, 0)), 'another')
self.assertEqual(m.data(m.index(1, 0)), 'final layer')
self.assertEqual(m.data(m.index(2, 0)), 'layer 1')
self.assertEqual(m.data(m.index(3, 0)), 'lAyEr 2')
self.assertTrue(m.acceptsLayer(l1))
self.assertTrue(m.acceptsLayer(l2))
self.assertTrue(m.acceptsLayer(l3))
self.assertTrue(m.acceptsLayer(l4))
m.setExceptedLayerList([l1, l3])
self.assertEqual(m.rowCount(), 2)
self.assertEqual(m.data(m.index(0, 0)), 'final layer')
self.assertEqual(m.data(m.index(1, 0)), 'lAyEr 2')
self.assertFalse(m.acceptsLayer(l1))
self.assertTrue(m.acceptsLayer(l2))
self.assertFalse(m.acceptsLayer(l3))
self.assertTrue(m.acceptsLayer(l4))
m.setExceptedLayerIds([l2.id(), l4.id()])
self.assertEqual(m.rowCount(), 2)
self.assertEqual(m.data(m.index(0, 0)), 'another')
self.assertEqual(m.data(m.index(1, 0)), 'layer 1')
self.assertTrue(m.acceptsLayer(l1))
self.assertFalse(m.acceptsLayer(l2))
self.assertTrue(m.acceptsLayer(l3))
self.assertFalse(m.acceptsLayer(l4))
m.setLayerAllowlist([l1])
self.assertEqual(m.rowCount(), 1)
self.assertEqual(m.data(m.index(0, 0)), 'layer 1')
self.assertTrue(m.acceptsLayer(l1))
self.assertFalse(m.acceptsLayer(l2))
self.assertFalse(m.acceptsLayer(l3))
self.assertFalse(m.acceptsLayer(l4))
m.setExceptedLayerIds([])
self.assertEqual(m.rowCount(), 1)
self.assertEqual(m.data(m.index(0, 0)), 'layer 1')
self.assertTrue(m.acceptsLayer(l1))
self.assertFalse(m.acceptsLayer(l2))
self.assertFalse(m.acceptsLayer(l3))
self.assertFalse(m.acceptsLayer(l4))
m.setLayerAllowlist([l2, l3])
self.assertEqual(m.rowCount(), 2)
self.assertEqual(m.data(m.index(0, 0)), 'another')
self.assertEqual(m.data(m.index(1, 0)), 'lAyEr 2')
self.assertFalse(m.acceptsLayer(l1))
self.assertTrue(m.acceptsLayer(l2))
self.assertTrue(m.acceptsLayer(l3))
self.assertFalse(m.acceptsLayer(l4))
m.setLayerAllowlist([])
self.assertEqual(m.rowCount(), 4)
self.assertTrue(m.acceptsLayer(l1))
self.assertTrue(m.acceptsLayer(l2))
self.assertTrue(m.acceptsLayer(l3))
self.assertTrue(m.acceptsLayer(l4))