Update TestQgsComposerLabel

nyalldawson · nyalldawson · commit 90fd4d4e6479 · 2016-08-10T17:56:01.000+10:00
diff --git a/tests/src/python/test_qgscomposerlabel.py b/tests/src/python/test_qgscomposerlabel.py
@@ -69,18 +69,17 @@ def evaluation_test(self, mComposition, mLabel):
         assert mLabel.displayText() == "__[NAME_1]42__"
 
     def feature_evaluation_test(self, mComposition, mLabel, mVectorLayer):
-        provider = mVectorLayer.dataProvider()
+        atlas = mComposition.atlasComposition()
+        atlas.setCoverageLayer(mVectorLayer)
+        atlas.setEnabled(True)
+        mComposition.setAtlasMode(QgsComposition.ExportAtlas)
 
-        fi = provider.getFeatures(QgsFeatureRequest())
-        feat = QgsFeature()
-
-        fi.nextFeature(feat)
-        mLabel.setExpressionContext(feat, mVectorLayer)
         mLabel.setText("[%\"NAME_1\"||'_ok'%]")
+        atlas.beginRender()
+        atlas.prepareForFeature(0)
         assert mLabel.displayText() == "Basse-Normandie_ok"
 
-        fi.nextFeature(feat)
-        mLabel.setExpressionContext(feat, mVectorLayer)
+        atlas.prepareForFeature(1)
         assert mLabel.displayText() == "Bretagne_ok"
 
     def page_evaluation_test(self, mComposition, mLabel, mVectorLayer):
