1 files changed, 134 insertions, 0 deletions
diff --git a/test/basic/linker.lua b/test/basic/linker.lua
new file mode 100644
index 00000000..bf4e50bb
--- /dev/null
+++ b/test/basic/linker.lua
@@ -0,0 +1,134 @@
+local linker = require 'core.linker'
+local files  = require 'files'
+local util   = require 'utility'
+local guide  = require 'core.guide'
+
+local function getSource(pos)
+    local ast = files.getAst('')
+    return guide.eachSourceContain(ast.ast, pos, function (source)
+        if source.type == 'local'
+        or source.type == 'getlocal'
+        or source.type == 'setlocal'
+        or source.type == 'setglobal'
+        or source.type == 'getglobal'
+        or source.type == 'field'
+        or source.type == 'method' then
+            return source
+        end
+    end)
+end
+
+local CARE = {}
+local function TEST(script)
+    return function (expect)
+        files.removeAll()
+        local start  = script:find('<?', 1, true)
+        local finish = script:find('?>', 1, true)
+        local pos = (start + finish) // 2 + 1
+        local newScript = script:gsub('<[!?]', '  '):gsub('[!?]>', '  ')
+        files.setText('', newScript)
+        local source = getSource(pos)
+        assert(source)
+        local result = linker.getLink(source)
+        for key in pairs(CARE) do
+            assert(result[key] == expect[key])
+        end
+    end
+end
+
+CARE['id'] = true
+TEST [[
+local <?x?>
+]] {
+    id = '9',
+}
+
+TEST [[
+local x
+print(<?x?>)
+]] {
+    id = '7',
+}
+
+TEST [[
+local x
+<?x?> = 1
+]] {
+    id = '7',
+}
+
+CARE['global'] = true
+TEST [[
+print(<?X?>)
+]] {
+    id     = '"X"',
+    global = true,
+}
+
+TEST [[
+print(<?X?>)
+]] {
+    id     = '"X"',
+    global = true,
+}
+
+TEST [[
+local x
+print(x.y.<?z?>)
+]] {
+    id     = '7|"y"|"z"',
+}
+
+TEST [[
+local x
+function x:<?f?>() end
+]] {
+    id     = '7|"f"',
+}
+
+TEST [[
+print(X.Y.<?Z?>)
+]] {
+    id     = '"X"|"Y"|"Z"',
+    global = true,
+}
+
+TEST [[
+function x:<?f?>() end
+]] {
+    id     = '"x"|"f"',
+    global = true,
+}
+
+CARE['tfield'] = true
+TEST [[
+{
+    <?x?> = 1,
+}
+]] {
+    id     = '1|"x"',
+    tfield = true,
+}
+
+CARE['freturn'] = true
+TEST [[
+return <?X?>
+]] {
+    id      = '"X"',
+    global  = true,
+    freturn = 0,
+}
+
+TEST [[
+function f()
+    return <?X?>
+end
+]] {
+    id      = '"X"',
+    global  = true,
+    freturn = 1,
+}
+
+TEST [[
+
+]]