Mock env.itTelemetryEnable

Author: robertbrignull

extensions/ql-vscode/test/vscode-tests/no-workspace/telemetry.test.ts

@@ -4,6 +4,7 @@ import {
   workspace,
   ConfigurationTarget,
   window,
+  env,
 } from "vscode";
 import {
   ExtensionTelemetryListener,
@@ -37,6 +38,11 @@ describe("telemetry reporting", () => {
     typeof TelemetryReporter.prototype.dispose
   >;
 
+  let isTelemetryEnabledSpy: jest.SpyInstance<
+    typeof env.isTelemetryEnabled,
+    []
+  >;
+
   let showInformationMessageSpy: jest.SpiedFunction<
     typeof window.showInformationMessage
   >;
@@ -78,6 +84,9 @@ describe("telemetry reporting", () => {
         .get<boolean>("codeQL.canary")).toString();
 
       // each test will default to telemetry being enabled
+      isTelemetryEnabledSpy = jest
+        .spyOn(env, "isTelemetryEnabled", "get")
+        .mockReturnValue(true);
       await enableTelemetry("telemetry", true);
       await enableTelemetry("codeQL.telemetry", true);
 
@@ -116,6 +125,7 @@ describe("telemetry reporting", () => {
   });
 
   it("should initialize telemetry when global option disabled", async () => {
+    isTelemetryEnabledSpy.mockReturnValue(false);
     await enableTelemetry("telemetry", false);
     await telemetryListener.initialize();
     expect(telemetryListener._reporter).toBeDefined();
@@ -133,6 +143,7 @@ describe("telemetry reporting", () => {
 
   it("should not initialize telemetry when both options disabled", async () => {
     await enableTelemetry("codeQL.telemetry", false);
+    isTelemetryEnabledSpy.mockReturnValue(false);
     await enableTelemetry("telemetry", false);
     await telemetryListener.initialize();
     expect(telemetryListener._reporter).toBeUndefined();
@@ -179,6 +190,7 @@ describe("telemetry reporting", () => {
     const reporter: any = telemetryListener._reporter;
     expect(reporter.userOptIn).toBe(true); // enabled
 
+    isTelemetryEnabledSpy.mockReturnValue(false);
     await enableTelemetry("telemetry", false);
     expect(reporter.userOptIn).toBe(false); // disabled
   });
@@ -344,6 +356,8 @@ describe("telemetry reporting", () => {
       resolveArg(3 /* "yes" item */),
     );
     await ctx.globalState.update("telemetry-request-viewed", false);
+    expect(env.isTelemetryEnabled).toBe(true);
+
     await enableTelemetry("codeQL.telemetry", false);
 
     await telemetryListener.initialize();
@@ -414,6 +428,7 @@ describe("telemetry reporting", () => {
     await ctx.globalState.update("telemetry-request-viewed", false);
 
     await telemetryListener.initialize();
+    isTelemetryEnabledSpy.mockReturnValue(false);
 
     // popup should not be shown even though we have initialized telemetry
     expect(showInformationMessageSpy).not.toBeCalled();