From 0e68a74e70a21d4b12c7460d7917ff31e5cb9300 Mon Sep 17 00:00:00 2001
From: ZLY <zhongliangyuan@gdngsk.com>
Date: Wed, 21 Jan 2026 10:44:56 +0800
Subject: [PATCH] =?UTF-8?q?feat(testInstance):=20=E6=B7=BB=E5=8A=A0?=
 =?UTF-8?q?=E6=B5=8B=E8=AF=95=E5=AE=9E=E4=BE=8B=E9=93=BE=E6=8E=A5=E5=8A=9F?=
 =?UTF-8?q?=E8=83=BD?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../componentTest/testInstance.tsx                | 15 ++++++++++-----
 1 file changed, 10 insertions(+), 5 deletions(-)

diff --git a/src/pages/componentDevelopment/componentTest/testInstance.tsx b/src/pages/componentDevelopment/componentTest/testInstance.tsx
index 0cffb53..477dba7 100644
--- a/src/pages/componentDevelopment/componentTest/testInstance.tsx
+++ b/src/pages/componentDevelopment/componentTest/testInstance.tsx
@@ -9,10 +9,11 @@ import {
   deleteTestCase,
   exportTemplate,
   exportTestCases,
-  importTestCases
+  importTestCases, startTestCase
 } from '@/api/componentTestCase';
 import TestCaseModal from './testCaseModal';
 import useWebSocket from '@/hooks/useWebSocket';
+import { getToken } from '@/utils/auth';
 
 const CollapseItem = Collapse.Item;
 
@@ -99,24 +100,27 @@ const TestInstance = ({ instance, parentId, onBack }: { instance: any; parentId:
   };
 
   // 链接实例
-  const handleLinkInstance = () => {
+  const handleLinkInstance = async () => {
     if (isSocketConnected) {
       disconnect();
       Message.info('已断开连接');
     }
     else {
+      // WebSocket连接前置
+      const res: any = await startTestCase(instance.id);
+      console.log('res:', res);
+
       // 构建WebSocket URL，根据你的实际后端配置调整
       const protocol = window.location.protocol === 'https:' ? 'wss:' : 'ws:';
       const host = window.location.host;
       let wsUrl = '';
       if (window.location.host.includes('localhost')) {
-        wsUrl = `${process.env.NEXT_PUBLIC_DEV_SOCKET_HOST}/ws/v1/bpms-workbench/test-case/${instance.id}/${cryptoRandom()}`;
+        wsUrl = `${process.env.NEXT_PUBLIC_DEV_SOCKET_HOST}/ws/v1/bpms-workbench/test-case/${instance.id}/${cryptoRandom()}?Authorization=Bearer ${getToken()}`;
       }
       else {
-        wsUrl = `${protocol}//${host}/ws/v1/bpms-workbench/test-case/${instance.id}/${cryptoRandom()}`;
+        wsUrl = `${protocol}//${host}/ws/v1/bpms-workbench/test-case/${instance.id}/${cryptoRandom()}?Authorization=Bearer ${getToken()}`;
       }
 
-
       connect(wsUrl);
       addLog('正在连接测试实例...');
     }
@@ -311,6 +315,7 @@ const TestInstance = ({ instance, parentId, onBack }: { instance: any; parentId:
         <div className={styles['tab-left']}>
           <Button
             type="primary"
+            status={isSocketConnected ?'danger' : 'default'}
             icon={<IconLink />}
             onClick={handleLinkInstance}
             style={{ minWidth: 250 }}