Add getStorageUri to API (to fix tests)

Author: andyleejordan

src/features/ExternalApi.ts

@@ -19,6 +19,7 @@ export interface IPowerShellExtensionClient {
     unregisterExternalExtension(uuid: string): boolean;
     getPowerShellVersionDetails(uuid: string): Promise<IExternalPowerShellDetails>;
     waitUntilStarted(uuid: string): Promise<void>;
+    getStorageUri(): vscode.Uri;
 }
 
 /*
@@ -166,6 +167,10 @@ export class ExternalApiFeature extends LanguageClientConsumer implements IPower
         return this.sessionManager.waitUntilStarted();
     }
 
+    public getStorageUri(): vscode.Uri {
+        return this.extensionContext.storageUri;
+    }
+
     public dispose() {
         // Nothing to dispose.
     }

src/main.ts

@@ -180,6 +180,7 @@ export async function activate(context: vscode.ExtensionContext): Promise<IPower
         unregisterExternalExtension: uuid => externalApi.unregisterExternalExtension(uuid),
         getPowerShellVersionDetails: uuid => externalApi.getPowerShellVersionDetails(uuid),
         waitUntilStarted: uuid => externalApi.waitUntilStarted(uuid),
+        getStorageUri: () => externalApi.getStorageUri(),
     };
 }
 

test/core/paths.test.ts

@@ -5,10 +5,15 @@ import * as assert from "assert";
 import * as fs from "fs";
 import * as path from "path";
 import * as vscode from "vscode";
+import { IPowerShellExtensionClient } from "../../src/features/ExternalApi";
 import utils = require("../utils");
 
 describe("Path assumptions", function () {
-    before(utils.ensureEditorServicesIsConnected);
+    let storageUri: vscode.Uri;
+    before(async () => {
+        const extension: IPowerShellExtensionClient = await utils.ensureEditorServicesIsConnected();
+        storageUri = extension.getStorageUri();
+    });
 
     // TODO: This is skipped because it interferes with other tests. Either
     // need to find a way to close the opened folder via a Code API, or find
@@ -22,6 +27,6 @@ describe("Path assumptions", function () {
     });
 
     it("Creates the log folder at the correct path", function () {
-        assert(fs.existsSync(path.resolve(utils.rootPath, "logs")));
+        assert(fs.existsSync(vscode.Uri.joinPath(storageUri, "logs").fsPath));
     });
 });

test/features/ExternalApi.test.ts

@@ -7,51 +7,50 @@ import { IExternalPowerShellDetails, IPowerShellExtensionClient } from "../../sr
 
 describe("ExternalApi feature", function () {
     describe("External extension registration", function () {
-        let powerShellExtensionClient: IPowerShellExtensionClient;
+        let extension: IPowerShellExtensionClient;
         before(async function () {
-            const powershellExtension = await utils.ensureExtensionIsActivated();
-            powerShellExtensionClient = powershellExtension!.exports as IPowerShellExtensionClient;
+            extension = await utils.ensureExtensionIsActivated();
         });
 
         it("Registers and unregisters an extension", function () {
-            const sessionId: string = powerShellExtensionClient.registerExternalExtension(utils.extensionId);
+            const sessionId: string = extension.registerExternalExtension(utils.extensionId);
             assert.notStrictEqual(sessionId, "");
             assert.notStrictEqual(sessionId, null);
             assert.strictEqual(
-                powerShellExtensionClient.unregisterExternalExtension(sessionId),
+                extension.unregisterExternalExtension(sessionId),
                 true);
         });
 
         it("Registers and unregisters an extension with a version", function () {
-            const sessionId: string = powerShellExtensionClient.registerExternalExtension(utils.extensionId, "v2");
+            const sessionId: string = extension.registerExternalExtension(utils.extensionId, "v2");
             assert.notStrictEqual(sessionId, "");
             assert.notStrictEqual(sessionId, null);
             assert.strictEqual(
-                powerShellExtensionClient.unregisterExternalExtension(sessionId),
+                extension.unregisterExternalExtension(sessionId),
                 true);
         });
 
         it("Rejects if not registered", async function () {
             assert.rejects(
-                async () => await powerShellExtensionClient.getPowerShellVersionDetails(""))
+                async () => await extension.getPowerShellVersionDetails(""))
         });
 
         it("Throws if attempting to register an extension more than once", async function () {
-            const sessionId: string = powerShellExtensionClient.registerExternalExtension(utils.extensionId);
+            const sessionId: string = extension.registerExternalExtension(utils.extensionId);
             try {
                 assert.throws(
-                    () => powerShellExtensionClient.registerExternalExtension(utils.extensionId),
+                    () => extension.registerExternalExtension(utils.extensionId),
                     {
                         message: `The extension '${utils.extensionId}' is already registered.`
                     });
             } finally {
-                powerShellExtensionClient.unregisterExternalExtension(sessionId);
+                extension.unregisterExternalExtension(sessionId);
             }
         });
 
         it("Throws when unregistering an extension that isn't registered", async function () {
             assert.throws(
-                () => powerShellExtensionClient.unregisterExternalExtension("not-real"),
+                () => extension.unregisterExternalExtension("not-real"),
                 {
                     message: `No extension registered with session UUID: not-real`
                 });
@@ -60,18 +59,17 @@ describe("ExternalApi feature", function () {
 
     describe("PowerShell version details", () => {
         let sessionId: string;
-        let powerShellExtensionClient: IPowerShellExtensionClient;
+        let extension: IPowerShellExtensionClient;
 
         before(async function () {
-            const powershellExtension = await utils.ensureExtensionIsActivated();
-            powerShellExtensionClient = powershellExtension!.exports as IPowerShellExtensionClient;
-            sessionId = powerShellExtensionClient.registerExternalExtension(utils.extensionId);
+            extension = await utils.ensureExtensionIsActivated();
+            sessionId = extension.registerExternalExtension(utils.extensionId);
         });
 
-        after(function () { powerShellExtensionClient.unregisterExternalExtension(sessionId); });
+        after(function () { extension.unregisterExternalExtension(sessionId); });
 
         it("Gets non-empty version details from the PowerShell Editor Services", async function () {
-            const versionDetails: IExternalPowerShellDetails = await powerShellExtensionClient.getPowerShellVersionDetails(sessionId);
+            const versionDetails: IExternalPowerShellDetails = await extension.getPowerShellVersionDetails(sessionId);
 
             assert.notStrictEqual(versionDetails.architecture, "");
             assert.notStrictEqual(versionDetails.architecture, null);

test/utils.ts

@@ -14,16 +14,16 @@ export const rootPath = path.resolve(__dirname, "../../")
 const packageJSON: any = require(path.resolve(rootPath, "package.json"));
 export const extensionId = `${packageJSON.publisher}.${packageJSON.name}`;
 
-export async function ensureExtensionIsActivated(): Promise<vscode.Extension<any>> {
+export async function ensureExtensionIsActivated(): Promise<IPowerShellExtensionClient> {
     const extension = vscode.extensions.getExtension(extensionId);
     if (!extension.isActive) { await extension.activate(); }
-    return extension;
+    return extension!.exports as IPowerShellExtensionClient;
 }
 
-export async function ensureEditorServicesIsConnected(): Promise<void> {
-    const powershellExtension = await ensureExtensionIsActivated();
-    const client = powershellExtension!.exports as IPowerShellExtensionClient;
-    const sessionId = client.registerExternalExtension(extensionId);
-    await client.waitUntilStarted(sessionId);
-    client.unregisterExternalExtension(sessionId);
+export async function ensureEditorServicesIsConnected(): Promise<IPowerShellExtensionClient> {
+    const extension = await ensureExtensionIsActivated();
+    const sessionId = extension.registerExternalExtension(extensionId);
+    await extension.waitUntilStarted(sessionId);
+    extension.unregisterExternalExtension(sessionId);
+    return extension;
 }