openclaw/src/config/config.pruning-defaults.test.ts

import path from "node:path";
import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
import { setBundledPluginsDirOverrideForTest } from "../plugins/bundled-dir.js";
import { resetBundledPluginPublicArtifactLoaderForTest } from "../plugins/public-surface-loader.js";
import type { OpenClawConfig } from "./config.js";
import { applyProviderConfigDefaultsForConfig } from "./provider-policy.js";

function expectAnthropicPruningDefaults(cfg: OpenClawConfig, heartbeatEvery = "30m") {
  expect(cfg.agents?.defaults?.contextPruning?.mode).toBe("cache-ttl");
  expect(cfg.agents?.defaults?.contextPruning?.ttl).toBe("1h");
  expect(cfg.agents?.defaults?.heartbeat?.every).toBe(heartbeatEvery);
}

function applyAnthropicDefaultsForTest(config: OpenClawConfig) {
  return applyProviderConfigDefaultsForConfig({ provider: "anthropic", config, env: {} });
}

describe("config pruning defaults", () => {
  beforeEach(() => {
    setBundledPluginsDirOverrideForTest(path.resolve(import.meta.dirname, "../../extensions"));
    resetBundledPluginPublicArtifactLoaderForTest();
    vi.stubEnv(
      "OPENCLAW_BUNDLED_PLUGINS_DIR",
      path.resolve(import.meta.dirname, "../../extensions"),
    );
  });

  afterEach(() => {
    setBundledPluginsDirOverrideForTest(undefined);
    resetBundledPluginPublicArtifactLoaderForTest();
    vi.unstubAllEnvs();
  });

  it("does not enable contextPruning by default", () => {
    const cfg = applyAnthropicDefaultsForTest({ agents: { defaults: {} } });

    expect(cfg.agents?.defaults?.contextPruning?.mode).toBeUndefined();
  });

  it("enables cache-ttl pruning + 1h heartbeat for Anthropic OAuth", () => {
    const cfg = applyAnthropicDefaultsForTest({
      auth: {
        profiles: {
          "anthropic:me": { provider: "anthropic", mode: "oauth", email: "me@example.com" },
        },
      },
      agents: { defaults: {} },
    });

    expectAnthropicPruningDefaults(cfg, "1h");
  });

  it("backfills raw and canonical Claude CLI policies for selected Anthropic CLI auth", () => {
    const cfg = applyAnthropicDefaultsForTest({
      auth: {
        order: { anthropic: ["anthropic:claude-cli"] },
        profiles: {
          "anthropic:claude-cli": { provider: "claude-cli", mode: "oauth" },
        },
      },
      agents: {
        defaults: {
          model: { primary: "anthropic/opus-4.7" },
          models: {
            "anthropic/opus-4.7": { params: { maxTokens: 1200 } },
          },
        },
      },
    });

    expect(cfg.agents?.defaults?.models?.["anthropic/opus-4.7"]).toEqual({
      params: { maxTokens: 1200 },
      agentRuntime: { id: "claude-cli" },
    });
    expect(cfg.agents?.defaults?.models?.["anthropic/claude-opus-4-7"]).toEqual({
      agentRuntime: { id: "claude-cli" },
    });
  });

  it("enables cache-ttl pruning + 1h cache TTL for Anthropic API keys", () => {
    const cfg = applyAnthropicDefaultsForTest({
      auth: {
        profiles: {
          "anthropic:api": { provider: "anthropic", mode: "api_key" },
        },
      },
      agents: {
        defaults: {
          model: { primary: "anthropic/claude-opus-4-6" },
        },
      },
    });

    expectAnthropicPruningDefaults(cfg);
    expect(
      cfg.agents?.defaults?.models?.["anthropic/claude-opus-4-6"]?.params?.cacheRetention,
    ).toBe("short");
  });

  it("adds default cacheRetention for Anthropic Claude models on Bedrock", () => {
    const cfg = applyAnthropicDefaultsForTest({
      auth: {
        profiles: {
          "anthropic:api": { provider: "anthropic", mode: "api_key" },
        },
      },
      agents: {
        defaults: {
          model: { primary: "amazon-bedrock/us.anthropic.claude-opus-4-6-v1" },
        },
      },
    });

    expect(
      cfg.agents?.defaults?.models?.["amazon-bedrock/us.anthropic.claude-opus-4-6-v1"]?.params
        ?.cacheRetention,
    ).toBe("short");
  });

  it("does not add default cacheRetention for non-Anthropic Bedrock models", () => {
    const cfg = applyAnthropicDefaultsForTest({
      auth: {
        profiles: {
          "anthropic:api": { provider: "anthropic", mode: "api_key" },
        },
      },
      agents: {
        defaults: {
          model: { primary: "amazon-bedrock/amazon.nova-micro-v1:0" },
        },
      },
    });

    expect(
      cfg.agents?.defaults?.models?.["amazon-bedrock/amazon.nova-micro-v1:0"]?.params
        ?.cacheRetention,
    ).toBeUndefined();
  });

  it("does not override explicit contextPruning mode", () => {
    const cfg = applyAnthropicDefaultsForTest({
      auth: {
        profiles: {
          "anthropic:api": { provider: "anthropic", mode: "api_key" },
        },
      },
      agents: { defaults: { contextPruning: { mode: "off" } } },
    });

    expect(cfg.agents?.defaults?.contextPruning?.mode).toBe("off");
  });
});