Skip to content

Commit 7f97c31

Browse files
committed
Add test for custom prefix on joins
1 parent 772033d commit 7f97c31

1 file changed

Lines changed: 16 additions & 6 deletions

File tree

tests/src/python/test_qgsvectorlayer.py

Lines changed: 16 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -843,7 +843,8 @@ def test_getFeatures(self):
843843
def test_join(self):
844844

845845
joinLayer = createJoinLayer()
846-
QgsMapLayerRegistry.instance().addMapLayers([joinLayer])
846+
joinLayer2 = createJoinLayer()
847+
QgsMapLayerRegistry.instance().addMapLayers([joinLayer,joinLayer2])
847848

848849
layer = createLayerWithOnePoint()
849850

@@ -855,10 +856,19 @@ def test_join(self):
855856

856857
layer.addJoin(join)
857858

859+
join2 = QgsVectorJoinInfo()
860+
join2.targetFieldName = "fldint"
861+
join2.joinLayerId = joinLayer2.id()
862+
join2.joinFieldName = "y"
863+
join2.memoryCache = True
864+
join2.prefix = "custom-prefix_"
865+
866+
layer.addJoin(join2)
867+
858868
flds = layer.pendingFields()
859-
assert len(flds) == 4
860-
assert flds[2].name() == "joinlayer_x"
861-
assert flds[3].name() == "joinlayer_z"
869+
assert len(flds) == 6
870+
assert flds[4].name() == "custom-prefix_x"
871+
assert flds[5].name() == "custom-prefix_z"
862872
assert flds.fieldOrigin(0) == QgsFields.OriginProvider
863873
assert flds.fieldOrigin(2) == QgsFields.OriginJoin
864874
assert flds.fieldOrigin(3) == QgsFields.OriginJoin
@@ -870,15 +880,15 @@ def test_join(self):
870880
fi = layer.getFeatures()
871881
assert fi.nextFeature(f) == True
872882
attrs = f.attributes()
873-
assert len(attrs) == 4
883+
assert len(attrs) == 6
874884
assert attrs[0] == "test"
875885
assert attrs[1] == 123
876886
assert attrs[2] == "foo"
877887
assert attrs[3] == 321
878888
assert fi.nextFeature(f) == False
879889

880890
f2 = layer.getFeatures(QgsFeatureRequest( f.id() )).next()
881-
assert len(f2.attributes()) == 4
891+
assert len(f2.attributes()) == 6
882892
assert f2[2] == "foo"
883893
assert f2[3] == 321
884894

0 commit comments

Comments
 (0)