wip: add more mock APIs

Author: kermanx

packages/mock/src/class/EventEmitter.ts

@@ -9,7 +9,7 @@ export class EventEmitter<T> extends Disposable implements vscode.EventEmitter<T
 
   _listeners = new Set<(e: T) => any>()
 
-  event: vscode.Event<T> = vi.fn((listener, thisArgs, disposables) => {
+  event = vi.fn<vscode.Event<T>>((listener, thisArgs, disposables) => {
     const cb = (e: T) => listener.call(thisArgs, e)
     this._listeners.add(cb)
     const disposable = new Disposable(() => this._listeners.delete(cb))

packages/mock/src/internal/WorkspaceConfiguration.ts

@@ -1,9 +1,10 @@
 import type vscode from 'vscode'
 import { vi } from 'vitest'
 import { Unimplemented } from '../utils/unimplemented'
+import type { MockVscode } from '../ns'
 
 export class WorkspaceConfiguration implements vscode.WorkspaceConfiguration {
-  constructor(public _data: any) {
+  constructor(public _context: MockVscode, public _data: any, public _baseSection?: string) {
   }
 
   get = vi.fn(<T>(section: string, defaultValue?: T): T | undefined => {
@@ -39,5 +40,10 @@ export class WorkspaceConfiguration implements vscode.WorkspaceConfiguration {
       obj = obj[key]
     }
     obj[keys[keys.length - 1]] = value
+    this._context.workspace._onDidChangeConfiguration.fire({
+      affectsConfiguration: vi.fn((check: string, _resource?: vscode.Uri) => {
+        return (this._baseSection ? `${this._baseSection}.` : `${section}`).startsWith(check)
+      }),
+    })
   })
 }

packages/mock/src/ns/window.ts

@@ -1,11 +1,12 @@
 import type vscode from 'vscode'
-
+import { vi } from 'vitest'
 import { EventEmitter } from '../class/EventEmitter'
 import { TextDocument } from '../internal/TextDocument'
 import { TextEditor } from '../internal/TextEditor'
 import type { WindowNS } from '../types'
 import { createState } from '../utils/state'
 import { ColorThemeKind } from '../enum/ColorThemeKind'
+import { Unimplemented } from '../utils/unimplemented'
 import type { MockVscode } from '.'
 
 export function createMockWindow(_context: MockVscode) {
@@ -60,13 +61,34 @@ export function createMockWindow(_context: MockVscode) {
       return editor
     }
 
+    showInformationMessage = vi.fn((..._args: any) => Unimplemented('showInformationMessage'))
+    showWarningMessage = vi.fn((..._args: any) => Unimplemented('showInformationMessage'))
+    showErrorMessage = vi.fn((..._args: any) => Unimplemented('showInformationMessage'))
+    showQuickPick = vi.fn((..._args: any) => Unimplemented('showQuickPick'))
+    showWorkspaceFolderPick = vi.fn((..._args: any) => Unimplemented('showWorkspaceFolderPick'))
+    showOpenDialog = vi.fn((..._args: any) => Unimplemented('showOpenDialog'))
+    showSaveDialog = vi.fn((..._args: any) => Unimplemented('showSaveDialog'))
+    showInputBox = vi.fn((..._args: any) => Unimplemented('showInputBox'))
+    createQuickPick = TODO
+    createInputBox = TODO
+    createStatusBarItem = TODO
+    createOutputChannel = TODO
+    createWebviewPanel = TODO
+    registerWebviewPanelSerializer = TODO
+    setStatusBarMessage = TODO
+    withScmProgress = TODO
+    withProgress = TODO
+    createTerminal = TODO
+    registerTreeDataProvider = TODO
+    createTreeView = TODO
+    registerUriHandler = TODO
+    registerWebviewViewProvider = TODO
+    registerCustomEditorProvider = TODO
+
     _windowState = createState(_init.state)
-    get windowState(): vscode.WindowState {
+    get state(): vscode.WindowState {
       return this._windowState.value
     }
-    set windowState(value: vscode.WindowState) {
-      this._windowState.update(value)
-    }
     _onDidChangeWindowState = this._windowState.eventEmitter
     onDidChangeWindowState = this._windowState.event
 

packages/mock/src/ns/workspace.ts

@@ -6,6 +6,7 @@ import type { WorkspaceNS } from '../types'
 import { Unimplemented } from '../utils/unimplemented'
 import { TextDocument } from '../internal/TextDocument'
 import { TextDocumentSaveReason } from '../enum/TextDocumentSaveReason'
+import { WorkspaceConfiguration } from '../internal/WorkspaceConfiguration'
 import type { MockVscode } from '.'
 
 export function createMockWorkspace(_context: MockVscode) {
@@ -151,6 +152,49 @@ export function createMockWorkspace(_context: MockVscode) {
 
     _onDidSaveTextDocument = new EventEmitter<TextDocument>()
     onDidSaveTextDocument = this._onDidSaveTextDocument.event
+
+    _onWillCreateFiles = new EventEmitter<vscode.FileWillCreateEvent>()
+    onWillCreateFiles = this._onWillCreateFiles.event
+
+    _onDidCreateFiles = new EventEmitter<vscode.FileCreateEvent>()
+    onDidCreateFiles = this._onDidCreateFiles.event
+
+    _onWillDeleteFiles = new EventEmitter<vscode.FileWillDeleteEvent>()
+    onWillDeleteFiles = this._onWillDeleteFiles.event
+
+    _onDidDeleteFiles = new EventEmitter<vscode.FileDeleteEvent>()
+    onDidDeleteFiles = this._onDidDeleteFiles.event
+
+    _onWillRenameFiles = new EventEmitter<vscode.FileWillRenameEvent>()
+    onWillRenameFiles = this._onWillRenameFiles.event
+
+    _onDidRenameFiles = new EventEmitter<vscode.FileRenameEvent>()
+    onDidRenameFiles = this._onDidRenameFiles.event
+
+    _workspaceConfiguration = _init.workspaceConfiguration
+    getConfiguration = vi.fn((section?: string, _scope?: vscode.ConfigurationScope | null): WorkspaceConfiguration => {
+      let obj = this._workspaceConfiguration
+      for (const key of section?.split('.') ?? []) {
+        if (!(key in obj))
+          return new WorkspaceConfiguration(_context, {}, section)
+        obj = obj[key]
+      }
+      return new WorkspaceConfiguration(_context, obj, section)
+    })
+
+    _onDidChangeConfiguration = new EventEmitter<vscode.ConfigurationChangeEvent>()
+    onDidChangeConfiguration = this._onDidChangeConfiguration.event
+
+    _isTrusted = _init.isTrusted
+    _trust = () => {
+      this._isTrusted = true
+      this._onDidGrantWorkspaceTrust.fire()
+    }
+    get isTrusted() {
+      return this._isTrusted
+    }
+    _onDidGrantWorkspaceTrust = new EventEmitter<void>()
+    onDidGrantWorkspaceTrust = this._onDidGrantWorkspaceTrust.event
   })()
 }
 
@@ -163,6 +207,8 @@ export const defaultWorkspaceInitConfig = {
     name: 'workspace',
     index: 0,
   }] as vscode.WorkspaceFolder[],
+  workspaceConfiguration: {} as any,
+  isTrusted: true,
 }
 
 export type MockWorkspace = ReturnType<typeof createMockWorkspace>

packages/mock/src/utils/state.ts

@@ -7,7 +7,7 @@ export function createState<T>(initial: T) {
     eventEmitter,
     update(newVal: T) {
       this.value = newVal
-      this.fire()
+      eventEmitter.fire(newVal)
     },
     event: eventEmitter.event,
   }