Refactor telemetryAIEvent parameters and add tests for telemetry functionality

Author: cx-sarah-chen

src/main/wrapper/CxConstants.ts

@@ -116,6 +116,7 @@ export enum CxConstants {
     SUB_CMD_TELEMETRY_AI = "ai",
     AI_PROVIDER = "--ai-provider",
     TIMESTAMP = "--timestamp",
-    CLICK_TYPE = "--click-type",
+    TYPE = "--type",
+    SUB_TYPE = "--sub-type",
     PROBLEM_SEVERITY = "--problem-severity"
 }

src/main/wrapper/CxWrapper.ts

@@ -448,14 +448,15 @@ export class CxWrapper {
         return new ExecutionService().executeCommands(this.config.pathToExecutable, commands, CxConstants.MASK_TYPE);
     }
 
-    telemetryAIEvent(aiProvider: string, agent: string, timestamp: Date, clickType: string, engine: string, problemSeverity: string): Promise<CxCommandOutput> {
+    telemetryAIEvent(aiProvider: string, agent: string, timestamp: Date, eventType: string, subType: string, engine: string, problemSeverity: string): Promise<CxCommandOutput> {
         const commands: string[] = [
             CxConstants.TELEMETRY,
             CxConstants.SUB_CMD_TELEMETRY_AI,
             CxConstants.AI_PROVIDER, aiProvider,
             CxConstants.AGENT, agent,
-            CxConstants.TIMESTAMP, timestamp.toISOString(),
-            CxConstants.CLICK_TYPE, clickType,
+            CxConstants.TIMESTAMP, timestamp?.toISOString(),
+            CxConstants.TYPE, eventType,
+            CxConstants.SUB_TYPE, subType,
             CxConstants.ENGINE, engine,
             CxConstants.PROBLEM_SEVERITY, problemSeverity
         ];

src/tests/TelemetryTest.test.ts

@@ -0,0 +1,70 @@
+import { CxWrapper } from '../main/wrapper/CxWrapper';
+import { CxCommandOutput } from "../main/wrapper/CxCommandOutput";
+import { BaseTest } from "./BaseTest";
+
+describe("Telemetry cases", () => {
+    const cxScanConfig = new BaseTest();
+
+    it('TelemetryAIEvent Successful case with minimal parameters', async () => {
+        const wrapper = new CxWrapper(cxScanConfig);
+        const timestamp = new Date();
+        const cxCommandOutput: CxCommandOutput = await wrapper.telemetryAIEvent(
+            "Cursor", 
+            "Cursos", 
+            timestamp,
+            "click",  
+            "ast-results.viewPackageDetails",
+            "secrets",   
+            "high"    
+        );
+        console.log("Json object from telemetryAIEvent successful case: " + JSON.stringify(cxCommandOutput));
+        expect(cxCommandOutput.exitCode).toBe(0);
+    });
+
+    it('TelemetryAIEvent Successful case with past timestamp', async () => {
+        const wrapper = new CxWrapper(cxScanConfig);
+        const pastTimestamp = new Date(Date.now() - 3600000); // 1 hour ago
+        const cxCommandOutput: CxCommandOutput = await wrapper.telemetryAIEvent(
+            "openai",           
+            "vscode",           
+            pastTimestamp,      
+            "click",
+            "ast-results.viewPackageDetails",      
+            "oss",             
+            "high"              
+        );
+        console.log("Json object from telemetryAIEvent with past timestamp: " + JSON.stringify(cxCommandOutput));
+        expect(cxCommandOutput.exitCode).toBe(0);
+    });
+
+    it('TelemetryAIEvent Successful case with edge case parameters', async () => {
+        const wrapper = new CxWrapper(cxScanConfig);
+        const timestamp = new Date();
+        const cxCommandOutput: CxCommandOutput = await wrapper.telemetryAIEvent(
+            "",              
+            "",         
+            timestamp,
+            "",
+            "",             
+            "",          
+            ""       
+        );
+        console.log("Json object from telemetryAIEvent with empty parameters: " + JSON.stringify(cxCommandOutput));
+        expect(typeof cxCommandOutput.exitCode).toBe(0);
+    });
+
+    it('TelemetryAIEvent Successful case without timestamp', async () => {
+        const wrapper = new CxWrapper(cxScanConfig);
+        const cxCommandOutput: CxCommandOutput = await wrapper.telemetryAIEvent(
+            "Copilot",       
+            "VS Code",  
+            undefined,   
+            "click",  
+            "ast-results.viewPackageDetails",   
+            "oss",        
+            "medium" 
+        );
+        console.log("Json object from telemetryAIEvent without timestamp: " + JSON.stringify(cxCommandOutput));
+        expect(typeof cxCommandOutput.exitCode).toBe(0);
+    });
+});