AI assistant e2e test

packages/web/src/buttons/ToolStripCommandSplitButton.svelte

@@ -17,4 +17,5 @@
   {menu}
   {hideDisabled}
   {buttonLabel}
+  {...$$restProps}
 />

packages/web/src/buttons/ToolStripSplitButton.svelte

@@ -21,7 +21,7 @@
 
 <div class="button" class:disabled {title}>
   <div class="inner" class:disabled>
-    <div class="main" class:disabled on:click={handleClick}>
+    <div class="main" class:disabled on:click={handleClick} data-testid={$$props['data-testid']}>
       <span class="icon" class:disabled><FontIcon {icon} /></span>
       <slot />
     </div>

packages/web/src/buttons/ToolStripSplitDropDownButton.svelte

@@ -19,6 +19,6 @@
   }
 </script>
 
-<ToolStripSplitButton {title} {icon} {disabled} on:splitclick={handleClick} on:click>
+<ToolStripSplitButton {title} {icon} {disabled} on:splitclick={handleClick} on:click {...$$restProps}>
   <slot />
 </ToolStripSplitButton>

packages/web/src/modals/ConfirmModal.svelte

@@ -21,8 +21,14 @@
           closeCurrentModal();
           onConfirm();
         }}
+        data-testid="ConfirmModal_okButton"
+      />
+      <FormStyledButton
+        type="button"
+        value="Close"
+        on:click={closeCurrentModal}
+        data-testid="ConfirmModal_closeButton"
       />
-      <FormStyledButton type="button" value="Close" on:click={closeCurrentModal} />
     </svelte:fragment>
   </ModalBase>
 </FormProvider>

packages/web/src/tabs/QueryTab.svelte

@@ -538,12 +538,16 @@
             editor.setValue(text);
           }
         }}
+        {tabid}
       />
     </svelte:fragment>
   </HorizontalSplitter>
   <svelte:fragment slot="toolstrip">
-    <ToolStripCommandSplitButton commands={['query.execute', 'query.executeCurrent']} />
-    <ToolStripCommandButton command="query.kill" />
+    <ToolStripCommandSplitButton
+      commands={['query.execute', 'query.executeCurrent']}
+      data-testid="QueryTab_executeButton"
+    />
+    <ToolStripCommandButton command="query.kill" data-testid="QueryTab_killButton" />
     <ToolStripSaveButton idPrefix="query" />
     <ToolStripCommandButton command="query.formatCode" />
     {#if resultCount == 1}
@@ -562,7 +566,13 @@
       icon="icon at"
       title="Query parameter style"
     />
-    <ToolStripCommandButton command="query.switchAiAssistant" hideDisabled />
+    <ToolStripCommandButton
+      command="query.switchAiAssistant"
+      hideDisabled
+      data-testid="QueryTab_switchAiAssistantButton"
+    >
+      AI Assistant
+    </ToolStripCommandButton>
   </svelte:fragment>
 </ToolStripContainer>