Fix python bindings for QgsRelation.fieldPairs()

qgis · Aug 18, 2014 · 1b205be · 1b205be
1 parent 978642a
commit 1b205be
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 1 deletion.
diff --git a/python/core/qgsrelation.sip b/python/core/qgsrelation.sip
@@ -165,7 +165,8 @@ class QgsRelation
      */
     QMap< QString, QString > fieldPairs() const;
 %MethodCode
-    QList< QgsRelation::FieldPair > pairs = sipCpp->fieldPairs();
+    const QList< QgsRelation::FieldPair >& pairs = sipCpp->fieldPairs();
+    sipRes = new QMap< QString, QString >();
     Q_FOREACH( const QgsRelation::FieldPair& pair, pairs )
     {
         sipRes->insert( pair.first, pair.second );

diff --git a/tests/src/python/test_qgsrelation.py b/tests/src/python/test_qgsrelation.py
@@ -119,5 +119,23 @@ def test_getRelatedFeatures(self):
 
         QgsMapLayerRegistry.instance().removeAllMapLayers()
 
+    def test_fieldPairs(self):
+        referencedLayer = createReferencedLayer()
+        referencingLayer = createReferencingLayer()
+        QgsMapLayerRegistry.instance().addMapLayers([referencedLayer,referencingLayer])
+
+        rel = QgsRelation()
+
+        rel.setRelationId( 'rel1' )
+        rel.setRelationName( 'Relation Number One' )
+        rel.setReferencingLayer( referencingLayer.id() )
+        rel.setReferencedLayer( referencedLayer.id() )
+        rel.addFieldPair( 'foreignkey', 'y' )
+
+        assert( rel.fieldPairs() == { 'foreignkey': 'y'} )
+
+        QgsMapLayerRegistry.instance().removeAllMapLayers()
+
+
 if __name__ == '__main__':
     unittest.main()