修正一些BUG

2 files changed, 10 insertions, 4 deletions

diff --git a/server/test/definition/init.lua b/server/test/definition/init.lua
index 263a73cb..216fc6ed 100644
--- a/server/test/definition/init.lua
+++ b/server/test/definition/init.lua
@@ -47,10 +47,10 @@ function TEST(script)
     local new_script = script:gsub('<[!?]', '  '):gsub('[!?]>', '  ')
     local ast = parser:ast(new_script)
     assert(ast)
-    local results = matcher.vm(ast)
-    assert(results)
+    local vm = matcher.vm(ast)
+    assert(vm)
 
-    local result = matcher.definition(results, pos)
+    local result = matcher.definition(vm, pos)
     if result then
         assert(founded(target, result))
     else
diff --git a/server/test/vm/init.lua b/server/test/vm/init.lua
index d8cfb3dc..799fa367 100644
--- a/server/test/vm/init.lua
+++ b/server/test/vm/init.lua
@@ -6,7 +6,9 @@ rawset(_G, 'TEST', true)
 function TEST(script)
     local ast = parser:ast(script)
     assert(ast)
-    local results = matcher.vm(ast)
+    local vm = matcher.vm(ast)
+    assert(vm)
+    local results = vm.results
     assert(results)
 end
 
@@ -74,6 +76,10 @@ local a, b, c = 1, 2, ...
 ]]
 
 TEST [[
+a[#a+1] = 1
+]]
+
+TEST [[
 xx(a, b, 2, 3, ...)
 ]]