diff --git a/extensions/ql-vscode/src/vscode-tests/no-workspace/index.ts b/extensions/ql-vscode/src/vscode-tests/no-workspace/index.ts
index fcc27eeb6..0744e6e47 100644
--- a/extensions/ql-vscode/src/vscode-tests/no-workspace/index.ts
+++ b/extensions/ql-vscode/src/vscode-tests/no-workspace/index.ts
@@ -4,6 +4,7 @@ import * as sinonChai from 'sinon-chai';
 import * as chai from 'chai';
 import * as chaiAsPromised from 'chai-as-promised';
 import 'chai/register-should';
+import { ExtensionContext } from 'vscode';
 
 import { runTestsInDirectory } from '../index-template';
 
@@ -13,3 +14,19 @@ chai.use(sinonChai);
 export function run(): Promise<void> {
   return runTestsInDirectory(__dirname);
 }
+
+export function createMockExtensionContext(): ExtensionContext {
+  return {
+    globalState: {
+      _state: {
+        'telemetry-request-viewed': true
+      } as Record<string, any>,
+      get(key: string) {
+        return this._state[key];
+      },
+      update(key: string, val: any) {
+        this._state[key] = val;
+      }
+    }
+  } as any;
+}
diff --git a/extensions/ql-vscode/src/vscode-tests/no-workspace/telemetry.test.ts b/extensions/ql-vscode/src/vscode-tests/no-workspace/telemetry.test.ts
index 8b08f6be2..3d7ee0321 100644
--- a/extensions/ql-vscode/src/vscode-tests/no-workspace/telemetry.test.ts
+++ b/extensions/ql-vscode/src/vscode-tests/no-workspace/telemetry.test.ts
@@ -6,6 +6,7 @@ import { TelemetryListener, telemetryListener as globalTelemetryListener } from
 import { UserCancellationException } from '../../commandRunner';
 import { fail } from 'assert';
 import { ENABLE_TELEMETRY } from '../../config';
+import { createMockExtensionContext } from './index';
 
 const sandbox = sinon.createSandbox();
 
@@ -339,22 +340,6 @@ describe('telemetry reporting', function() {
     expect(window.showInformationMessage).to.have.been.calledOnce;
   });
 
-  function createMockExtensionContext(): ExtensionContext {
-    return {
-      globalState: {
-        _state: {
-          'telemetry-request-viewed': true
-        } as Record<string, any>,
-        get(key: string) {
-          return this._state[key];
-        },
-        update(key: string, val: any) {
-          this._state[key] = val;
-        }
-      }
-    } as any;
-  }
-
   async function enableTelemetry(section: string, value: boolean | undefined) {
     await workspace.getConfiguration(section).update(
       'enableTelemetry', value, ConfigurationTarget.Global