fennel-ls/test/completion.fnl
Michele Campeotto 1061db98d0 Don't drop multisym delimiters at the end of a symbol
added a test for completions
2025-10-19 15:24:33 -05:00

322 lines
12 KiB
Fennel

(local faith (require :faith))
(local {: create-client
: position-past-end-of-text} (require :test.utils))
(local {: view} (require :fennel))
(local kinds
{:Text 1 :Method 2 :Function 3 :Constructor 4 :Field 5 :Variable 6 :Class 7
:Interface 8 :Module 9 :Property 10 :Unit 11 :Value 12 :Enum 13 :Keyword 14
:Snippet 15 :Color 16 :File 17 :Reference 18 :Folder 19 :EnumMember 20
:Constant 21 :Struct 22 :Event 23 :Operator 24 :TypeParameter 25})
(fn find [client params e]
(let [completions (or params.items params)]
(accumulate [result nil
i c (ipairs completions)
&until result]
(let [c (if params.itemDefaults
(collect [k v (pairs params.itemDefaults) &into (collect [k v (pairs c)] k v)] k v)
c)]
(if (or (and (= (type e) :string)
(= c.label e))
(and (= (type e) :table)
(or (= e.label nil)
(and (= (type e.label) :string) (= e.label c.label))
(and (= (type e.label) :function) (e.label c.label)))
(or (= e.kind nil)
(and (= (type e.kind) :number) (= e.kind c.kind))
(and (= (type e.kind) :function) (e.kind c.kind)))
(or (= e.filterText nil)
(and (= (type e.filterText) :string) (= e.filterText c.filterText))
(and (= (type e.filterText) :function) (e.filterText c.filterText)))
(or (= e.insertText nil)
(and (= (type e.insertText) :string) (= e.insertText c.insertText))
(and (= (type e.insertText) :function) (e.insertText c.insertText)))
(or (= e.documentation nil)
(let [c (if params.items (-> (client:completion-item-resolve c)
(. 1 :result))
c)]
(or
(and (= (type e.documentation) :string) (= e.documentation c.documentation))
(and (= (type e.documentation) :function) (e.documentation c.documentation))
(and (= e.documentation true) (not= nil c.documentation)))))
(or (= e.textEdit nil)
(let [c-textEdit (or c.textEdit {:range c.editRange :newText (or c.insertText c.label)})]
(and (= (type e.textEdit) :table)
(= e.textEdit.range.start.line c-textEdit.range.start.line)
(= e.textEdit.range.start.character c-textEdit.range.start.character)
(= e.textEdit.range.end.line c-textEdit.range.end.line)
(= e.textEdit.range.end.character c-textEdit.range.end.character)
(= e.textEdit.newText c-textEdit.newText)))
(and (= (type e.textEdit) :function) (e.textEdit c.textEdit)))))
i)))))
(fn check [file-contents expected unexpected ?bad-completions?]
(let [{: client : uri : cursor : text} (create-client file-contents (if (not ?bad-completions?)
{:capabilities
{:textDocument
{:completion
{:completionList
{:itemDefaults
[:editRange :data]}}}}}))
[{:result ?result}] (client:completion uri
(or cursor
(position-past-end-of-text text)))
completions (or ?result [])]
(each [_ e (ipairs unexpected)]
(let [i (find client completions e)]
(faith.= nil i (.. "Got unexpected completion: " (view e) "\n"
"from: " (view file-contents) "\n"
(view (. (or completions.items completions) i) {:escape-newlines? true})))))
(if (= (type expected) :table)
(each [_ e (ipairs expected)]
(let [i (find client completions e)]
(faith.is i (.. "Didn't get completion: " (view e) "\n"
"from: " (view file-contents) "\n"
(if (= (type e) :table)
(let [candidate (find client completions {:label e.label})]
(if candidate
(.. "Candidate that didn't match:\n"
(view (. (or completions.items completions) candidate)
{:escape-newlines? true}))
""))
"")))))
(expected completions))))
(fn test-basic []
;; basic scoping rules
(check "(local yes1 10)
(fn yes2 [no2])
(do (local no1 11))
(let [yes3 (fn [no3] no3)]
(let [{:item y} {:item {:es4 12}}]
(fn [yes5 {: yes6}]
(each [no4 {: no5} (pairs _G)]
nil)
(each [yes7 {: yes8} (pairs _G)]
|"
[:yes1 :yes2 :yes3 :y.es4 :yes5 :yes6 :yes7 :yes8
:_G :debug :table :table.insert :io :getmetatable :_VERSION :ipairs :pairs :next {:label :setmetatable :kind kinds.Function}
:true :false :.nan :.inf :nil
{:label "coroutine.yield"
:documentation #(and $.value ($.value:find "```fnl\n(coroutine.yield ...)\n```" 1 true))}]
[:no1 :no2 :no3 :no4 :no5 :+
:this-variable-does-not-exist
:_G.coroutine.yield
:_G._G.coroutine.yield
#(and (= nil $.documentation)
(not= $.label :yes8))])
;; no duplicates
(check "(let [x 10] (let [x 10] x"
(fn [completions]
(faith.= 1 (accumulate [number-of-x 0 _ completion (ipairs completions.items)]
(if (= completion.label :x)
(+ number-of-x 1)
number-of-x))))
[])
(check "(let [f| 10] foo foo foo"
(fn [completions]
(faith.= 1 (accumulate [number-of-foo 0 _ completion (ipairs completions.items)]
(if (= completion.label :foo)
(+ number-of-foo 1)
number-of-foo))))
[])
;; completions of fields (nested)
(check "(local x {:y {:z {:a {:b 1}}}}) ; deep tables
(local m {}) ; split modules
(fn m.function [])
(fn m.method [self])
(local m2 {:function m.function :method m.method})
|"
[:x :x.y :x.y.z.a.b
:m.function
:m:method
:m2.function
:m2:method]
[])
;; regression test for not crashing
(check "(local x {:field (fn [self])})\n(x::f" [:x:field] [])
nil)
(fn test-builtin []
;; specials and macros are only suggested in callable positions
(check "(macro funny [] `nil)
(|)"
[:do :let :fn :doto :-> :-?>> :?. :funny]
[])
(check "(do |)"
[]
[:do :let :fn :doto :-> :-?>> :?.])
;; tricky case
(check "(doto f |)"
[:do :let :fn :doto :-> :-?>> :?.]
[])
nil)
(fn test-local-in-macro []
(check "(local item 10)\n(case 1 1 it|)" [:item] [])
nil)
(fn test-field []
(check
"(let [my-table {:foo 10 :bar 20}]\n my-table.|)))"
[{:label :my-table.foo :kind kinds.Field}
{:label :my-table.bar :kind kinds.Field}]
[])
nil)
(fn test-docs []
;; things that aren't present in lua5.4 but are in other versions, I guess??
(local things-that-are-allowed-to-have-missing-docs
{:lua 1 :set-forcibly! 1}) ;:unpack 1 :setfenv 1 :getfenv 1 :module 1 :newproxy 1 :gcinfo 1 :loadstring 1 :bit 1 :jit 1 :bit32 1})
(check "(let [x (fn x [a b c]
\"docstring\"
nil)
t {: x}]
(t."
[:x]
[{:documentation #(= nil $)}
{:kind #(= nil $)}
{:label #(= nil $)}])
(check "(case {}
identifier |"
[:identifier]
[]) ; TODO [{:documentation #(= nil $)}])
(each [_ mode (ipairs [true false])]
(check "(fn x [a b c]
\"docstring\"
nil)
(let [str :hi]
(|))"
[;; builtin specials
{:label :local
:kind kinds.Operator
:documentation true}
;; builtin macros
{:label :-?>
:kind kinds.Keyword
:documentation true}
;; builtin globals
{:label :table
:kind kinds.Module
:documentation true}
{:label :_G
:kind kinds.Variable
:documentation true}
;; method fields
{:label :str:gsub :kind kinds.Method :documentation true}
{:label :str:match :kind kinds.Method :documentation true}
{:label :str:match :kind kinds.Method :documentation true}
{:label :str:sub :kind kinds.Method :documentation true}
{:label :str:len :kind kinds.Method :documentation true}
{:label :str:find :kind kinds.Method :documentation true}
;; things in scope
{:label :x :kind kinds.Function :documentation true}]
[{:documentation #(= nil $) :label #(not (. things-that-are-allowed-to-have-missing-docs $))}
{:kind #(= nil $) :label #(not (. things-that-are-allowed-to-have-missing-docs $))}
{:label #(= nil $)}]
mode))
nil)
(fn test-module []
(check
{:main.fnl "(let [foo (require :fooo)
bar (require :baar)]
|)"
:fooo.fnl "(fn my-export [x] (print x))
{: my-export :constant 10}"
:baar.fnl "(local M {:constant 10})
(fn M.my-export [x] (print x))
M"}
[:foo.my-export :foo.constant
:bar.my-export :bar.constant]
[])
nil)
(fn test-destructure []
;; this is in a destructure location, so we don't want all the normal completions
(check "(local |)
(print foo)"
[:foo]
[:math])
(check "(let [f|]
(print foo))"
[:foo]
[:math])
nil)
(fn test-no-completion []
;; comment
(check "; ("
[]
[:math])
;; inside of a string
(check "\" (|\n\""
[]
[:math])
;; inside of a string, but the starting quote is on a different line
(check "\"\n(|\""
[]
[:math])
(check "(fn foo| [])"
[]
["foo|"])
nil)
(fn test-compiler-env []
(check ";; fennel-ls: macro-file\n("
[:sym :sym? :list :quote :icollect :math :ipairs]
[:os])
;; TODO #73
; (check "(macro foo [] (|"
; [:sym :sym? :list :quote :icollect :math]
; [:os])
nil)
;; ensure that we get completions for both the symbol and the table fields
(fn test-sym-ending-in-delimiter []
(check "(local a {:b 1 :c 2})
(local a. 3)
(set a.|"
["a." "a.b" "a.c"]
[])
nil)
;; ;; Future tests / features
;; ;; Scope Ordering Rules
;; (it "does not suggest locals past the suggestion location when a symbol is partially typed")
;; (it "does not suggest locals past the suggestion location without a symbol")
;; (it "does not suggest locals past the suggestion point at the top level")
;; (it "does not suggest items from later definitions in the same `let`")
;; (it "does suggest items from earlier definitions in the same `let`")
;; (it "does not suggest macros defined from later definitions")
;; (it "offers rich information about module completions")
;; (it "offers rich information about macro-module completions")))
;; (it "suggests known module names in `require` and `include` and `import-macros` and `require-macros` and friends")
;; (it "suggests known fn keys when using the `:` special")
;; (it "suggests known keys when using the `.` special")
;; (it "does not suggest special forms for the \"call\" position when a list isn't actually a call, ie destructuring assignment")
;; (it "suggests keys when typing out destructuring, as in `(local {: typinghere} (require :mod))`")
;; (it "only suggests tables for `ipairs` / begin work on type checking system")
{: test-basic
: test-builtin
: test-local-in-macro
: test-field
: test-docs
: test-module
: test-destructure
: test-no-completion
: test-compiler-env
: test-sym-ending-in-delimiter}