deer-flow/frontend/tests/unit/core/channels/provider-state.test.ts

import { describe, expect, it } from "@rstest/core";

import {
  providerCanConnect,
  providerCanEditRuntimeConfig,
  providerNeedsRuntimeConfig,
} from "@/core/channels/provider-state";
import type { ChannelProvider } from "@/core/channels/types";

function makeProvider(overrides: Partial<ChannelProvider>): ChannelProvider {
  return {
    provider: "slack",
    display_name: "Slack",
    enabled: true,
    configured: true,
    connectable: true,
    auth_mode: "binding_code",
    connection_status: "not_connected",
    credential_fields: [
      {
        name: "bot_token",
        label: "Bot token",
        type: "password",
        required: true,
      },
    ],
    ...overrides,
  };
}

describe("providerCanConnect", () => {
  it("allows connecting a configured, not yet connected provider", () => {
    expect(providerCanConnect(makeProvider({}))).toBe(true);
  });

  it("rejects an already connected provider", () => {
    expect(
      providerCanConnect(makeProvider({ connection_status: "connected" })),
    ).toBe(false);
  });

  it("rejects a non-connectable provider", () => {
    expect(providerCanConnect(makeProvider({ connectable: false }))).toBe(
      false,
    );
  });

  it("falls back to enabled+configured when connectable is missing", () => {
    expect(providerCanConnect(makeProvider({ connectable: undefined }))).toBe(
      true,
    );
    expect(
      providerCanConnect(
        makeProvider({ connectable: undefined, configured: false }),
      ),
    ).toBe(false);
  });
});

describe("providerNeedsRuntimeConfig", () => {
  it("requires setup only when enabled and unconfigured with fields", () => {
    expect(
      providerNeedsRuntimeConfig(makeProvider({ configured: false })),
    ).toBe(true);
    expect(providerNeedsRuntimeConfig(makeProvider({}))).toBe(false);
    expect(
      providerNeedsRuntimeConfig(
        makeProvider({ configured: false, enabled: false }),
      ),
    ).toBe(false);
    expect(
      providerNeedsRuntimeConfig(
        makeProvider({ configured: false, credential_fields: [] }),
      ),
    ).toBe(false);
  });
});

describe("providerCanEditRuntimeConfig", () => {
  it("is editable whenever enabled with credential fields", () => {
    expect(providerCanEditRuntimeConfig(makeProvider({}))).toBe(true);
    expect(providerCanEditRuntimeConfig(makeProvider({ enabled: false }))).toBe(
      false,
    );
    expect(
      providerCanEditRuntimeConfig(makeProvider({ credential_fields: [] })),
    ).toBe(false);
  });
});