add cli test for scret redacting

Author: Min Shi

src/promptflow/tests/sdk_cli_test/e2etests/test_cli.py

@@ -1407,3 +1407,45 @@ def test_flow_test_with_image_input_and_output(self):
         assert output_path.exists()
         image_path = Path(FLOWS_DIR) / "python_tool_with_image_input_and_output" / ".promptflow" / "intermediate"
         assert image_path.exists()
+
+    def test_data_scrubbing(self):
+        # Prepare connection
+        run_pf_command(
+            "connection",
+            "create",
+            "--file",
+            f"{CONNECTIONS_DIR}/custom_connection.yaml",
+            "--name",
+            "custom_connection")
+        # Test flow run
+        run_pf_command(
+            "flow",
+            "test",
+            "--flow",
+            f"{FLOWS_DIR}/print_secret_flow",
+        )
+        output_path = Path(FLOWS_DIR) / "print_secret_flow" / ".promptflow" / "flow.output.json"
+        assert output_path.exists()
+        log_path = Path(FLOWS_DIR) / "print_secret_flow" / ".promptflow" / "flow.log"
+        with open(log_path, "r") as f:
+            log_content = f.read()
+            assert "**data_scrubbed**" in log_content
+
+        # Test node run
+        run_pf_command(
+            "flow",
+            "test",
+            "--flow",
+            f"{FLOWS_DIR}/print_secret_flow",
+            "--node",
+            "print_secret",
+            "--inputs",
+            "conn=custom_connection",
+            "inputs.topic=atom"
+        )
+        output_path = Path(FLOWS_DIR) / "print_secret_flow" / ".promptflow" / "flow-print_secret.node.detail.json"
+        assert output_path.exists()
+        log_path = Path(FLOWS_DIR) / "print_secret_flow" / ".promptflow" / "print_secret.node.log"
+        with open(log_path, "r") as f:
+            log_content = f.read()
+        assert "**data_scrubbed**" in log_content

src/promptflow/tests/test_configs/flows/print_secret_flow/flow.dag.yaml

@@ -0,0 +1,17 @@
+inputs:
+  key:
+    type: string
+    default: text
+outputs:
+  output:
+    type: string
+    reference: ${print_secret.output}
+nodes:
+- name: print_secret
+  type: python
+  source:
+    type: code
+    path: print_secret.py
+  inputs:
+    connection: custom_connection
+    text: ${inputs.key}

src/promptflow/tests/test_configs/flows/print_secret_flow/print_secret.py

@@ -0,0 +1,11 @@
+import os
+
+from promptflow import tool
+from promptflow.connections import CustomConnection
+
+
+@tool
+def print_secret(text: str, connection: CustomConnection):
+    print(connection["key1"])
+    print(connection["key2"])
+    return text