From 3ce873467b32614a33b8d172dfb90d66cedc7e5a Mon Sep 17 00:00:00 2001
From: Christopher Christou <chrstou@amazon.com>
Date: Tue, 19 May 2020 08:53:09 -0700
Subject: [PATCH] Remove extension activation from unit tests

---
 src/test/globalSetup.test.ts                       | 5 -----
 src/test/shared/codelens/localLambdaRunner.test.ts | 6 ++++++
 2 files changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/test/globalSetup.test.ts b/src/test/globalSetup.test.ts
index 0dc059abd6f..a231bca75a4 100644
--- a/src/test/globalSetup.test.ts
+++ b/src/test/globalSetup.test.ts
@@ -6,7 +6,6 @@
 /**
  * Before/After hooks for all "unit" tests
  */
-import * as vscode from 'vscode'
 import * as assert from 'assert'
 import { appendFileSync, mkdirpSync } from 'fs-extra'
 import { join } from 'path'
@@ -20,7 +19,6 @@ import { TelemetryPublisher } from '../shared/telemetry/telemetryPublisher'
 import { FakeExtensionContext } from './fakeExtensionContext'
 import { TestLogger } from './testLogger'
 import { FakeAwsContext } from './utilities/fakeAwsContext'
-import { VSCODE_EXTENSION_ID } from '../shared/extensions'
 
 const testReportDir = join(__dirname, '../../../.test-reports')
 const testLogOutput = join(testReportDir, 'testLog.log')
@@ -45,9 +43,6 @@ before(async () => {
     }
     const service = new DefaultTelemetryService(mockContext, mockAws, mockPublisher)
     ext.telemetry = service
-
-    // Warm up vscode before tests run
-    await vscode.extensions.getExtension(VSCODE_EXTENSION_ID.awstoolkit)!.activate()
 })
 
 beforeEach(async function() {
diff --git a/src/test/shared/codelens/localLambdaRunner.test.ts b/src/test/shared/codelens/localLambdaRunner.test.ts
index a8c3e9c3d45..431779f99ee 100644
--- a/src/test/shared/codelens/localLambdaRunner.test.ts
+++ b/src/test/shared/codelens/localLambdaRunner.test.ts
@@ -12,11 +12,17 @@ import { DebugConfiguration } from '../../../lambda/local/debugConfiguration'
 import * as localLambdaRunner from '../../../shared/codelens/localLambdaRunner'
 import * as fsUtils from '../../../shared/filesystemUtilities'
 import { ChildProcessResult } from '../../../shared/utilities/childProcess'
+import { ExtensionDisposableFiles } from '../../../shared/utilities/disposableFiles'
+import { FakeExtensionContext } from '../../fakeExtensionContext'
 import { FakeChannelLogger } from '../fakeChannelLogger'
 import { assertRejects } from '../utilities/assertUtils'
 
 describe('localLambdaRunner', async () => {
     let tempDir: string
+    before(async () => {
+        await ExtensionDisposableFiles.initialize(new FakeExtensionContext())
+    })
+
     beforeEach(async () => {
         tempDir = await fsUtils.makeTemporaryToolkitFolder()
     })