From e02321107277a63e7dcb12ab163c9942ac101b87 Mon Sep 17 00:00:00 2001
From: Li Jin <dragon-fly@qq.com>
Date: Mon, 26 Jan 2026 17:45:26 +0800
Subject: Updated tests.

---
 spec/outputs/test/return_spec.lua | 147 ++++++++++++++++++++++++++++++++++++++
 1 file changed, 147 insertions(+)
 create mode 100644 spec/outputs/test/return_spec.lua

(limited to 'spec/outputs/test/return_spec.lua')

diff --git a/spec/outputs/test/return_spec.lua b/spec/outputs/test/return_spec.lua
new file mode 100644
index 0000000..9479773
--- /dev/null
+++ b/spec/outputs/test/return_spec.lua
@@ -0,0 +1,147 @@
+return describe("return", function()
+	it("should return from comprehension", function()
+		local fn
+		fn = function()
+			local _accum_0 = { }
+			local _len_0 = 1
+			for x = 1, 5 do
+				_accum_0[_len_0] = x * 2
+				_len_0 = _len_0 + 1
+			end
+			return _accum_0
+		end
+		local result = fn()
+		return assert.same(result, {
+			2,
+			4,
+			6,
+			8,
+			10
+		})
+	end)
+	it("should return from table comprehension", function()
+		local fn
+		fn = function()
+			local _tbl_0 = { }
+			for k, v in pairs({
+				a = 1,
+				b = 2
+			}) do
+				_tbl_0[k] = v
+			end
+			return _tbl_0
+		end
+		local result = fn()
+		return assert.same(type(result), "table")
+	end)
+	it("should return from nested if", function()
+		local fn
+		fn = function(a, b)
+			if a then
+				if b then
+					return "both"
+				else
+					return "only a"
+				end
+			else
+				return "neither"
+			end
+		end
+		assert.same(fn(true, true), "both")
+		assert.same(fn(true, false), "only a")
+		return assert.same(fn(false, false), "neither")
+	end)
+	it("should return from switch", function()
+		local fn
+		fn = function(value)
+			if 1 == value then
+				return "one"
+			elseif 2 == value then
+				return "two"
+			else
+				return "other"
+			end
+		end
+		assert.same(fn(1), "one")
+		assert.same(fn(2), "two")
+		return assert.same(fn(3), "other")
+	end)
+	it("should return table literal", function()
+		local fn
+		fn = function()
+			return {
+				value = 42,
+				name = "test"
+			}
+		end
+		local result = fn()
+		assert.same(result.value, 42)
+		return assert.same(result.name, "test")
+	end)
+	it("should return array literal", function()
+		local fn
+		fn = function()
+			return {
+				1,
+				2,
+				3
+			}
+		end
+		local result = fn()
+		return assert.same(result, {
+			1,
+			2,
+			3
+		})
+	end)
+	it("should return from with statement", function()
+		local fn
+		fn = function(obj)
+			local result = obj.value
+			return result
+		end
+		return assert.same(fn({
+			value = 100
+		}), 100)
+	end)
+	it("should return nil implicitly", function()
+		local fn
+		fn = function()
+			local _ = "no return"
+		end
+		return assert.same(fn(), nil)
+	end)
+	it("should return multiple values", function()
+		local fn
+		fn = function()
+			return 1, 2, 3
+		end
+		local a, b, c = fn()
+		assert.same(a, 1)
+		assert.same(b, 2)
+		return assert.same(c, 3)
+	end)
+	it("should return from function call", function()
+		local fn
+		fn = function()
+			local inner
+			inner = function()
+				return 42
+			end
+			return inner()
+		end
+		return assert.same(fn(), 42)
+	end)
+	return it("should handle return in expression context", function()
+		local fn
+		fn = function(cond)
+			if cond then
+				return "yes"
+			else
+				return "no"
+			end
+		end
+		assert.same(fn(true), "yes")
+		return assert.same(fn(false), "no")
+	end)
+end)
-- 
cgit v1.2.3-55-g6feb