Replace register with an extension activate

test/global.test.ts

@@ -3,10 +3,9 @@
  *  Licensed under the MIT License. See LICENSE.md in the project root for license information.
  *--------------------------------------------------------------------------------------------*/
 
-import { registerAzureUtilsExtensionVariables } from '@microsoft/vscode-azext-azureutils';
 import { TestOutputChannel, TestUserInput } from '@microsoft/vscode-azext-dev';
 import * as vscode from 'vscode';
-import { ext, registerOnActionStartHandler, registerUIExtensionVariables } from '../extension.bundle';
+import { ext, registerOnActionStartHandler } from '../extension.bundle';
 
 const longRunningLocalTestsEnabled: boolean = !/^(false|0)?$/i.test(process.env.AzCode_EnableLongRunningTestsLocal || '');
 const longRunningRemoteTestsEnabled: boolean = !/^(false|0)?$/i.test(process.env.AzCode_UseAzureFederatedCredentials || '');
@@ -16,12 +15,14 @@ export const longRunningTestsEnabled: boolean = longRunningLocalTestsEnabled ||
 // Runs before all tests
 suiteSetup(async function (this: Mocha.Context): Promise<void> {
     this.timeout(120 * 1000);
-    await vscode.commands.executeCommand('appService.Refresh'); // activate the extension before tests begin
 
-    ext.outputChannel = new TestOutputChannel();
-    registerUIExtensionVariables(ext);
-    registerAzureUtilsExtensionVariables(ext);
+    const appServiceExtension = vscode.extensions.getExtension('ms-azuretools.vscode-azureappservice');
+    if (!appServiceExtension) {
+        throw new Error('Could not find the Azure App Service extension.');
+    }
+    await appServiceExtension.activate();
 
+    ext.outputChannel = new TestOutputChannel();
     registerOnActionStartHandler(context => {
         // Use `TestUserInput` by default so we get an error if an unexpected call to `context.ui` occurs, rather than timing out
         context.ui = new TestUserInput(vscode);