Move restore flag behind feature flag. Whitelist restorable tabs in for Fabric. Restore e2e tests.

src/Explorer/Explorer.tsx

@@ -1166,8 +1166,11 @@ export default class Explorer {
     }
 
     await this.refreshSampleData();
+
+    if (userContext.features.restoreTabs) {
       this.restoreOpenTabs();
     }
+  }
 
   public async configureCopilot(): Promise<void> {
     if (

src/Explorer/OpenActions/OpenActions.tsx

@@ -1,4 +1,5 @@
 // TODO convert this file to an action registry in order to have actions and their handlers be more tightly coupled.
+import { configContext, Platform } from "ConfigContext";
 import { useDatabases } from "Explorer/useDatabases";
 import React from "react";
 import { ActionContracts } from "../../Contracts/ExplorerContracts";
@@ -56,6 +57,19 @@ function openCollectionTab(
         continue;
       }
 
+      if (
+        configContext.platform === Platform.Fabric &&
+        !(
+          // whitelist the tab kinds that are allowed to be opened in Fabric
+          (
+            action.tabKind === ActionContracts.TabKind.SQLDocuments ||
+            action.tabKind === ActionContracts.TabKind.SQLQuery
+          )
+        )
+      ) {
+        continue;
+      }
+
       //expand database first if not expanded to load the collections
       if (!database.isDatabaseExpanded?.()) {
         database.expandDatabase?.();

src/Platform/Hosted/extractFeatures.ts

@@ -38,6 +38,7 @@ export type Features = {
   readonly copilotChatFixedMonacoEditorHeight: boolean;
   readonly enablePriorityBasedExecution: boolean;
   readonly disableConnectionStringLogin: boolean;
+  readonly restoreTabs: boolean;
 
   // can be set via both flight and feature flag
   autoscaleDefault: boolean;
@@ -108,6 +109,7 @@ export function extractFeatures(given = new URLSearchParams(window.location.sear
     copilotChatFixedMonacoEditorHeight: "true" === get("copilotchatfixedmonacoeditorheight"),
     enablePriorityBasedExecution: "true" === get("enableprioritybasedexecution"),
     disableConnectionStringLogin: "true" === get("disableconnectionstringlogin"),
+    restoreTabs: "true" === get("restoretabs"),
   };
 }
 

test/sql/query.spec.ts

@@ -1,6 +1,5 @@
 import { expect, test } from "@playwright/test";
 
-import { StorageKey } from "../../src/Shared/StorageUtility";
 import { DataExplorer, Editor, QueryTab, TestAccount } from "../fx";
 import { TestContainerContext, TestItem, createTestSQLContainer } from "../testData";
 
@@ -17,9 +16,6 @@ test.beforeEach("Open new query tab", async ({ page }) => {
   // Open a query tab
   explorer = await DataExplorer.open(page, TestAccount.SQL);
 
-  // Clear previous settings
-  page.evaluate(() => localStorage.removeItem(StorageKey[StorageKey.AppState]));
-
   // Container nodes should be visible. The explorer auto-expands database nodes when they are first loaded.
   const containerNode = await explorer.waitForContainerNode(context.database.id, context.container.id);
   await containerNode.openContextMenu();