Python: Rebase durable task feature branch with main (#2806)

2026-06-16 21:04:09 +08:00 · 2025-12-17 14:02:36 -08:00
parent a48a8dd524
commit 87a38bc7da
227 changed files with 11968 additions and 2637 deletions
@@ -394,4 +394,61 @@ public class WorkflowVisualizerTests
        // Check fan-in (should have intermediate node)
        mermaidContent.Should().Contain("((fan-in))");
    }
+
+    [Fact]
+    public void Test_WorkflowViz_Mermaid_Edge_Label_With_Pipe()
+    {
+        // Test that pipe characters in labels are properly escaped
+        var start = new MockExecutor("start");
+        var end = new MockExecutor("end");
+
+        var workflow = new WorkflowBuilder("start")
+            .AddEdge(start, end, label: "High | Low Priority")
+            .Build();
+
+        var mermaidContent = workflow.ToMermaidString();
+
+        // Should escape pipe character
+        mermaidContent.Should().Contain("start -->|High &#124; Low Priority| end");
+        // Should not contain unescaped pipe that would break syntax
+        mermaidContent.Should().NotContain("-->|High | Low");
+    }
+
+    [Fact]
+    public void Test_WorkflowViz_Mermaid_Edge_Label_With_Special_Chars()
+    {
+        // Test that special characters are properly escaped
+        var start = new MockExecutor("start");
+        var end = new MockExecutor("end");
+
+        var workflow = new WorkflowBuilder("start")
+            .AddEdge(start, end, label: "Score >= 90 & < 100")
+            .Build();
+
+        var mermaidContent = workflow.ToMermaidString();
+
+        // Should escape special characters
+        mermaidContent.Should().Contain("&amp;");
+        mermaidContent.Should().Contain("&gt;");
+        mermaidContent.Should().Contain("&lt;");
+    }
+
+    [Fact]
+    public void Test_WorkflowViz_Mermaid_Edge_Label_With_Newline()
+    {
+        // Test that newlines are converted to <br/>
+        var start = new MockExecutor("start");
+        var end = new MockExecutor("end");
+
+        var workflow = new WorkflowBuilder("start")
+            .AddEdge(start, end, label: "Line 1\nLine 2")
+            .Build();
+
+        var mermaidContent = workflow.ToMermaidString();
+
+        // Should convert newline to <br/>
+        mermaidContent.Should().Contain("Line 1<br/>Line 2");
+        // Should not contain literal newline in the label (but the overall output has newlines between statements)
+        mermaidContent.Should().NotContain("Line 1\nLine 2");
+    }
 }