1 files changed, 145 insertions, 0 deletions
diff --git a/test/plugins/ast/test.lua b/test/plugins/ast/test.lua
new file mode 100644
index 00000000..a01b9e2e
--- /dev/null
+++ b/test/plugins/ast/test.lua
@@ -0,0 +1,145 @@
+local parser  = require 'parser'
+local guide   = require 'parser.guide'
+local helper  = require 'plugins.astHelper'
+
+---@diagnostic disable: await-in-sync
+---@param ... function checkers
+local function TestPlugin(script, plugin, ...)
+    local state = parser.compile(script, "Lua", "Lua 5.4")
+    state.ast = plugin(TESTURI, state.ast) or state.ast
+    parser.luadoc(state)
+    for i = 1, select('#', ...) do
+        select(i, ...)(state)
+    end
+end
+
+local function isDocClass(ast)
+    return ast.bindDocs[1].type == 'doc.class'
+end
+
+local function TestAIsClass(state, next)
+    assert(isDocClass(state.ast[1]))
+end
+
+--- when call Class
+local function plugin_AddClass(uri, ast)
+    guide.eachSourceType(ast, "call", function (source)
+        local node = source.node
+        if not guide.isGet(node) then
+            return
+        end
+        if not guide.isGlobal(node) then
+            return
+        end
+        if guide.getKeyName(node) ~= 'Class' then
+            return
+        end
+        local wants = {
+            ['local'] = true,
+            ['setglobal'] = true
+        }
+        local classnameNode = guide.getParentTypes(source, wants)
+        if not classnameNode then
+            return
+        end
+        local classname = guide.getKeyName(classnameNode)
+        if classname then
+            helper.addClassDoc(ast, classnameNode, classname)
+        end
+    end)
+end
+
+local function plugin_AddClassAtParam(uri, ast)
+    guide.eachSourceType(ast, "function", function (src)
+        helper.addClassDocAtParam(ast, "A", src, 1)
+    end)
+end
+
+local function TestSelfIsClass(state, next)
+    guide.eachSourceType(state.ast, "local", function (source)
+        if source[1] == 'self' then
+            assert(source.bindDocs)
+            assert(source.parent.type == 'function')
+            assert(#source.parent.args == 0)
+        end
+    end)
+end
+
+local function TestAHasField(state, next)
+    local foundFields = false
+    local cb = function (source)
+        if not source.bindDocs or not source.bindDocs[1].class or source.bindDocs[1].class[1] ~= "A" then
+            return
+        end
+        assert(#source.bindDocs == 1)
+        assert(#source.bindDocs[1].fields >= 2)
+        assert(source.bindDocs[1].fields[1].field[1] == "x")
+        assert(source.bindDocs[1].fields[2].field[1] == "y")
+        foundFields = true
+    end
+    guide.eachSourceType(state.ast, "local", cb)
+    guide.eachSourceType(state.ast, "setglobal", cb)
+    assert(foundFields)
+end
+
+local function plugin_AddMultipleDocs(uri, ast)
+    guide.eachSourceType(ast, "call", function(source)
+        local node = source.node
+        if guide.getKeyName(node) ~= 'Class' then
+            return
+        end
+        local wants = {
+            ['local'] = true,
+            ['setglobal'] = true
+        }
+        local classnameNode = guide.getParentTypes(source, wants)
+        if not classnameNode then
+            return
+        end
+        local classname = guide.getKeyName(classnameNode)
+        if classname then
+            local group = {}
+            helper.addClassDoc(ast, classnameNode, classname, group)
+            helper.addDoc(ast, classnameNode, "field", "x number", group)
+            helper.addDoc(ast, classnameNode, "field", "y string", group)
+        end
+    end)
+end
+
+local function TestPlugin1(script)
+    TestPlugin(script, plugin_AddClass, TestAIsClass)
+end
+
+local function TestPlugin2(script)
+    TestPlugin(script, plugin_AddClassAtParam, TestSelfIsClass)
+end
+
+local function TestPlugin3(script)
+    TestPlugin(script, plugin_AddMultipleDocs, TestAIsClass, TestAHasField)
+end
+
+TestPlugin1 [[
+    local A = Class(function() end)
+]]
+
+TestPlugin1 [[
+    A = Class(function() end)
+]]
+
+TestPlugin2 [[
+    local function ctor(self) end
+]]
+
+TestPlugin2 [[
+    function ctor(self) end
+]]
+
+TestPlugin3 [[
+    local A = Class()
+]]
+
+TestPlugin3 [[
+    A = Class()
+]]
+
+require 'plugins.ast.helper'