fix

2 files changed, 73 insertions, 0 deletions

diff --git a/test/tclient/tests/folders-with-single-file.lua b/test/tclient/tests/folders-with-single-file.lua
new file mode 100644
index 00000000..326dd623
--- /dev/null
+++ b/test/tclient/tests/folders-with-single-file.lua
@@ -0,0 +1,66 @@
+local lclient = require 'tclient.lclient'
+local fs      = require 'bee.filesystem'
+local util    = require 'utility'
+local furi    = require 'file-uri'
+local ws      = require 'workspace'
+local files   = require 'files'
+local scope   = require 'workspace.scope'
+
+---@async
+lclient():start(function (client)
+    client:registerFakers()
+
+    client:initialize {
+        rootUri = 'abc',
+    }
+
+    client:notify('textDocument/didOpen', {
+        textDocument = {
+            uri = furi.encode('abc/1.lua'),
+            languageId = 'lua',
+            version = 0,
+            text = [[
+local x
+print(x)
+]]
+        }
+    })
+
+    ws.awaitReady('abc')
+
+    local locations = client:awaitRequest('textDocument/definition', {
+        textDocument = { uri = furi.encode('abc/1.lua') },
+        position = { line = 1, character = 7 },
+    })
+
+    assert(util.equal(locations, {
+        {
+            uri = furi.encode('abc/1.lua'),
+            range = {
+                start   = { line = 0, character = 6 },
+                ['end'] = { line = 0, character = 7 },
+            }
+        }
+    }))
+
+    client:notify('textDocument/didOpen', {
+        textDocument = {
+            uri = 'test://single-file.lua',
+            languageId = 'lua',
+            version = 0,
+            text = [[
+local x
+print(x)
+]]
+        }
+    })
+
+    ws.awaitReady(nil)
+
+    local locations = client:awaitRequest('textDocument/definition', {
+        textDocument = { uri = 'test://single-file.lua' },
+        position = { line = 1, character = 0 },
+    })
+
+    assert(#locations > 0)
+end)
diff --git a/test/tclient/tests/single-mode.lua b/test/tclient/tests/single-mode.lua
index 0a7f85ff..10f3e2da 100644
--- a/test/tclient/tests/single-mode.lua
+++ b/test/tclient/tests/single-mode.lua
@@ -35,4 +35,11 @@ print(x)
             }
         }
     }))
+
+    local locations = client:awaitRequest('textDocument/definition', {
+        textDocument = { uri = 'test://single-file.lua' },
+        position = { line = 1, character = 0 },
+    })
+
+    assert(#locations > 0)
 end)