From 8049d92b3e9aef35e4f124e5ae361bdd1f5c5435 Mon Sep 17 00:00:00 2001
From: OpportunityLiu <opportunity@live.in>
Date: Sun, 9 Jun 2019 11:46:22 +0800
Subject: fix

---
 tests/test_utils/test_assert.lua | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

(limited to 'tests/test_utils/test_assert.lua')

diff --git a/tests/test_utils/test_assert.lua b/tests/test_utils/test_assert.lua
index f9af3c89a..42a276d87 100644
--- a/tests/test_utils/test_assert.lua
+++ b/tests/test_utils/test_assert.lua
@@ -45,19 +45,23 @@ function test_assert:are_not_equal(actual, expacted)
     end
 end
 
+test_assert._will_raise_stack = {}
 function test_assert:will_raise(func, message_pattern)
+    table.insert(self._will_raise_stack, 1, debug.getinfo(2).func)
     try{
         func,
         finally{
             function (ok, error)
+                local funcs = { func, unpack(self._will_raise_stack) }
                 if ok then
-                    self:print_error("expected raise but finished successfully", func)
+                    self:print_error("expected raise but finished successfully", funcs)
                 elseif message_pattern and not string.find(error, message_pattern) then
-                    self:print_error(format("expected raise with message ${green}%s${reset} but got ${red}%s${reset}", message_pattern, error), func)
+                    self:print_error(format("expected raise with message ${green}%s${reset} but got ${red}%s${reset}", message_pattern, error), funcs)
                 end
             end
         }
     }
+    table.remove(self._will_raise_stack, 1)
 end
 
 function main(context)
-- 
cgit v1.3.1