@@ -310,8 +310,12 @@ def test_convert_plan_preserves_snapshot_id(self):
310310
311311 split = MagicMock ()
312312 original_plan = Plan ([split ], snapshot_id = 42 )
313+ predicate_json = (
314+ '{"kind":"LEAF","transform":{"name":"FIELD_REF",'
315+ '"fieldRef":{"name":"id"}},"function":"EQUAL","literals":[1]}'
316+ )
313317 auth_result = TableQueryAuthResult (
314- filter = ['{"kind":"LEAF","transform":{"name":"FIELD_REF","fieldRef":{"name":"id"}},"function":"EQUAL","literals":[1]}' ],
318+ filter = [predicate_json ],
315319 column_masking = None )
316320 converted = auth_result .convert_plan (original_plan )
317321 assert converted .snapshot_id == 42
@@ -323,8 +327,12 @@ def test_convert_plan_preserves_none_snapshot_id(self):
323327
324328 split = MagicMock ()
325329 original_plan = Plan ([split ], snapshot_id = None )
330+ predicate_json = (
331+ '{"kind":"LEAF","transform":{"name":"FIELD_REF",'
332+ '"fieldRef":{"name":"id"}},"function":"EQUAL","literals":[1]}'
333+ )
326334 auth_result = TableQueryAuthResult (
327- filter = ['{"kind":"LEAF","transform":{"name":"FIELD_REF","fieldRef":{"name":"id"}},"function":"EQUAL","literals":[1]}' ],
335+ filter = [predicate_json ],
328336 column_masking = None )
329337 converted = auth_result .convert_plan (original_plan )
330338 assert converted .snapshot_id is None
0 commit comments