Fix the bug that quotes are not accepted in test cases (LeetCode-OpenSource#111)

jdneo · web-flow · commit e8da8334c2c7 · 2019-02-09T21:12:03.000+08:00
diff --git a/src/commands/test.ts b/src/commands/test.ts
@@ -59,15 +59,15 @@ export async function testSolution(uri?: vscode.Uri): Promise<void> {
                     ignoreFocusOut: true,
                 });
                 if (testString) {
-                    result = await leetCodeExecutor.testSolution(filePath, testString.replace(/"/g, ""));
+                    result = await leetCodeExecutor.testSolution(filePath, testString);
                 }
                 break;
             case ":file":
                 const testFile: vscode.Uri[] | undefined = await showFileSelectDialog();
                 if (testFile && testFile.length) {
                     const input: string = await fse.readFile(testFile[0].fsPath, "utf-8");
                     if (input.trim()) {
-                        result = await leetCodeExecutor.testSolution(filePath, input.replace(/"/g, "").replace(/\r?\n/g, "\\n"));
+                        result = await leetCodeExecutor.testSolution(filePath, input.replace(/\r?\n/g, "\\n"));
                     } else {
                         vscode.window.showErrorMessage("The selected test file must not be empty.");
                     }
diff --git a/src/leetCodeExecutor.ts b/src/leetCodeExecutor.ts
@@ -96,7 +96,7 @@ class LeetCodeExecutor {
 
     public async testSolution(filePath: string, testString?: string): Promise<string> {
         if (testString) {
-            return await this.executeCommandWithProgressEx("Submitting to LeetCode...", "node", [await this.getLeetCodeBinaryPath(), "test", `"${filePath}"`, "-t", `"${testString}"`]);
+            return await this.executeCommandWithProgressEx("Submitting to LeetCode...", "node", [await this.getLeetCodeBinaryPath(), "test", `"${filePath}"`, "-t", `'${testString}'`]);
         }
         return await this.executeCommandWithProgressEx("Submitting to LeetCode...", "node", [await this.getLeetCodeBinaryPath(), "test", `"${filePath}"`]);
     }
