Mock more fs

Author: PIC123

src/plugins/login/utils/simpleFileTokenCache.test.ts

@@ -2,17 +2,32 @@ import fs from "fs";
 import mockFs from "mock-fs";
 import { MockFactory } from "../../../test/mockFactory";
 import { SimpleFileTokenCache } from "./simpleFileTokenCache";
-import os from "os";
 
 describe("Simple File Token Cache", () => {
   const tokenFilePath = "slsTokenCache.json";
+  const mockReadDirSync = jest.fn();
+  const mockWriteFileSync = jest.fn(() => {
+    return
+  });
+  const mockMkDir = jest.fn((path) => {
+    return
+  });
 
   let fileContent = {
     entries: [],
     subscriptions: []
   };
 
+  beforeEach(() => {
+    fs.readdirSync = mockReadDirSync;
+    fs.mkdirSync = mockMkDir;
+    fs.writeFileSync = mockWriteFileSync;
+  })
+
   afterEach(() => {
+    mockReadDirSync.mockRestore();
+    mockMkDir.mockRestore();
+    mockWriteFileSync.mockRestore();
     mockFs.restore();
   });
 
@@ -36,19 +51,9 @@ describe("Simple File Token Cache", () => {
 
   it("Create .azure default directory if it doesn't exist", () => {
     mockFs();
-    const mockMkDir = jest.fn((path) => {
-      return
-    });
-    const mockHomedir = jest.fn(() => {
-      return ""
-    });
-    fs.mkdirSync = mockMkDir;
-    os.homedir = mockHomedir;
     new SimpleFileTokenCache();
 
     expect(mockMkDir).toBeCalled();
-    mockMkDir.mockRestore();
-    mockHomedir.mockRestore();
   });
 
   it("Load file on creation if available", () => {

src/plugins/login/utils/simpleFileTokenCache.ts

@@ -3,17 +3,19 @@ import path from "path";
 import os from "os";
 import * as adal from "adal-node";
 
-let CONFIG_DIRECTORY = path.join(os.homedir(), ".azure");
-const DEFAULT_SLS_TOKEN_FILE = path.join(CONFIG_DIRECTORY, "slsTokenCache.json");
+// let CONFIG_DIRECTORY = path.join(os.homedir(), ".azure");
+// const DEFAULT_SLS_TOKEN_FILE = path.join(CONFIG_DIRECTORY, "slsTokenCache.json");
 
 export class SimpleFileTokenCache implements adal.TokenCache {
   private entries: any[] = [];
   private subscriptions: any[] = [];
 
-  public constructor(private tokenPath: string = DEFAULT_SLS_TOKEN_FILE) {
+  public constructor(private tokenPath: string = path.join(path.join(os.homedir(), ".azure"), "slsTokenCache.json")) {
+    const CONFIG_DIRECTORY = path.join(os.homedir(), ".azure");
+    const DEFAULT_SLS_TOKEN_FILE = path.join(CONFIG_DIRECTORY, "slsTokenCache.json");
     if(tokenPath === DEFAULT_SLS_TOKEN_FILE && !fs.existsSync(CONFIG_DIRECTORY)) {
-      CONFIG_DIRECTORY = path.join(os.homedir(), ".azure");
-      this.tokenPath = CONFIG_DIRECTORY;
+      // CONFIG_DIRECTORY = path.join(os.homedir(), ".azure");
+      // this.tokenPath = CONFIG_DIRECTORY;
       fs.mkdirSync(CONFIG_DIRECTORY);
     }
     this.load();