test: add tests for reference (#3546)

Co-authored-by: Johnson Chu <[email protected]>

Author: so1ve

.vscode/settings.json

@@ -16,8 +16,5 @@
 	"[jsonc]": {
 		"editor.defaultFormatter": "vscode.json-language-features"
 	},
-	"[vue]": {
-		"editor.defaultFormatter": "Vue.volar"
-	},
 	"vue.server.path": "./packages/vue-language-server/bin/vue-language-server.js",
 }

packages/vue-language-service/tests/index.spec.ts

@@ -1,3 +1,4 @@
 import './complete';
 import './findDefinition';
 import './rename';
+import './reference';

packages/vue-language-service/tests/reference.ts

@@ -0,0 +1,77 @@
+import { describe, expect, it } from 'vitest';
+import * as path from 'path';
+import { tester } from './utils/createTester';
+import { TextDocument } from 'vscode-languageserver-textdocument';
+import * as fs from 'fs';
+
+const baseDir = path.resolve(__dirname, '../../vue-test-workspace/reference');
+const testDirs = fs.readdirSync(baseDir);
+
+for (const dirName of testDirs) {
+
+	describe(`find reference: ${dirName}`, async () => {
+
+		const dir = path.join(baseDir, dirName);
+		const inputFiles = readFiles(dir);
+
+		for (const file in inputFiles) {
+
+			const filePath = path.join(dir, file);
+			const uri = tester.fileNameToUri(filePath);
+			const fileText = inputFiles[file];
+			const document = TextDocument.create('', '', 0, fileText);
+			const actions = findActions(fileText);
+
+			for (const action of actions) {
+
+				const position = document.positionAt(action.offset);
+
+				position.line--;
+
+				const location = `${filePath}:${position.line + 1}:${position.character + 1}`;
+
+				it(`${location} => count: ${action.count}`, async () => {
+
+					const locations = await tester.languageService.findReferences(
+						uri,
+						position,
+					);
+
+					expect(locations).toBeDefined();
+
+					expect(locations?.length).toBe(action.count);
+				});
+			}
+		}
+	});
+}
+
+function readFiles(dir: string) {
+
+	const filesText: Record<string, string> = {};
+	const files = fs.readdirSync(dir);
+
+	for (const file of files) {
+		const filePath = path.join(dir, file);
+		filesText[file] = fs.readFileSync(filePath, 'utf8');
+	}
+
+	return filesText;
+}
+
+const referenceReg = /(\^*)reference:\s*([\S]*)/g;
+
+function findActions(text: string) {
+
+	return [...text.matchAll(referenceReg)].map(flag => {
+
+		const offset = flag.index!;
+		// The definition itself is also counted
+		const count = Number(flag[2]) + 1;
+
+		return {
+			offset,
+			count,
+		};
+	});
+}

packages/vue-test-workspace/reference/slot-default/entry.vue

@@ -0,0 +1,4 @@
+<template>
+	<slot></slot>
+	<!-- ^reference: 1 -->
+</template>

packages/vue-test-workspace/reference/slot-default/foo.vue

@@ -0,0 +1,9 @@
+<script setup lang="ts">
+import Entry from './entry.vue';
+</script>
+
+<template>
+	<Entry>
+		<div></div>
+	</Entry>
+</template>

packages/vue-test-workspace/reference/slot-named/entry.vue

@@ -0,0 +1,4 @@
+<template>
+	<slot name="foo"></slot>
+	<!-- ^reference: 1 -->
+</template>

packages/vue-test-workspace/reference/slot-named/foo.vue

@@ -0,0 +1,7 @@
+<script setup lang="ts">
+import Entry from './entry.vue';
+</script>
+
+<template>
+	<Entry #foo></Entry>
+</template>