Add test for condition

Author: lazebnyi

unit_tests/sources/declarative/transformations/test_add_fields.py

@@ -12,12 +12,13 @@
 
 
 @pytest.mark.parametrize(
-    ["input_record", "field", "field_type", "kwargs", "expected"],
+    ["input_record", "field", "field_type", "condition", "kwargs", "expected"],
     [
         pytest.param(
             {"k": "v"},
             [(["path"], "static_value")],
             None,
+            "",
             {},
             {"k": "v", "path": "static_value"},
             id="add new static value",
@@ -26,6 +27,7 @@
             {"k": "v"},
             [(["path"], "{{ 1 }}")],
             None,
+            "",
             {},
             {"k": "v", "path": 1},
             id="add an expression evaluated as a number",
@@ -34,6 +36,7 @@
             {"k": "v"},
             [(["path"], "{{ 1 }}")],
             str,
+            "",
             {},
             {"k": "v", "path": "1"},
             id="add an expression evaluated as a string using the value_type field",
@@ -42,6 +45,7 @@
             {"k": "v"},
             [(["path"], "static_value"), (["path2"], "static_value2")],
             None,
+            "",
             {},
             {"k": "v", "path": "static_value", "path2": "static_value2"},
             id="add new multiple static values",
@@ -50,6 +54,7 @@
             {"k": "v"},
             [(["nested", "path"], "static_value")],
             None,
+            "",
             {},
             {"k": "v", "nested": {"path": "static_value"}},
             id="set static value at nested path",
@@ -58,6 +63,7 @@
             {"k": "v"},
             [(["k"], "new_value")],
             None,
+            "",
             {},
             {"k": "new_value"},
             id="update value which already exists",
@@ -66,6 +72,7 @@
             {"k": [0, 1]},
             [(["k", 3], "v")],
             None,
+            "",
             {},
             {"k": [0, 1, None, "v"]},
             id="Set element inside array",
@@ -74,6 +81,7 @@
             {"k": "v"},
             [(["k2"], '{{ config["shop"] }}')],
             None,
+            "",
             {"config": {"shop": "in-n-out"}},
             {"k": "v", "k2": "in-n-out"},
             id="set a value from the config using bracket notation",
@@ -82,6 +90,7 @@
             {"k": "v"},
             [(["k2"], "{{ config.shop }}")],
             None,
+            "",
             {"config": {"shop": "in-n-out"}},
             {"k": "v", "k2": "in-n-out"},
             id="set a value from the config using dot notation",
@@ -90,6 +99,7 @@
             {"k": "v"},
             [(["k2"], '{{ stream_slice["start_date"] }}')],
             None,
+            "",
             {"stream_slice": {"start_date": "oct1"}},
             {"k": "v", "k2": "oct1"},
             id="set a value from the stream slice using bracket notation",
@@ -98,6 +108,7 @@
             {"k": "v"},
             [(["k2"], "{{ stream_slice.start_date }}")],
             None,
+            "",
             {"stream_slice": {"start_date": "oct1"}},
             {"k": "v", "k2": "oct1"},
             id="set a value from the stream slice using dot notation",
@@ -106,6 +117,7 @@
             {"k": "v"},
             [(["k2"], "{{ record.k }}")],
             None,
+            "",
             {},
             {"k": "v", "k2": "v"},
             id="set a value from a field in the record using dot notation",
@@ -114,6 +126,7 @@
             {"k": "v"},
             [(["k2"], '{{ record["k"] }}')],
             None,
+            "",
             {},
             {"k": "v", "k2": "v"},
             id="set a value from a field in the record using bracket notation",
@@ -122,6 +135,7 @@
             {"k": {"nested": "v"}},
             [(["k2"], "{{ record.k.nested }}")],
             None,
+            "",
             {},
             {"k": {"nested": "v"}, "k2": "v"},
             id="set a value from a nested field in the record using bracket notation",
@@ -130,6 +144,7 @@
             {"k": {"nested": "v"}},
             [(["k2"], '{{ record["k"]["nested"] }}')],
             None,
+            "",
             {},
             {"k": {"nested": "v"}, "k2": "v"},
             id="set a value from a nested field in the record using bracket notation",
@@ -138,22 +153,42 @@
             {"k": "v"},
             [(["k2"], "{{ 2 + 2 }}")],
             None,
+            "",
             {},
             {"k": "v", "k2": 4},
             id="set a value from a jinja expression",
         ),
+        pytest.param(
+            {"k": "v"},
+            [(["path"], "static_value")],
+            None,
+            "{{ False }}",
+            {},
+            {"k": "v"},
+            id="do not add any field if condition is boolean False",
+        ),
+        pytest.param(
+            {"k": "v"},
+            [(["path"], "static_value")],
+            None,
+            "{{ True }}",
+            {},
+            {"k": "v", "path": "static_value"},
+            id="add all fields if condition is boolean True",
+        ),
     ],
 )
 def test_add_fields(
     input_record: Mapping[str, Any],
     field: List[Tuple[FieldPointer, str]],
     field_type: Optional[str],
+    condition: Optional[str],
     kwargs: Mapping[str, Any],
     expected: Mapping[str, Any],
 ):
     inputs = [
         AddedFieldDefinition(path=v[0], value=v[1], value_type=field_type, parameters={})
         for v in field
     ]
-    AddFields(fields=inputs, parameters={"alas": "i live"}).transform(input_record, **kwargs)
+    AddFields(fields=inputs, condition=condition, parameters={"alas": "i live"}).transform(input_record, **kwargs)
     assert input_record == expected