rename

2 files changed, 86 insertions, 0 deletions

diff --git a/server/test/main.lua b/server/test/main.lua
index 7a498929..e6fdcbe2 100644
--- a/server/test/main.lua
+++ b/server/test/main.lua
@@ -27,6 +27,7 @@ local function main()
     test 'core'
     test 'full'
     test 'definition'
+    test 'rename'
     test 'diagnostics'
     test 'type_inference'
     test 'find_lib'
diff --git a/server/test/rename/init.lua b/server/test/rename/init.lua
new file mode 100644
index 00000000..09900e3d
--- /dev/null
+++ b/server/test/rename/init.lua
@@ -0,0 +1,85 @@
+local core = require 'core'
+local parser  = require 'parser'
+local buildVM = require 'vm'
+
+local function catch_target(script)
+    local list = {}
+    local cur = 1
+    while true do
+        local start, finish  = script:find('<[!?].-[!?]>', cur)
+        if not start then
+            break
+        end
+        list[#list+1] = { start + 2, finish - 2 }
+        cur = finish + 1
+    end
+    return list
+end
+
+local function founded(targets, results)
+    if #targets ~= #results then
+        return false
+    end
+    for _, target in ipairs(targets) do
+        for _, result in ipairs(results) do
+            if target[1] == result[1] and target[2] == result[2] then
+                goto NEXT
+            end
+        end
+        do return false end
+        ::NEXT::
+    end
+    return true
+end
+
+function TEST(newName)
+    return function (script)
+        local target = catch_target(script)
+        local start  = script:find('<?', 1, true)
+        local finish = script:find('?>', 1, true)
+        local pos = (start + finish) // 2 + 1
+        local new_script = script:gsub('<[!?]', '  '):gsub('[!?]>', '  ')
+        local ast = parser:ast(new_script)
+        assert(ast)
+        local vm = buildVM(ast)
+        assert(vm)
+
+        local positions = core.rename(vm, pos, newName)
+        if positions then
+            assert(founded(target, positions))
+        else
+            assert(#target == 0)
+        end
+    end
+end
+
+TEST 'b' [[
+local <?a?> = 1
+]]
+
+TEST 'b' [[
+local <?a?> = 1
+<!a!> = 2
+<!a!> = <!a!>
+]]
+
+TEST 'b' [[
+t.<?a?> = 1
+a = t.<!a!>
+]]
+
+TEST 'b' [[
+t[<!'a'!>] = 1
+a = t.<?a?>
+]]
+
+TEST 'b' [[
+:: <?a?> ::
+goto <!a!>
+]]
+
+TEST 'b' [[
+local function f(<!a!>)
+    return <?a?>
+end
+]]