向前找符号

1 files changed, 31 insertions, 1 deletions

diff --git a/server/test/completion/init.lua b/server/test/completion/init.lua
index 874b0948..cc2c36be 100644
--- a/server/test/completion/init.lua
+++ b/server/test/completion/init.lua
@@ -68,6 +68,20 @@ local function findStartPos(pos, buf)
             break
         end
     end
+    if not res then
+        for i = pos-1, 1, -1 do
+            local c = buf:sub(i, i)
+            if c:find '[%.%:]' then
+                res = i
+            elseif c:find '[%s%c]' then
+            else
+                break
+            end
+        end
+    end
+    if not res then
+        return pos
+    end
     return res
 end
 
@@ -93,7 +107,7 @@ function TEST(script)
         local vm = buildVM(ast)
         assert(vm)
         local word = findWord(pos, new_script)
-        local startPos = findStartPos(pos, new_script) or pos
+        local startPos = findStartPos(pos, new_script)
         local result = core.completion(vm, startPos, word)
         if expect then
             assert(result)
@@ -262,6 +276,22 @@ t.@
 
 TEST [[
 t.a = {}
+t.b = {}
+t.   @
+]]
+{
+    {
+        label = 'a',
+        kind = CompletionItemKind.Field,
+    },
+    {
+        label = 'b',
+        kind = CompletionItemKind.Field,
+    },
+}
+
+TEST [[
+t.a = {}
 function t:b()
 end
 t:@