[test]: Update unit tests

Author: zzyangh

packages/sqle/src/page/DataSourceComparison/ComparisonEntry/__tests__/__snapshots__/useDataSourceSelectorTree.test.tsx.snap

@@ -24,6 +24,17 @@ exports[`useDataSourceSelectorTree should initialize data correctly 1`] = `
     />,
     "value": "PostgreSQL",
   },
+  {
+    "dbType": "TiDB",
+    "disabled": true,
+    "id": "TiDB",
+    "label": "TiDB",
+    "pId": "TREE_DATA_PARENT_NODE_ID",
+    "title": <DatabaseTypeLogo
+      dbType="TiDB"
+    />,
+    "value": "TiDB",
+  },
   {
     "dbType": "MySQL",
     "id": "1739531854064652288",
@@ -119,5 +130,24 @@ exports[`useDataSourceSelectorTree should initialize data correctly 1`] = `
     </Styled(Styled(span))>,
     "value": "1743145219550875648",
   },
+  {
+    "dbType": "TiDB",
+    "id": "1743145219550875649",
+    "label": "tidb-1",
+    "pId": "TiDB",
+    "title": <Styled(Styled(span))>
+      <DatabaseFilled
+        color="#4583ff"
+        height={18}
+        width={18}
+      />
+      <span
+        className="content"
+      >
+        tidb-1(10.186.62.17:4000)
+      </span>
+    </Styled(Styled(span))>,
+    "value": "1743145219550875649",
+  },
 ]
 `;

packages/sqle/src/page/DataSourceComparison/ComparisonEntry/__tests__/useDataSourceSelectorTree.test.tsx

@@ -46,12 +46,20 @@ describe('useDataSourceSelectorTree', () => {
     expect(result.current.getTreeDataPending).toBeFalsy();
 
     expect(result.current.treeData).toHaveLength(
-      instanceTipsMockData.length + 2
-    ); // 2 db type +  instance
+      instanceTipsMockData.length + 3
+    ); // 3 db type +  instance
     expect(result.current.treeData).toMatchSnapshot();
 
-    expect(result.current.treeExpandedKeys).toEqual(['MySQL', 'PostgreSQL']);
-    expect(result.current.treeLoadedKeys).toEqual(['MySQL', 'PostgreSQL']);
+    expect(result.current.treeExpandedKeys).toEqual([
+      'MySQL',
+      'PostgreSQL',
+      'TiDB'
+    ]);
+    expect(result.current.treeLoadedKeys).toEqual([
+      'MySQL',
+      'PostgreSQL',
+      'TiDB'
+    ]);
     expect(result.current.selectedBaselineInstanceValue).toBeUndefined();
     expect(result.current.selectedComparisonInstanceValue).toBeUndefined();
   });
@@ -138,7 +146,7 @@ describe('useDataSourceSelectorTree', () => {
     expect(
       result.current.disableTreeNodesBasedOnSelection(
         result.current.treeData[5].value
-      )?.[4].disabled
+      )?.[5].disabled
     ).toBeTruthy();
   });
 });