Update connection in tests

Author: skovhus

server/src/__tests__/analyzer.test.ts

@@ -1,4 +1,5 @@
 import FIXTURES, { FIXTURE_FOLDER } from '../../../testing/fixtures'
+import { getMockConnection } from '../../../testing/mocks'
 import Analyzer from '../analyser'
 import { initializeParser } from '../parser'
 
@@ -113,11 +114,7 @@ describe('fromRoot', () => {
 
     jest.spyOn(Date, 'now').mockImplementation(() => 0)
 
-    const connection: any = {
-      console: {
-        log: jest.fn(),
-      },
-    }
+    const connection = getMockConnection()
 
     const newAnalyzer = await Analyzer.fromRoot({
       connection,
@@ -127,7 +124,7 @@ describe('fromRoot', () => {
 
     expect(newAnalyzer).toBeDefined()
 
-    const FIXTURE_FILES_MATCHING_GLOB = 9
+    const FIXTURE_FILES_MATCHING_GLOB = 10
 
     // Intro, stats on glob, one file skipped due to shebang, and outro
     const LOG_LINES = FIXTURE_FILES_MATCHING_GLOB + 4

server/src/__tests__/server.test.ts

@@ -1,71 +1,14 @@
 import * as lsp from 'vscode-languageserver'
 
 import { FIXTURE_FOLDER, FIXTURE_URI } from '../../../testing/fixtures'
+import { getMockConnection } from '../../../testing/mocks'
 import LspServer from '../server'
 import { CompletionItemDataType } from '../types'
 
 async function initializeServer() {
   const diagnostics: Array<lsp.PublishDiagnosticsParams | undefined> = undefined
-  const console: any = {
-    error: jest.fn(),
-    warn: jest.fn(),
-    info: jest.fn(),
-    log: jest.fn(),
-  }
 
-  const connection: jest.Mocked<lsp.Connection> = {
-    client: {} as any,
-    console,
-    dispose: jest.fn(),
-    listen: jest.fn(),
-    onCodeAction: jest.fn(),
-    onCodeLens: jest.fn(),
-    onCodeLensResolve: jest.fn(),
-    onColorPresentation: jest.fn(),
-    onCompletion: jest.fn(),
-    onCompletionResolve: jest.fn(),
-    onDeclaration: jest.fn(),
-    onDefinition: jest.fn(),
-    onDidChangeConfiguration: jest.fn(),
-    onDidChangeTextDocument: jest.fn(),
-    onDidChangeWatchedFiles: jest.fn(),
-    onDidCloseTextDocument: jest.fn(),
-    onDidOpenTextDocument: jest.fn(),
-    onDidSaveTextDocument: jest.fn(),
-    onDocumentColor: jest.fn(),
-    onDocumentFormatting: jest.fn(),
-    onDocumentHighlight: jest.fn(),
-    onDocumentLinkResolve: jest.fn(),
-    onDocumentLinks: jest.fn(),
-    onDocumentOnTypeFormatting: jest.fn(),
-    onDocumentRangeFormatting: jest.fn(),
-    onDocumentSymbol: jest.fn(),
-    onExecuteCommand: jest.fn(),
-    onExit: jest.fn(),
-    onFoldingRanges: jest.fn(),
-    onHover: jest.fn(),
-    onImplementation: jest.fn(),
-    onInitialize: jest.fn(),
-    onInitialized: jest.fn(),
-    onNotification: jest.fn(),
-    onPrepareRename: jest.fn(),
-    onReferences: jest.fn(),
-    onRenameRequest: jest.fn(),
-    onRequest: jest.fn(),
-    onShutdown: jest.fn(),
-    onSignatureHelp: jest.fn(),
-    onTypeDefinition: jest.fn(),
-    onWillSaveTextDocument: jest.fn(),
-    onWillSaveTextDocumentWaitUntil: jest.fn(),
-    onWorkspaceSymbol: jest.fn(),
-    sendDiagnostics: jest.fn(),
-    sendNotification: jest.fn(),
-    sendRequest: jest.fn(),
-    telemetry: {} as any,
-    tracer: {} as any,
-    window: {} as any,
-    workspace: {} as any,
-  }
+  const connection = getMockConnection()
 
   const server = await LspServer.initialize(connection, {
     rootPath: FIXTURE_FOLDER,

testing/mocks.ts

@@ -0,0 +1,64 @@
+import * as lsp from 'vscode-languageserver'
+
+export function getMockConnection(): jest.Mocked<lsp.Connection> {
+  const console: any = {
+    error: jest.fn(),
+    warn: jest.fn(),
+    info: jest.fn(),
+    log: jest.fn(),
+  }
+
+  return {
+    client: {} as any,
+    console,
+    dispose: jest.fn(),
+    listen: jest.fn(),
+    onCodeAction: jest.fn(),
+    onCodeLens: jest.fn(),
+    onCodeLensResolve: jest.fn(),
+    onColorPresentation: jest.fn(),
+    onCompletion: jest.fn(),
+    onCompletionResolve: jest.fn(),
+    onDeclaration: jest.fn(),
+    onDefinition: jest.fn(),
+    onDidChangeConfiguration: jest.fn(),
+    onDidChangeTextDocument: jest.fn(),
+    onDidChangeWatchedFiles: jest.fn(),
+    onDidCloseTextDocument: jest.fn(),
+    onDidOpenTextDocument: jest.fn(),
+    onDidSaveTextDocument: jest.fn(),
+    onDocumentColor: jest.fn(),
+    onDocumentFormatting: jest.fn(),
+    onDocumentHighlight: jest.fn(),
+    onDocumentLinkResolve: jest.fn(),
+    onDocumentLinks: jest.fn(),
+    onDocumentOnTypeFormatting: jest.fn(),
+    onDocumentRangeFormatting: jest.fn(),
+    onDocumentSymbol: jest.fn(),
+    onExecuteCommand: jest.fn(),
+    onExit: jest.fn(),
+    onFoldingRanges: jest.fn(),
+    onHover: jest.fn(),
+    onImplementation: jest.fn(),
+    onInitialize: jest.fn(),
+    onInitialized: jest.fn(),
+    onNotification: jest.fn(),
+    onPrepareRename: jest.fn(),
+    onReferences: jest.fn(),
+    onRenameRequest: jest.fn(),
+    onRequest: jest.fn(),
+    onShutdown: jest.fn(),
+    onSignatureHelp: jest.fn(),
+    onTypeDefinition: jest.fn(),
+    onWillSaveTextDocument: jest.fn(),
+    onWillSaveTextDocumentWaitUntil: jest.fn(),
+    onWorkspaceSymbol: jest.fn(),
+    sendDiagnostics: jest.fn(),
+    sendNotification: jest.fn(),
+    sendRequest: jest.fn(),
+    telemetry: {} as any,
+    tracer: {} as any,
+    window: {} as any,
+    workspace: {} as any,
+  }
+}