@@ -206,15 +206,30 @@ def test_get_all_latest_flag_variations():
206206 assert client .initialize
207207 all_states = client .get_all_latest_flag_variations (USER_1 )
208208 ed = all_states .get ("ff-test-bool" , False )
209+ value = all_states .get_variation ("ff-test-bool" , False )
209210 assert ed is not None and ed .variation is True
211+ assert value is True
212+ assert "ff-test-bool" in all_states .keys ()
210213 ed = all_states .get ("ff-test-number" , - 1 )
214+ value = all_states .get_variation ("ff-test-number" , - 1 )
211215 assert ed is not None and ed .variation == 1
216+ assert value == 1
217+ assert "ff-test-number" in all_states .keys ()
212218 ed = all_states .get ("ff-test-string" , 'error' )
219+ value = all_states .get_variation ("ff-test-string" , 'error' )
213220 assert ed is not None and ed .variation == "others"
221+ assert value == "others"
222+ assert "ff-test-string" in all_states .keys ()
214223 ed = all_states .get ("ff-test-seg" , 'error' )
224+ value = all_states .get_variation ("ff-test-seg" , 'error' )
215225 assert ed is not None and ed .variation == "teamA"
226+ assert value == "teamA"
227+ assert "ff-test-seg" in all_states .keys ()
216228 ed = all_states .get ("ff-test-json" , {})
229+ value = all_states .get_variation ("ff-test-json" , {})
217230 assert ed is not None and ed .variation ["code" ] == 200
231+ assert value ["code" ] == 200
232+ assert "ff-test-json" in all_states .keys ()
218233
219234
220235def test_variation_argument_error ():
0 commit comments