OmniRoute/tests/unit/volcengine-plan-binding-upsert.test.ts

import test from "node:test";
import assert from "node:assert/strict";
import fs from "node:fs";
import path from "node:path";
import os from "node:os";
import { resetDbInstance } from "../../src/lib/db/core.ts";
import { createProviderConnection, getProviderConnectionById } from "../../src/models/index.ts";
import {
  detectPlan,
  __testing as bindingTesting,
} from "../../src/lib/providers/volcenginePlanBinding.ts";

test("detectPlan returns available: false when account has no active quota windows (unsubscribed)", async (t) => {
  const originalFetch = globalThis.fetch;
  t.after(() => {
    globalThis.fetch = originalFetch;
  });

  // Mock console response with empty QuotaUsage
  globalThis.fetch = async () =>
    new Response(
      JSON.stringify({
        ResponseMetadata: {},
        Result: {
          QuotaUsage: [],
        },
      }),
      { status: 200, headers: { "Content-Type": "application/json" } }
    );

  const coding = await detectPlan("coding", "cookie=1", "csrf=1");
  assert.equal(coding.available, false);

  // Mock console response for Agent with empty quota object
  globalThis.fetch = async () =>
    new Response(
      JSON.stringify({
        ResponseMetadata: {},
        Result: {},
      }),
      { status: 200, headers: { "Content-Type": "application/json" } }
    );

  const agent = await detectPlan("agent", "cookie=1", "csrf=1");
  assert.equal(agent.available, false);
});

test("volcenginePlanBinding upsert rules with SQLite temp isolation", async (t) => {
  const tempDir = fs.mkdtempSync(path.join(os.tmpdir(), "or-volc-test-"));
  const prevDataDir = process.env.DATA_DIR;
  process.env.DATA_DIR = tempDir;

  t.after(() => {
    resetDbInstance();
    if (prevDataDir) process.env.DATA_DIR = prevDataDir;
    else delete process.env.DATA_DIR;
    fs.rmSync(tempDir, { recursive: true, force: true });
  });

  resetDbInstance();

  // T-1: Single connection named 'main' is adopted and updated, preserving name and autoFetchModels
  const conn1 = await createProviderConnection({
    provider: "volcengine-coding-plan",
    name: "main",
    apiKey: "ark-original-key-1",
    providerSpecificData: { autoFetchModels: true, customTag: "keep-me" },
  });

  const updated1 = await bindingTesting.upsertConnection(
    "coding",
    "ark-new-key-1",
    "new-cookie-1",
    "new-csrf-1",
    123,
    { dummy: 1 },
    undefined
  );

  assert.equal(updated1.id, conn1.id);
  assert.equal(updated1.name, "main", "Preserves original custom name 'main'");
  assert.equal(updated1.apiKey, "ark-new-key-1");
  assert.equal(
    updated1.providerSpecificData.autoFetchModels,
    true,
    "Preserves existing PSD autoFetchModels"
  );
  assert.equal(
    updated1.providerSpecificData.customTag,
    "keep-me",
    "Preserves existing PSD customTag"
  );
  assert.equal(updated1.providerSpecificData.volcConsoleCookie, "new-cookie-1");
  assert.equal(updated1.providerSpecificData.volcApiKeyId, 123);

  // T-4: Multiple connections exist without match -> safely creates new connection without clobbering
  await createProviderConnection({
    provider: "volcengine-coding-plan",
    name: "secondary",
    apiKey: "ark-other-key-2",
    providerSpecificData: {},
  });

  const createdNew = await bindingTesting.upsertConnection(
    "coding",
    "ark-brand-new-key-3",
    "new-cookie-3",
    "new-csrf-3",
    999,
    {},
    undefined
  );

  assert.notEqual(createdNew.id, conn1.id);
  assert.equal(createdNew.name, "Volcano Ark Coding Plan");
  const preserved = await getProviderConnectionById(conn1.id as string);
  assert.equal(preserved.apiKey, "ark-new-key-1", "Original connection was NOT clobbered");

  // T-5: targetConnectionId with cross-provider guard
  const agentConn = await createProviderConnection({
    provider: "volcengine-agent-plan",
    name: "agent-main",
    apiKey: "ark-agent-key",
    providerSpecificData: {},
  });

  // Passing conn1.id (which is coding-plan) into agent upsert must NOT match conn1
  const agentUpsertResult = await bindingTesting.upsertConnection(
    "agent",
    "ark-agent-new-key",
    "agent-cookie",
    "agent-csrf",
    888,
    {},
    conn1.id as string // Mismatched provider
  );
  assert.notEqual(agentUpsertResult.id, conn1.id);
  assert.equal(agentUpsertResult.id, agentConn.id, "Matched the single agent connection instead");
});

test("findTargetConnection pure matching logic and single-connection adoption intent", () => {
  const criteria = {
    provider: "volcengine-coding-plan",
    defaultName: "Volcano Ark Coding Plan",
    apiKey: "ark-key-match",
  };

  // 1. targetConnectionId priority match (strictly same provider)
  const list1 = [
    { id: "c1", provider: "volcengine-coding-plan", name: "any" },
    { id: "c2", provider: "volcengine-coding-plan", name: "other" },
  ];
  assert.equal(
    bindingTesting.findTargetConnection(list1, { ...criteria, targetConnectionId: "c2" })?.id,
    "c2"
  );
  // targetConnectionId with mismatched provider does not match
  assert.equal(
    bindingTesting.findTargetConnection(
      [{ id: "c-other", provider: "different-provider", name: "any" }],
      { ...criteria, targetConnectionId: "c-other" }
    ),
    undefined
  );

  // 2. ApiKey match
  const list2 = [
    { id: "c1", provider: "volcengine-coding-plan", apiKey: "ark-key-match" },
    { id: "c2", provider: "volcengine-coding-plan", apiKey: "ark-diff-key" },
  ];
  assert.equal(bindingTesting.findTargetConnection(list2, criteria)?.id, "c1");

  // 3. VolcApiKeyId match
  const list3 = [
    { id: "c1", provider: "volcengine-coding-plan", providerSpecificData: { volcApiKeyId: 777 } },
    { id: "c2", provider: "volcengine-coding-plan", providerSpecificData: { volcApiKeyId: 888 } },
  ];
  assert.equal(
    bindingTesting.findTargetConnection(list3, { ...criteria, apiKey: undefined, apiKeyId: 888 })
      ?.id,
    "c2"
  );

  // 4. Default name match
  const list4 = [
    { id: "c1", provider: "volcengine-coding-plan", name: "custom-name" },
    { id: "c2", provider: "volcengine-coding-plan", name: "Volcano Ark Coding Plan" },
  ];
  assert.equal(
    bindingTesting.findTargetConnection(list4, { ...criteria, apiKey: undefined })?.id,
    "c2"
  );

  // 5. Intentional fallback: single existing connection adoption
  const list5 = [{ id: "c-single", provider: "volcengine-coding-plan", name: "main" }];
  assert.equal(
    bindingTesting.findTargetConnection(list5, { ...criteria, apiKey: undefined })?.id,
    "c-single",
    "Adopts sole existing connection for provider"
  );

  // 6. Multiple connections without match -> undefined (new connection will be created)
  const list6 = [
    { id: "c1", provider: "volcengine-coding-plan", name: "account-a" },
    { id: "c2", provider: "volcengine-coding-plan", name: "account-b" },
  ];
  assert.equal(
    bindingTesting.findTargetConnection(list6, { ...criteria, apiKey: undefined }),
    undefined,
    "Does not clobber when multiple connections exist"
  );
});