@@ -43,15 +43,17 @@ func TestTraverseFlow_LinearSequence(t *testing.T) {
4343 visited := make (map [model.ID ]bool )
4444 e .traverseFlow (mkID ("start" ), activityMap , flowsByOrigin , nil , visited , nil , nil , & lines , 1 , nil , 0 , nil )
4545
46- // StartEvent produces no output, EndEvent with no return produces no output .
47- // Each activity now has a @position line before it.
48- if len (lines ) != 4 {
49- t .Fatalf ("expected 4 lines, got %d: %v" , len (lines ), lines )
46+ // StartEvent produces no output. Void EndEvent emits an explicit return .
47+ // Each emitted activity has a @position line before it.
48+ if len (lines ) != 6 {
49+ t .Fatalf ("expected 6 lines, got %d: %v" , len (lines ), lines )
5050 }
5151 assertContains (t , lines [0 ], "@position(0, 0)" )
5252 assertContains (t , lines [1 ], "$Obj = create Mod.Entity;" )
5353 assertContains (t , lines [2 ], "@position(0, 0)" )
5454 assertContains (t , lines [3 ], "commit $Obj;" )
55+ assertContains (t , lines [4 ], "@position(0, 0)" )
56+ assertContains (t , lines [5 ], "return;" )
5557}
5658
5759// =============================================================================
@@ -206,11 +208,12 @@ func TestCollectErrorHandlerStatements_Simple(t *testing.T) {
206208 }
207209
208210 stmts := e .collectErrorHandlerStatements (mkID ("err_log" ), activityMap , flowsByOrigin , nil , nil )
209- if len (stmts ) != 1 {
210- t .Fatalf ("expected 1 statement , got %d: %v" , len (stmts ), stmts )
211+ if len (stmts ) != 2 {
212+ t .Fatalf ("expected 2 statements , got %d: %v" , len (stmts ), stmts )
211213 }
212214 assertContains (t , stmts [0 ], "log error" )
213215 assertContains (t , stmts [0 ], "Something failed" )
216+ assertContains (t , stmts [1 ], "return;" )
214217}
215218
216219func TestCollectErrorHandlerStatements_StopsAtMerge (t * testing.T ) {
0 commit comments