def testProperties(self):
model = QgsFieldMappingModel(self.source_fields, self.destination_fields)
model.setDestinationFields(self.destination_fields, {'destination_field1': '5',
'destination_field2': 'source_field2',
'destination_field3': 'source_field2 * @myvar'})
mapping = model.mapping()
self.assertEqual(mapping[0].field.name(), 'destination_field1')
self.assertEqual(mapping[1].field.name(), 'destination_field2')
self.assertEqual(mapping[2].field.name(), 'destination_field3')
self.assertEqual(mapping[0].expression, '5')
self.assertEqual(mapping[1].expression, 'source_field2')
self.assertEqual(mapping[2].expression, 'source_field2 * @myvar')
self.assertEqual(model.fieldPropertyMap(), {'destination_field1': QgsProperty.fromExpression('5'),
'destination_field2': QgsProperty.fromField('source_field2'),
'destination_field3': QgsProperty.fromExpression(
'source_field2 * @myvar'),
})
model = QgsFieldMappingModel(self.source_fields, self.destination_fields)
self.assertEqual(model.fieldPropertyMap(), {'destination_field1': QgsProperty.fromField('source_field2'),
'destination_field2': QgsProperty.fromField('source_field1'),
'destination_field3': QgsProperty.fromExpression(''),
})
model.setFieldPropertyMap({
'destination_field1': QgsProperty.fromField('source_field1'),
'destination_field2': QgsProperty.fromExpression('55*6'),
'destination_field3': QgsProperty.fromValue(6),
})
self.assertEqual(model.fieldPropertyMap(), {
'destination_field1': QgsProperty.fromField('source_field1'),
'destination_field2': QgsProperty.fromExpression('55*6'),
'destination_field3': QgsProperty.fromExpression('6'),
})
def testModel(self):
"""Test the mapping model"""
model = QgsFieldMappingModel(self.source_fields, self.destination_fields)
self.assertEqual(model.rowCount(QModelIndex()), 3)
self.assertIsNone(model.data(model.index(9999, 0), Qt.DisplayRole))
# We now have this default mapping:
# source exp | destination fld
# -------------------------------------------
# source_field2 | destination_field1
# source_field1 | destination_field2
# NOT SET (NULL) | destination_field3
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole), '"source_field2"')
self.assertEqual(model.data(model.index(0, 1), Qt.DisplayRole), 'destination_field1')
self.assertEqual(model.data(model.index(0, 3), Qt.DisplayRole), 10)
self.assertEqual(model.data(model.index(0, 4), Qt.DisplayRole), 8)
self.assertEqual(model.data(model.index(1, 0), Qt.DisplayRole), '"source_field1"')
self.assertEqual(model.data(model.index(1, 1), Qt.DisplayRole), 'destination_field2')
self.assertEqual(model.data(model.index(2, 0), Qt.DisplayRole), QVariant())
self.assertEqual(model.data(model.index(2, 1), Qt.DisplayRole), 'destination_field3')
# Test expression scope
ctx = model.contextGenerator().createExpressionContext()
self.assertTrue('source_field1' in ctx.fields().names())
# Test add fields
model.appendField(QgsField('destination_field4', QVariant.String))
self.assertEqual(model.rowCount(QModelIndex()), 4)
self.assertEqual(model.data(model.index(3, 1), Qt.DisplayRole), 'destination_field4')
# Test remove field
model.removeField(model.index(3, 0))
self.assertEqual(model.rowCount(QModelIndex()), 3)
self.assertEqual(model.data(model.index(2, 1), Qt.DisplayRole), 'destination_field3')
# Test edit fields
mapping = model.mapping()
self.assertEqual(mapping[0].field.name(), 'destination_field1')
self.assertEqual(mapping[1].field.name(), 'destination_field2')
self.assertEqual(mapping[2].field.name(), 'destination_field3')
self.assertEqual(mapping[0].originalName, 'destination_field1')
self.assertEqual(mapping[1].originalName, 'destination_field2')
self.assertEqual(mapping[2].originalName, 'destination_field3')
# Test move up or down
self.assertFalse(model.moveUp(model.index(0, 0)))
self.assertFalse(model.moveUp(model.index(100, 0)))
self.assertFalse(model.moveDown(model.index(2, 0)))
self.assertFalse(model.moveDown(model.index(100, 0)))
self.assertTrue(model.moveDown(model.index(0, 0)))
mapping = model.mapping()
self.assertEqual(mapping[1].field.name(), 'destination_field1')
self.assertEqual(mapping[0].field.name(), 'destination_field2')
self.assertEqual(mapping[2].field.name(), 'destination_field3')
self.assertEqual(mapping[1].originalName, 'destination_field1')
self.assertEqual(mapping[0].originalName, 'destination_field2')
self.assertEqual(mapping[2].originalName, 'destination_field3')
self.assertTrue(model.moveUp(model.index(1, 0)))
mapping = model.mapping()
self.assertEqual(mapping[0].field.name(), 'destination_field1')
self.assertEqual(mapping[1].field.name(), 'destination_field2')
self.assertEqual(mapping[2].field.name(), 'destination_field3')
self.assertEqual(mapping[0].originalName, 'destination_field1')
self.assertEqual(mapping[1].originalName, 'destination_field2')
self.assertEqual(mapping[2].originalName, 'destination_field3')
self.assertTrue(model.moveUp(model.index(2, 0)))
mapping = model.mapping()
self.assertEqual(mapping[0].field.name(), 'destination_field1')
self.assertEqual(mapping[2].field.name(), 'destination_field2')
self.assertEqual(mapping[1].field.name(), 'destination_field3')
self.assertEqual(mapping[0].originalName, 'destination_field1')
self.assertEqual(mapping[2].originalName, 'destination_field2')
self.assertEqual(mapping[1].originalName, 'destination_field3')
def testSetSourceFields(self):
"""Test that changing source fields also empty expressions are updated"""
model = QgsFieldMappingModel(self.source_fields, self.destination_fields)
self.assertEqual(model.data(model.index(2, 0), Qt.DisplayRole), QVariant())
self.assertEqual(model.data(model.index(2, 1), Qt.DisplayRole), 'destination_field3')
f = QgsField('source_field3', QVariant.String)
fields = self.source_fields
fields.append(f)
model.setSourceFields(fields)
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole), '"source_field2"')
self.assertEqual(model.data(model.index(0, 1), Qt.DisplayRole), 'destination_field1')
self.assertEqual(model.data(model.index(1, 0), Qt.DisplayRole), '"source_field1"')
self.assertEqual(model.data(model.index(1, 1), Qt.DisplayRole), 'destination_field2')
self.assertEqual(model.data(model.index(2, 0), Qt.DisplayRole), '"source_field3"')
self.assertEqual(model.data(model.index(2, 1), Qt.DisplayRole), 'destination_field3')
