|
| 1 | +import * as assert from 'assert'; |
| 2 | +import * as path from 'path'; |
| 3 | +import ts from 'typescript'; |
| 4 | +import { Location, Position, Range } from 'vscode-languageserver'; |
| 5 | +import { Document, DocumentManager } from '../../../../src/lib/documents'; |
| 6 | +import { FindReferencesProviderImpl } from '../../../../src/plugins/typescript/features/FindReferencesProvider'; |
| 7 | +import { LSAndTSDocResolver } from '../../../../src/plugins/typescript/LSAndTSDocResolver'; |
| 8 | +import { pathToUrl } from '../../../../src/utils'; |
| 9 | + |
| 10 | +const testDir = path.join(__dirname, '..'); |
| 11 | + |
| 12 | +describe('FindReferencesProvider', () => { |
| 13 | + function getFullPath(filename: string) { |
| 14 | + return path.join(testDir, 'testfiles', filename); |
| 15 | + } |
| 16 | + function getUri(filename: string) { |
| 17 | + const filePath = path.join(testDir, 'testfiles', filename); |
| 18 | + return pathToUrl(filePath); |
| 19 | + } |
| 20 | + |
| 21 | + function setup(filename: string) { |
| 22 | + const docManager = new DocumentManager( |
| 23 | + (textDocument) => new Document(textDocument.uri, textDocument.text), |
| 24 | + ); |
| 25 | + const lsAndTsDocResolver = new LSAndTSDocResolver(docManager, [testDir]); |
| 26 | + const provider = new FindReferencesProviderImpl(lsAndTsDocResolver); |
| 27 | + const document = openDoc(filename); |
| 28 | + return { provider, document }; |
| 29 | + |
| 30 | + function openDoc(filename: string) { |
| 31 | + const filePath = getFullPath(filename); |
| 32 | + const doc = docManager.openDocument(<any>{ |
| 33 | + uri: pathToUrl(filePath), |
| 34 | + text: ts.sys.readFile(filePath) || '', |
| 35 | + }); |
| 36 | + return doc; |
| 37 | + } |
| 38 | + } |
| 39 | + |
| 40 | + async function test(position: Position, includeDeclaration: boolean) { |
| 41 | + const { provider, document } = setup('find-references.svelte'); |
| 42 | + |
| 43 | + const results = await provider.findReferences(document, position, { includeDeclaration }); |
| 44 | + |
| 45 | + let expectedResults = [ |
| 46 | + Location.create( |
| 47 | + getUri('find-references.svelte'), |
| 48 | + Range.create(Position.create(2, 8), Position.create(2, 14)), |
| 49 | + ), |
| 50 | + Location.create( |
| 51 | + getUri('find-references.svelte'), |
| 52 | + Range.create(Position.create(3, 8), Position.create(3, 14)), |
| 53 | + ), |
| 54 | + ]; |
| 55 | + if (includeDeclaration) { |
| 56 | + expectedResults = [ |
| 57 | + Location.create( |
| 58 | + getUri('find-references.svelte'), |
| 59 | + Range.create(Position.create(1, 10), Position.create(1, 16)), |
| 60 | + ), |
| 61 | + ].concat(expectedResults); |
| 62 | + } |
| 63 | + |
| 64 | + assert.deepStrictEqual(results, expectedResults); |
| 65 | + } |
| 66 | + |
| 67 | + it('finds references', async () => { |
| 68 | + await test(Position.create(1, 11), true); |
| 69 | + }); |
| 70 | + |
| 71 | + it('finds references, exluding definition', async () => { |
| 72 | + await test(Position.create(1, 11), false); |
| 73 | + }); |
| 74 | + |
| 75 | + it('finds references (not searching from declaration)', async () => { |
| 76 | + await test(Position.create(2, 8), true); |
| 77 | + }); |
| 78 | +}); |
0 commit comments