fix: allow whitespaces at end of expression (#121)

fix: allow whitespaces at end of line
microsoft · Aug 5, 2024 · e407b10 · e407b10
1 parent b916bb7
commit e407b10
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/...y_testing_framework/_expression_runtime/data_factory_expression/expression_transformer.py b/...y_testing_framework/_expression_runtime/data_factory_expression/expression_transformer.py
@@ -40,7 +40,7 @@ def __init__(self) -> None:
 
         expression_grammar = f"""
             expression_start: "@" expression_evaluation
-            expression_evaluation: (expression_logical_bool | expression_branch | expression_call) ((("." EXPRESSION_PARAMETER_NAME) | EXPRESSION_ARRAY_INDEX)+)?
+            expression_evaluation: (expression_logical_bool | expression_branch | expression_call) ((("." EXPRESSION_PARAMETER_NAME) | EXPRESSION_ARRAY_INDEX)+)? EXPRESSION_WS*
             ?expression_call: expression_function_call
                               // used to translate to expression_pipeline_reference
                               | expression_datafactory_parameters_reference

diff --git a/tests/unit/functions/test_data_factory_testing_framework_expression_evaluator.py b/tests/unit/functions/test_data_factory_testing_framework_expression_evaluator.py
@@ -120,6 +120,7 @@
             id="linked_service_reference",
         ),
         p("@item()", PipelineRunState(iteration_item="value"), "@item()", "value", id="item_reference"),
+        p("@item() ", PipelineRunState(iteration_item="value"), "@item() ", "value", id="item_reference_with_ws"),
         p("@concat('a', 'b' )", PipelineRunState(), "@concat('a', 'b' )", "ab", id="function_call"),
         p(
             "@concat('https://example.com/jobs/', '123''', concat('&', 'abc,'))",