Improve tests

Author: paleite

src/git.test.ts

@@ -106,10 +106,11 @@ describe("getDiffFileList", () => {
     jest.mock("child_process").resetAllMocks();
     mockedChildProcess.execSync.mockReturnValueOnce(Buffer.from(diffFileList));
     const staged = false;
+    expect(mockedChildProcess.execSync).toHaveBeenCalledTimes(0);
     const fileListA = getDiffFileList(staged);
     const fileListB = getDiffFileList(staged);
 
-    expect(mockedChildProcess.execSync).toHaveBeenCalled();
+    expect(mockedChildProcess.execSync).toHaveBeenCalledTimes(1);
     expect(fileListA).toEqual(
       ["file1", "file2", "file3"].map((p) => path.resolve(p))
     );
@@ -119,6 +120,7 @@ describe("getDiffFileList", () => {
 
 describe("getGitFileList", () => {
   it("should get the list of committed files", () => {
+    mockedChildProcess.execSync.mockReturnValueOnce(Buffer.from(diffFileList));
     expect(getGitFileList()).toEqual(
       ["file1", "file2", "file3"].map((p) => path.resolve(p))
     );

src/git.ts

@@ -102,6 +102,27 @@ const hasCleanIndex = (filePath: string): boolean => {
   return result;
 };
 
+let untrackedFileListCache: string[] | undefined;
+const getUntrackedFileList = (staged = false): string[] => {
+  if (untrackedFileListCache === undefined) {
+    const command = ["git", "ls-files", "--exclude-standard", "--others"]
+      .filter(Boolean)
+      .join(" ");
+
+    if (staged === false) {
+      untrackedFileListCache = child_process
+        .execSync(command)
+        .toString()
+        .trim()
+        .split("\n")
+        .map((filePath) => path.resolve(filePath));
+    } else {
+      untrackedFileListCache = [];
+    }
+  }
+  return untrackedFileListCache;
+};
+
 const isHunkHeader = (input: string) => {
   const hunkHeaderRE = /^@@ [^@]* @@/u;
   return hunkHeaderRE.exec(input);
@@ -157,10 +178,11 @@ const getRangesForDiff = (diff: string): Range[] =>
   }, []);
 
 export {
+  getDiffFileList,
   getDiffForFile,
   getRangesForDiff,
-  getDiffFileList,
   getGitFileList,
+  getUntrackedFileList,
   hasCleanIndex,
 };
 export type { Range };