fix(compare): let oracle pick schemas in data compare

apps/desktop/src/components/diff/DataCompareDialog.vue

@@ -319,13 +319,12 @@ async function loadDatabases(connectionId: string, side: "source" | "target") {
   if (!connectionId) return;
   await store.ensureConnected(connectionId);
   const config = store.getConfig(connectionId);
-  const names =
-    config?.db_type === "dameng"
-      ? await fetchNamespaceOptionsForConnection(connectionId, config)
-      : databaseOptionsForConnection(
-          (await api.listDatabases(connectionId)).map((database) => database.name),
-          config,
-        );
+  const names = config
+    ? await fetchNamespaceOptionsForConnection(connectionId, config)
+    : databaseOptionsForConnection(
+        (await api.listDatabases(connectionId)).map((database) => database.name),
+        config,
+      );
   if (side === "source") {
     sourceDatabases.value = names;
     sourceDatabase.value = names.length === 1 ? names[0] : "";

apps/desktop/src/components/diff/__tests__/DataCompareDialog.prefill.spec.ts

@@ -7,18 +7,44 @@ import DataCompareDialog from "@/components/diff/DataCompareDialog.vue";
 
 const mocks = vi.hoisted(() => ({
   ensureConnected: vi.fn().mockResolvedValue(undefined),
-  listDatabases: vi.fn().mockResolvedValue([{ name: "DBX_TEST" }, { name: "OPS$ORACLE" }]),
-  listSchemas: vi.fn().mockResolvedValue(["DBX_TEST"]),
-  listTables: vi.fn().mockResolvedValue([{ name: "CODEX_7046_META", table_type: "TABLE" }]),
+  listDatabases: vi.fn().mockResolvedValue([]),
+  listSchemas: vi.fn().mockResolvedValue(["DBX_TEST", "REPORTING", "SYS"]),
+  listTables: vi.fn().mockResolvedValue([{ name: "CODEX_7467_META", table_type: "TABLE" }]),
   getColumns: vi.fn().mockResolvedValue([{ name: "ID", data_type: "NUMBER", is_primary_key: true }]),
 }));
 
-vi.mock("@/stores/connectionStore", () => ({
-  useConnectionStore: () => ({
-    connections: [{ id: "oracle-11g", name: "Oracle XE 11g", db_type: "oracle", driver_profile: "oracle", database: "XE" }],
-    getConfig: (id: string) => (id === "oracle-11g" ? { id, name: "Oracle XE 11g", db_type: "oracle", driver_profile: "oracle", database: "XE" } : undefined),
-    ensureConnected: mocks.ensureConnected,
-  }),
+function passthrough(tag: string) {
+  return defineComponent({
+    inheritAttrs: false,
+    setup(_, { attrs, slots }) {
+      return () => h(tag, attrs, slots.default?.());
+    },
+  });
+}
+
+vi.mock("@/stores/connectionStore", () => {
+  const connections = [
+    { id: "oracle-11g", name: "Oracle XE 11g", db_type: "oracle", driver_profile: "oracle", database: "XE" },
+    {
+      id: "oracle-jdbc-11g",
+      name: "Oracle JDBC 11g",
+      db_type: "jdbc",
+      driver_profile: "oracle",
+      connection_string: "jdbc:oracle:thin:@//localhost:1521/XE",
+      jdbc_driver_class: "oracle.jdbc.OracleDriver",
+    },
+  ];
+  return {
+    useConnectionStore: () => ({
+      connections,
+      getConfig: (id: string) => connections.find((connection) => connection.id === id),
+      ensureConnected: mocks.ensureConnected,
+    }),
+  };
+});
+
+vi.mock("@/components/connection/ConnectionGroupBadge.vue", () => ({
+  default: passthrough("span"),
 }));
 
 vi.mock("@/lib/backend/api", () => ({
@@ -39,7 +65,7 @@ async function flushAsyncSetup() {
 
 afterEach(() => {
   for (const app of mountedApps.splice(0)) app.unmount();
-  document.body.innerHTML = "";
+  document.body.textContent = "";
   vi.clearAllMocks();
 });
 
@@ -64,8 +90,71 @@ describe("DataCompareDialog source prefill", () => {
     app.mount(container);
     await flushAsyncSetup();
 
+    expect(mocks.listDatabases).not.toHaveBeenCalled();
+    expect(mocks.listSchemas).toHaveBeenCalledWith("oracle-11g", "XE", true);
+
+    const searchableSelectTriggers = [...document.querySelectorAll<HTMLButtonElement>("button.dbx-searchable-select-trigger")];
+    const sourceDatabaseTrigger = searchableSelectTriggers[1];
+    expect(sourceDatabaseTrigger?.title).toBe("DBX_TEST");
+    expect(sourceDatabaseTrigger?.disabled).toBe(false);
+    sourceDatabaseTrigger?.click();
+    await flushAsyncSetup();
+
+    const databaseOptions = [...document.querySelectorAll<HTMLButtonElement>(".dbx-searchable-select-list button")].map((button) => button.textContent?.trim());
+    expect(databaseOptions).toEqual(expect.arrayContaining(["DBX_TEST", "REPORTING"]));
     expect(mocks.listTables).toHaveBeenCalledWith("oracle-11g", "DBX_TEST", "DBX_TEST");
-    expect(document.body.textContent).toContain("CODEX_7046_META");
+    expect(document.body.textContent).toContain("CODEX_7467_META");
     expect(document.body.textContent).not.toContain("暂无可比较的表");
   });
+
+  it("loads schemas and tables after selecting an Oracle JDBC target connection", async () => {
+    const container = document.createElement("div");
+    document.body.append(container);
+    const app = createApp(
+      defineComponent({
+        setup: () => () =>
+          h(DataCompareDialog, {
+            open: true,
+            prefillConnectionId: "oracle-11g",
+            prefillDatabase: "DBX_TEST",
+            prefillSchema: "DBX_TEST",
+            prefillTable: "CODEX_7467_META",
+          }),
+      }),
+    );
+    mountedApps.push(app);
+    app.use(i18n);
+    app.mount(container);
+    await flushAsyncSetup();
+
+    const searchableSelectTriggers = [...document.querySelectorAll<HTMLButtonElement>("button.dbx-searchable-select-trigger")];
+    const targetConnectionTrigger = searchableSelectTriggers[3];
+    expect(targetConnectionTrigger).toBeDefined();
+    targetConnectionTrigger?.click();
+    await flushAsyncSetup();
+
+    const jdbcConnectionOption = [...document.querySelectorAll<HTMLButtonElement>(".dbx-searchable-select-list button")].find((button) => button.textContent?.trim() === "Oracle JDBC 11g");
+    expect(jdbcConnectionOption).toBeDefined();
+    jdbcConnectionOption?.click();
+    await flushAsyncSetup();
+
+    expect(mocks.listSchemas).toHaveBeenCalledWith("oracle-jdbc-11g", "", true);
+
+    const triggersAfterTargetLoad = [...document.querySelectorAll<HTMLButtonElement>("button.dbx-searchable-select-trigger")];
+    const targetDatabaseTrigger = triggersAfterTargetLoad[4];
+    expect(targetDatabaseTrigger?.disabled).toBe(false);
+    targetDatabaseTrigger?.click();
+    await flushAsyncSetup();
+
+    const targetDatabaseOptions = [...document.querySelectorAll<HTMLButtonElement>(".dbx-searchable-select-list button")].map((button) => button.textContent?.trim());
+    expect(targetDatabaseOptions).toEqual(expect.arrayContaining(["DBX_TEST", "REPORTING"]));
+
+    const reportingOption = [...document.querySelectorAll<HTMLButtonElement>(".dbx-searchable-select-list button")].find((button) => button.textContent?.trim() === "REPORTING");
+    expect(reportingOption).toBeDefined();
+    reportingOption?.click();
+    await flushAsyncSetup();
+
+    expect(mocks.listTables).toHaveBeenCalledWith("oracle-jdbc-11g", "REPORTING", "DBX_TEST");
+    expect(document.body.textContent).toContain("CODEX_7467_META");
+  });
 });