From ef0ff9c4c523980d71e49d65e8db57d670daf85c Mon Sep 17 00:00:00 2001
From: Li Jin <dragon-fly@qq.com>
Date: Mon, 5 Feb 2024 10:27:47 +0800
Subject: fix more invalid in-expression use cases.

---
 spec/inputs/in_expression.yue  | 39 +++++++++++++++++++++++++++++++++
 spec/outputs/in_expression.lua | 49 ++++++++++++++++++++++++++++++++++++++++++
 2 files changed, 88 insertions(+)

(limited to 'spec')

diff --git a/spec/inputs/in_expression.yue b/spec/inputs/in_expression.yue
index 6faff4e..7ee0816 100644
--- a/spec/inputs/in_expression.yue
+++ b/spec/inputs/in_expression.yue
@@ -35,5 +35,44 @@ do
 	check -> return x in tb
 	check -> x not in tb
 
+do
+	f1 = ->
+		a = 2
+		a in [
+			1
+			2
+			3
+			4
+		]
+	f2 = ->
+		a not in [
+			1
+			2
+			3
+			4
+		]
+	f3 = ->
+		a = 2
+		a in { b, c, d
+			e, f
+			g
+		}
+	f4 = ->
+		a = 2
+		b = a in [ 1, 2
+			3, 4
+		]
+	f5 = ->
+		a = 2
+		return a in [ 1, 2
+			3, 4
+		]
+	f6 = ->
+		b = a not in {
+			1
+			2
+			x: 3
+		}
+
 nil
 
diff --git a/spec/outputs/in_expression.lua b/spec/outputs/in_expression.lua
index 04c08e7..afcc030 100644
--- a/spec/outputs/in_expression.lua
+++ b/spec/outputs/in_expression.lua
@@ -162,4 +162,53 @@ do
 		return true
 	end)
 end
+do
+	local f1
+	f1 = function()
+		local a = 2
+		return (1 == a or 2 == a or 3 == a or 4 == a)
+	end
+	local f2
+	f2 = function()
+		local _val_0 = a
+		return not (1 == _val_0 or 2 == _val_0 or 3 == _val_0 or 4 == _val_0)
+	end
+	local f3
+	f3 = function()
+		local a = 2
+		return (b == a or c == a or d == a or e == a or f == a or g == a)
+	end
+	local f4
+	f4 = function()
+		local a = 2
+		local b
+		b = (1 == a or 2 == a or 3 == a or 4 == a)
+	end
+	local f5
+	f5 = function()
+		local a = 2
+		return (1 == a or 2 == a or 3 == a or 4 == a)
+	end
+	local f6
+	f6 = function()
+		local b
+		do
+			local _check_0 = {
+				1,
+				2,
+				x = 3
+			}
+			local _val_0 = a
+			local _find_0 = false
+			for _index_0 = 1, #_check_0 do
+				local _item_0 = _check_0[_index_0]
+				if _item_0 == _val_0 then
+					_find_0 = true
+					break
+				end
+			end
+			b = not _find_0
+		end
+	end
+end
 return nil
-- 
cgit v1.2.3-55-g6feb