Add tests for the new webview message handler

Author: wolmir

extension/src/test/suite/experiments/index.test.ts

@@ -344,6 +344,30 @@ suite('Experiments Test Suite', () => {
       return { experiments, experimentsModel, mockExecuteCommand }
     }
 
+    it('should be able to handle a message to hide a table column', async () => {
+      const { experiments, columnsModel } = buildExperiments(disposable)
+
+      const mockToggleStatus = stub(columnsModel, 'toggleStatus')
+      const mockSendTelemetryEvent = stub(Telemetry, 'sendTelemetryEvent')
+      const webview = await experiments.showWebview()
+      const mockMessageReceived = getMessageReceivedEmitter(webview)
+      const mockColumnId = 'mock-column-id'
+
+      mockMessageReceived.fire({
+        payload: mockColumnId,
+        type: MessageFromWebviewType.HIDE_EXPERIMENTS_TABLE_COLUMN
+      })
+
+      expect(mockToggleStatus).to.be.calledOnce
+      expect(mockToggleStatus).to.be.calledWithExactly(mockColumnId)
+
+      expect(mockSendTelemetryEvent).to.be.calledWithExactly(
+        EventName.VIEWS_EXPERIMENTS_TABLE_HIDE_COLUMN,
+        { path: mockColumnId },
+        undefined
+      )
+    })
+
     it('should be able to handle a message to apply an experiment to workspace', async () => {
       const { experiments, mockExecuteCommand } =
         setupExperimentsAndMockCommands()

extension/src/test/suite/experiments/util.ts

@@ -19,6 +19,7 @@ import { CheckpointsModel } from '../../../experiments/checkpoints/model'
 import { FileSystemData } from '../../../fileSystem/data'
 import * as Watcher from '../../../fileSystem/watcher'
 import { ExperimentsModel } from '../../../experiments/model'
+import { ColumnsModel } from '../../../experiments/columns/model'
 
 const hasCheckpoints = (data: ExperimentsOutput) => {
   const [experimentsWithBaseline] = Object.values(omit(data, 'workspace'))
@@ -73,6 +74,8 @@ export const buildExperiments = (
     cliExecutor,
     cliReader,
     cliRunner,
+    // eslint-disable-next-line @typescript-eslint/no-explicit-any
+    columnsModel: (experiments as any).columns as ColumnsModel,
     experiments,
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     experimentsModel: (experiments as any).experiments as ExperimentsModel,