coroutine in test

Roblox · May 23, 2021 · 72863ab · 72863ab
1 parent 7ea5eeb
commit 72863ab
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/src/RobloxRenderer.spec.lua b/src/RobloxRenderer.spec.lua
@@ -959,6 +959,8 @@ return function()
 				})
 			end
 
+			local childCoroutine
+
 			function ChildComponent:render()
 				if self.state.firstTime then
 					return createElement("Frame")
@@ -968,7 +970,7 @@ return function()
 			end
 
 			function ChildComponent:didMount()
-				spawn(function()
+				childCoroutine = coroutine.create(function()
 					self:setState({
 						firstTime = false
 					})
@@ -1007,6 +1009,8 @@ return function()
 			local hostKey = "Some Key"
 			local instance = reconciler.mountVirtualNode(tree, parent, hostKey)
 
+			coroutine.resume(childCoroutine)
+
 			expect(#parent:GetChildren()).to.equal(1)
 
 			local frame = parent:GetChildren()[1]