fix(tests): Update tests for ggshieldAuthStatus

Author: salome-voltz

package.json

@@ -135,13 +135,15 @@
     "@types/glob": "^8.1.0",
     "@types/mocha": "^10.0.1",
     "@types/node": "20.2.5",
+    "@types/simple-mock": "^0.8.6",
     "@types/vscode": "^1.81.0",
     "@typescript-eslint/eslint-plugin": "^5.59.8",
     "@typescript-eslint/parser": "^5.59.8",
     "@vscode/test-electron": "^2.3.2",
     "eslint": "^8.41.0",
     "glob": "^8.1.0",
     "mocha": "^10.2.0",
+    "simple-mock": "^0.8.0",
     "typescript": "^5.1.3"
   },
   "dependencies": {

src/lib/ggshield-api.ts

@@ -3,7 +3,7 @@ import {
   SpawnOptionsWithoutStdio,
   spawn,
 } from "child_process";
-import { window, WebviewView } from "vscode";
+import { window, WebviewView, ExtensionContext, commands } from "vscode";
 import axios from 'axios';
 import { GGShieldConfiguration } from "./ggshield-configuration";
 import { GGShieldScanResults } from "./api-types";

src/test/suite/authentication.test.ts

@@ -0,0 +1,57 @@
+import * as assert from 'assert';
+import * as simple from 'simple-mock';
+import { ExtensionContext, Memento } from 'vscode';
+import { ggshieldAuthStatus } from '../../lib/ggshield-api';
+import * as runGGShield from '../../lib/run-ggshield';
+import { GGShieldConfiguration } from '../../lib/ggshield-configuration';
+
+suite('ggshieldAuthStatus', function () {
+    let isAuthenticated: boolean;
+    let mockGlobalState: Memento & { setKeysForSync(keys: readonly string[]): void; };
+    let mockContext: Partial<ExtensionContext>;
+
+    setup(function () {
+        isAuthenticated = false;
+
+        mockGlobalState = {
+            get: (key: string) => (key === 'isAuthenticated' ? isAuthenticated : undefined),
+            update: (key: string, value: any) => {
+                if (key === 'isAuthenticated') {
+                    isAuthenticated = value;
+                }
+                return Promise.resolve();
+            },
+            keys: () => [],
+            setKeysForSync: (keys: readonly string[]) => {},
+        };
+
+        mockContext = {
+            globalState: mockGlobalState,
+        };
+    });
+    teardown(function () {
+        simple.restore();
+    });
+
+    test('Valid authentication should update isAuthenticated to true', async function () {
+        simple.mock(runGGShield, "runGGShieldCommand").returnWith({
+            status: 0,
+            stdout: '{"detail": "Valid API key.", "status_code": 200, "app_version": "v2.111.0", "secrets_engine_version": "2.124.1"}',
+            stderr: ''
+        });
+
+        await ggshieldAuthStatus({} as GGShieldConfiguration, mockContext as ExtensionContext);
+        assert.strictEqual(isAuthenticated, true);
+    });
+
+    test('Invalid authentication should keep isAuthenticated to false', async function () {
+        simple.mock(runGGShield, "runGGShieldCommand").returnWith({
+            status: 0,
+            stdout: '{"detail": "Invalid API key.", "status_code": 401, "app_version": "v2.111.0", "secrets_engine_version": "2.124.1"}',
+            stderr: ''
+        });
+
+        await ggshieldAuthStatus({} as GGShieldConfiguration, mockContext as ExtensionContext);
+        assert.strictEqual(isAuthenticated, false);
+    });
+});