From 27780690255dae225a809392ea9ce0d8c37908e5 Mon Sep 17 00:00:00 2001
From: Danil Alexeev <danil@alexeev.xyz>
Date: Mon, 18 Mar 2024 14:51:46 +0300
Subject: GDScript: Fix object iterator opcodes

---
 .../scripts/runtime/features/object_iterators.gd   | 49 ++++++++++++++++++++++
 .../scripts/runtime/features/object_iterators.out  | 30 +++++++++++++
 2 files changed, 79 insertions(+)
 create mode 100644 modules/gdscript/tests/scripts/runtime/features/object_iterators.gd
 create mode 100644 modules/gdscript/tests/scripts/runtime/features/object_iterators.out

(limited to 'modules/gdscript/tests/scripts')

diff --git a/modules/gdscript/tests/scripts/runtime/features/object_iterators.gd b/modules/gdscript/tests/scripts/runtime/features/object_iterators.gd
new file mode 100644
index 0000000000..6fe28c6f78
--- /dev/null
+++ b/modules/gdscript/tests/scripts/runtime/features/object_iterators.gd
@@ -0,0 +1,49 @@
+class MyIterator:
+	var count: int
+
+	func _init(p_count: int) -> void:
+		count = p_count
+
+	func _iter_init(arg: Array) -> bool:
+		prints("_iter_init", arg)
+		arg[0] = 0
+		return arg[0] < count
+
+	func _iter_next(arg: Array) -> bool:
+		prints("_iter_next", arg)
+		arg[0] += 1
+		return arg[0] < count
+
+	func _iter_get(arg: Variant) -> Variant:
+		prints("_iter_get", arg)
+		return arg
+
+func test():
+	var container := PackedDataContainer.new()
+	var _err := container.pack([{
+		id = 123,
+		node_path = ^"/some/path",
+		data = PackedByteArray(),
+	}])
+
+	for ref: PackedDataContainerRef in container:
+		for key: String in ref:
+			print(key)
+
+	print("===")
+
+	for ref: Variant in container:
+		for key: String in ref:
+			print(key)
+
+	print("===")
+
+	var hard_custom := MyIterator.new(3)
+	for x in hard_custom:
+		print(x)
+
+	print("===")
+
+	var weak_custom: Variant = MyIterator.new(3)
+	for x in weak_custom:
+		print(x)
diff --git a/modules/gdscript/tests/scripts/runtime/features/object_iterators.out b/modules/gdscript/tests/scripts/runtime/features/object_iterators.out
new file mode 100644
index 0000000000..942a2c9dd8
--- /dev/null
+++ b/modules/gdscript/tests/scripts/runtime/features/object_iterators.out
@@ -0,0 +1,30 @@
+GDTEST_OK
+id
+node_path
+data
+===
+id
+node_path
+data
+===
+_iter_init [<null>]
+_iter_get 0
+0
+_iter_next [0]
+_iter_get 1
+1
+_iter_next [1]
+_iter_get 2
+2
+_iter_next [2]
+===
+_iter_init [<null>]
+_iter_get 0
+0
+_iter_next [0]
+_iter_get 1
+1
+_iter_next [1]
+_iter_get 2
+2
+_iter_next [2]
-- 
cgit v1.2.3