Validate loop control behavior across runtime contexts

Author: mgomes

ROADMAP.md

@@ -226,7 +226,7 @@ Goal: improve language ergonomics for complex script logic and recovery behavior
 
 - [x] Ensure new control flow integrates with step quota accounting.
 - [ ] Ensure new constructs integrate with recursion/memory quotas.
-- [ ] Validate behavior inside class methods, blocks, and capability callbacks.
+- [x] Validate behavior inside class methods, blocks, and capability callbacks.
 
 ### Testing and Docs
 

vibes/capability_db_test.go

@@ -165,6 +165,48 @@ end`)
 	}
 }
 
+func TestDBCapabilityEachLoopControlCannotCrossCallbackBoundary(t *testing.T) {
+	stub := &dbCapabilityStub{
+		eachRows: []Value{
+			NewHash(map[string]Value{"id": NewString("p-1")}),
+			NewHash(map[string]Value{"id": NewString("p-2")}),
+		},
+	}
+	engine := MustNewEngine(Config{})
+	script, err := engine.Compile(`def break_from_callback()
+  db.each("Player") do |row|
+    if row[:id] == "p-2"
+      break
+    end
+  end
+end
+
+def next_from_callback()
+  db.each("Player") do |row|
+    if row[:id] == "p-2"
+      next
+    end
+  end
+end`)
+	if err != nil {
+		t.Fatalf("compile failed: %v", err)
+	}
+
+	_, err = script.Call(context.Background(), "break_from_callback", nil, CallOptions{
+		Capabilities: []CapabilityAdapter{MustNewDBCapability("db", stub)},
+	})
+	if err == nil || !strings.Contains(err.Error(), "break used outside of loop") {
+		t.Fatalf("expected callback break outside-loop error, got %v", err)
+	}
+
+	_, err = script.Call(context.Background(), "next_from_callback", nil, CallOptions{
+		Capabilities: []CapabilityAdapter{MustNewDBCapability("db", stub)},
+	})
+	if err == nil || !strings.Contains(err.Error(), "next used outside of loop") {
+		t.Fatalf("expected callback next outside-loop error, got %v", err)
+	}
+}
+
 func TestDBCapabilityRejectsCallableUpdateAttributes(t *testing.T) {
 	stub := &dbCapabilityStub{}
 	engine := MustNewEngine(Config{})

vibes/runtime_test.go

@@ -1005,6 +1005,35 @@ func TestLoopControlNestedAndBlockBoundaryBehavior(t *testing.T) {
 	}
 }
 
+func TestLoopControlInsideClassMethods(t *testing.T) {
+	script := compileScript(t, `
+    class Counter
+      def self.collect(limit)
+        out = []
+        n = 0
+        while n < limit
+          n = n + 1
+          if n % 2 == 0
+            next
+          end
+          if n > 5
+            break
+          end
+          out = out + [n]
+        end
+        out
+      end
+    end
+
+    def run(limit)
+      Counter.collect(limit)
+    end
+    `)
+
+	result := callFunc(t, script, "run", []Value{NewInt(10)})
+	compareArrays(t, result, []Value{NewInt(1), NewInt(3), NewInt(5)})
+}
+
 func TestDurationMethods(t *testing.T) {
 	script := compileScript(t, `
     def duration_helpers()