Added unit tests

Author: BenAlvo1

src/main/osinstaller/CxInstaller.ts

@@ -25,7 +25,7 @@ export class CxInstaller {
     private readonly client: Client;
 
     private static readonly PLATFORMS: Record<SupportedPlatforms, PlatformData> = {
-        win32: { platform: winOS, extension: 'zip' },
+        win32: { platform: 'windows', extension: 'zip' },
         darwin: { platform: macOS, extension: 'tar.gz' },
         linux: { platform: linuxOS, extension: 'tar.gz' }
     };
@@ -36,7 +36,7 @@ export class CxInstaller {
         this.client = client;
     }
 
-    private async getDownloadURL(): Promise<string> {
+    async getDownloadURL(): Promise<string> {
         const cliVersion = await this.readASTCLIVersion();
         const platformData = CxInstaller.PLATFORMS[this.platform];
 
@@ -65,7 +65,7 @@ export class CxInstaller {
     }
 
     public getExecutablePath(): string {
-        const executableName = this.platform === 'win32' ? 'cx.exe' : 'cx';
+        const executableName = this.platform === winOS ? 'cx.exe' : 'cx';
         return path.join(this.resourceDirPath, executableName);
     }
 
@@ -170,7 +170,7 @@ export class CxInstaller {
         }
     }
 
-    private checkExecutableExists(): boolean {
+    public checkExecutableExists(): boolean {
         return fs.existsSync(this.getExecutablePath());
     }
 
@@ -195,4 +195,8 @@ export class CxInstaller {
     private getCompressFolderName(): string {
         return `ast-cli.${this.platform === winOS ? 'zip' : 'tar.gz'}`;
     }
+    
+    public getPlatform(): SupportedPlatforms {
+        return this.platform;
+    }
 }

src/tests/CxInstallerTest.test.ts

@@ -0,0 +1,78 @@
+import { CxInstaller } from "../main/osinstaller/CxInstaller";
+import { anyString, mock, instance, when, verify } from "ts-mockito";
+import { HttpClient } from "../main/client/HttpClient";
+
+const clientMock = mock(HttpClient);
+const clientMockInstance = instance(clientMock);
+
+const cxInstallerLinux = new CxInstaller("linux", clientMockInstance);
+const cxInstallerMac = new CxInstaller("darwin", clientMockInstance);
+const cxInstallerWindows = new CxInstaller("win32", clientMockInstance);
+
+describe("CxInstaller cases", () => {
+    it('CxInstaller getDownloadURL Linux Successful case', async () => {
+        const url = await cxInstallerLinux.getDownloadURL();
+        const architecture = getArchitecture(cxInstallerLinux.getPlatform());
+        expect(url).toBe(`https://download.checkmarx.com/CxOne/CLI/2.2.5/ast-cli_2.2.5_linux_${architecture}.tar.gz`);
+    });
+
+    it('CxInstaller getDownloadURL Mac Successful case', async () => {
+        const url = await cxInstallerMac.getDownloadURL();
+        const architecture = getArchitecture(cxInstallerMac.getPlatform());
+        expect(url).toBe(`https://download.checkmarx.com/CxOne/CLI/2.2.5/ast-cli_2.2.5_darwin_${architecture}.tar.gz`);
+    });
+
+    it('CxInstaller getDownloadURL Windows Successful case', async () => {
+        const url = await cxInstallerWindows.getDownloadURL();
+        const architecture = getArchitecture(cxInstallerWindows.getPlatform());
+        expect(url).toBe(`https://download.checkmarx.com/CxOne/CLI/2.2.5/ast-cli_2.2.5_windows_${architecture}.zip`);
+    });
+});
+
+describe("CxInstaller getExecutablePath cases", () => {
+    it('CxInstaller getExecutablePath Linux Successful case', () => {
+        const executablePath = cxInstallerLinux.getExecutablePath();
+        expect(executablePath).toContain(`src/main/wrapper/resources/cx`);
+    });
+
+    it('CxInstaller getExecutablePath Mac Successful case', () => {
+        const executablePath = cxInstallerMac.getExecutablePath();
+        expect(executablePath).toContain(`src/main/wrapper/resources/cx`);
+    });
+
+    it('CxInstaller getExecutablePath Windows Successful case', () => {
+        const executablePath = cxInstallerWindows.getExecutablePath();
+        expect(executablePath).toContain(`src/main/wrapper/resources/cx.exe`);
+    });
+});
+
+describe("CxInstaller checkExecutableExists cases", () => {
+    beforeAll(async () => {
+        // Set up mock behavior
+        when(clientMock.downloadFile(anyString(), anyString())).thenResolve();
+
+        // Trigger the download
+        await cxInstallerWindows.downloadIfNotInstalledCLI();
+    });
+
+    it('CxInstaller checkExecutableExists Windows Successful case', () => {
+        const exists = cxInstallerWindows.checkExecutableExists();
+        expect(exists).toBe(false);
+
+        // Verify if downloadFile was called with the expected arguments
+        verify(clientMock.downloadFile(anyString(), anyString())).called();
+    });
+});
+
+function getArchitecture(platform: string): string {
+    if (platform !== 'linux') {
+        return 'x64';
+    }
+
+    const archMap: Record<string, string> = {
+        'arm64': 'arm64',
+        'arm': 'armv6'
+    };
+
+    return archMap[process.arch] || 'x64';
+}