1 files changed, 51 insertions, 0 deletions
diff --git a/server/test/find_lib/init.lua b/server/test/find_lib/init.lua
new file mode 100644
index 00000000..29297bcf
--- /dev/null
+++ b/server/test/find_lib/init.lua
@@ -0,0 +1,51 @@
+local matcher    = require 'matcher'
+local parser     = require 'parser'
+local findResult = require 'matcher.find_result'
+local findLib    = require 'matcher.find_lib'
+
+rawset(_G, 'TEST', true)
+
+function TEST(fullkey)
+    return function (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 results = matcher.compile(ast)
+        assert(results)
+        local result = findResult(results, pos)
+        assert(result)
+        assert(result.type == 'var')
+        local var = result.var
+        assert(var)
+        local _, name = findLib(var)
+        assert(name)
+        assert(name == fullkey)
+    end
+end
+
+TEST 'require' [[
+<?require?> 'xxx'
+]]
+
+TEST 'require' [[
+local <?req?> = require
+]]
+
+TEST 'require' [[
+local req = require
+local t = {
+    xx = req,
+}
+t[<?'xx'?>]()
+]]
+
+TEST 'table' [[
+<?table?>.unpack()
+]]
+
+TEST 'table' [[
+local <?xx?> = require 'table'
+]]