Commit e87f10e

committed

feat. Support custom test case in source code.

1 parent 1893d34 commit e87f10eCopy full SHA for e87f10e

File tree

+38

-3

lines changed

+38

-3

lines changed

Lines changed: 23 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -56,10 +56,32 @@ export class CustomCodeLensProvider implements vscode.CodeLensProvider {`
`56`	`56`	`}`
`57`	`57`
`58`	`58`	`if (shortcuts.indexOf("test") >= 0) {`
	`59`	`+ let testLineStart: number = -1;`
	`60`	`+ for (let i: number = document.lineCount - 1; i >= 0; i--) {`
	`61`	`+ const lineContent: string = document.lineAt(i).text;`
	`62`	`+ if (lineContent.indexOf("@lc test=start") >= 0) {`
	`63`	`+ testLineStart = i;`
	`64`	`+ break;`
	`65`	`+ }`
	`66`	`+ }`
	`67`	`+ let testLineEnd: number = -1;`
	`68`	`+ if (testLineStart >= 0) {`
	`69`	`+ for (let i: number = document.lineCount - 1; i >= 0; i--) {`
	`70`	`+ const lineContent: string = document.lineAt(i).text;`
	`71`	`+ if (lineContent.indexOf("@lc test=end") >= 0) {`
	`72`	`+ testLineEnd = i;`
	`73`	`+ break;`
	`74`	`+ }`
	`75`	`+ }`
	`76`	`+ }`
	`77`	`+ const testStringInSource: string \| null = testLineEnd >= 0`
	`78`	`+ ? document.getText(new vscode.Range(testLineStart + 1, 0, testLineEnd, 0))`
	`79`	`+ : null;`
	`80`	`+`
`59`	`81`	`codeLens.push(new vscode.CodeLens(range, {`
`60`	`82`	`title: "Test",`
`61`	`83`	`command: "leetcode.testSolution",`
`62`		`- arguments: [document.uri],`
	`84`	`+ arguments: [document.uri,testStringInSource],`
`63`	`85`	`}));`
`64`	`86`	`}`
`65`	`87`

Lines changed: 14 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,7 @@ import { getActiveFilePath } from "../utils/workspaceUtils";`
`12`	`12`	`import * as wsl from "../utils/wslUtils";`
`13`	`13`	`import { leetCodeSubmissionProvider } from "../webview/leetCodeSubmissionProvider";`
`14`	`14`
`15`		`-export async function testSolution(uri?: vscode.Uri): Promise<void> {`
	`15`	`+export async function testSolution(uri?: vscode.Uri,testContentInSource?: string): Promise<void> {`
`16`	`16`	`try {`
`17`	`17`	`if (leetCodeManager.getStatus() === UserStatus.SignedOut) {`
`18`	`18`	`return;`
`@@ -30,6 +30,12 @@ export async function testSolution(uri?: vscode.Uri): Promise<void> {`
`30`	`30`	`detail: "Test with the default cases",`
`31`	`31`	`value: ":default",`
`32`	`32`	`},`
	`33`	`+ {`
	`34`	`+ label: "$(code) From source",`
	`35`	`+ description: "",`
	`36`	`+ detail: "Test with the cases in source code",`
	`37`	`+ value: ":source",`
	`38`	`+ },`
`33`	`39`	`{`
`34`	`40`	`label: "$(pencil) Write directly...",`
`35`	`41`	`description: "",`
`@@ -75,6 +81,13 @@ export async function testSolution(uri?: vscode.Uri): Promise<void> {`
`75`	`81`	`}`
`76`	`82`	`}`
`77`	`83`	`break;`
	`84`	`+ case ":source":`
	`85`	`+ if (testContentInSource) {`
	`86`	`+ result = await leetCodeExecutor.testSolution(filePath, parseTestString(testContentInSource.replace(/\r?\n/g, "\\n")));`
	`87`	`+ } else {`
	`88`	`+ vscode.window.showErrorMessage("Failed to load test cases from source code.");`
	`89`	`+ }`
	`90`	`+ break;`
`78`	`91`	`default:`
`79`	`92`	`break;`
`80`	`93`	`}`

Lines changed: 1 addition & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -59,7 +59,7 @@ export async function activate(context: vscode.ExtensionContext): Promise<void>`
`59`	`59`	`vscode.commands.registerCommand("leetcode.searchProblem", () => show.searchProblem()),`
`60`	`60`	`vscode.commands.registerCommand("leetcode.showSolution", (input: LeetCodeNode \| vscode.Uri) => show.showSolution(input)),`
`61`	`61`	`vscode.commands.registerCommand("leetcode.refreshExplorer", () => leetCodeTreeDataProvider.refresh()),`
`62`		`- vscode.commands.registerCommand("leetcode.testSolution", (uri?: vscode.Uri) => test.testSolution(uri)),`
	`62`	`+ vscode.commands.registerCommand("leetcode.testSolution", (uri?: vscode.Uri,testContentInSource?: string) => test.testSolution(uri,testContentInSource)),`
`63`	`63`	`vscode.commands.registerCommand("leetcode.submitSolution", (uri?: vscode.Uri) => submit.submitSolution(uri)),`
`64`	`64`	`vscode.commands.registerCommand("leetcode.switchDefaultLanguage", () => switchDefaultLanguage()),`
`65`	`65`	`vscode.commands.registerCommand("leetcode.addFavorite", (node: LeetCodeNode) => star.addFavorite(node)),`

Comments

(0)