1 files changed, 106 insertions, 0 deletions
diff --git a/test/find_lib/init.lua b/test/find_lib/init.lua
new file mode 100644
index 00000000..51f53b75
--- /dev/null
+++ b/test/find_lib/init.lua
@@ -0,0 +1,106 @@
+local core    = require 'core'
+local parser  = require 'parser'
+local buildVM = require 'vm'
+
+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:parse(new_script, 'lua', 'Lua 5.3')
+        assert(ast)
+        local vm = buildVM(ast)
+        assert(vm)
+        local source = core.findSource(vm, pos)
+        local _, name = core.findLib(source)
+        assert(name == fullkey)
+    end
+end
+
+TEST 'require' [[
+<?require?> 'xxx'
+]]
+
+TEST 'req<require>' [[
+local <?req?> = require
+]]
+
+TEST 'req<require>' [[
+local req = require
+local t = {
+    xx = req,
+}
+t[<?'xx'?>]()
+]]
+
+TEST 'table' [[
+<?table?>.unpack()
+]]
+
+TEST 'xx<table>' [[
+local <?xx?> = require 'table'
+]]
+
+TEST 'xx<table>' [[
+local rq = require
+local lib = 'table'
+local <?xx?> = rq(lib)
+]]
+
+TEST 'table.insert' [[
+table.<?insert?>()
+]]
+
+TEST 'table.insert' [[
+local t = table
+t.<?insert?>()
+]]
+
+TEST 'table.insert' [[
+local insert = table.insert
+<?insert?>()
+]]
+
+TEST 'table.insert' [[
+local t = require 'table'
+t.<?insert?>()
+]]
+
+TEST 'table.insert' [[
+require 'table'.<?insert?>()
+]]
+
+TEST '*string.sub' [[
+local str = 'xxx'
+str.<?sub?>()
+]]
+
+TEST '*string:sub' [[
+local str = 'xxx'
+str:<?sub?>(1, 1)
+]]
+
+TEST '*string.sub' [[
+('xxx').<?sub?>()
+]]
+
+TEST 'fs<bee::filesystem>' [[
+local <?fs?> = require 'bee.filesystem'
+]]
+
+TEST 'fs.current_path' [[
+local filesystem = require 'bee.filesystem'
+
+ROOT = filesystem.<?current_path?>()
+]]
+
+TEST(nil)[[
+print(<?insert?>)
+]]
+
+TEST '_G' [[
+local x = <?_G?>
+]]