Fix NoYield to give a useful stack trace in all cases (#27)

Roblox · May 2, 2018 · d1161cd · d1161cd
1 parent 576767d
commit d1161cd
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/lib/NoYield.lua b/lib/NoYield.lua
@@ -9,8 +9,8 @@
 
 local function resultHandler(co, ok, ...)
 	if not ok then
-		local err = (...)
-		error(err, 2)
+		local message = (...)
+		error(debug.traceback(co, message), 2)
 	end
 
 	if coroutine.status(co) ~= "dead" then

diff --git a/lib/NoYield.spec.lua b/lib/NoYield.spec.lua
@@ -51,5 +51,6 @@ return function()
 
 		expect(ok).to.equal(false)
 		expect(err:find("foo")).to.be.ok()
+		expect(err:find("NoYield.spec")).to.be.ok()
 	end)
 end