vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-06 05:30:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(agents): isolate auth profile store fixtures

Browse Source
This commit is contained in:
Vincent Koc
2026-05-05 14:53:35 -07:00
parent d38e30e02c
commit a36981a2c5
1 changed files with 28 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

70
src/agents/auth-profiles.ensureauthprofilestore.test.ts
View File

@@ -104,6 +104,10 @@ describe("ensureAuthProfileStore", () => {
restoreEnvValue("PI_CODING_AGENT_DIR", params.previousPiAgentDir);
}
function resolveDefaultMainAgentDir(stateDir: string): string {
return path.join(stateDir, "agents", "main", "agent");
}
function expectApiKeyProfile(
profile: AuthProfileCredential,
): Extract<AuthProfileCredential, { type: "api_key" }> {
@@ -167,17 +171,14 @@ describe("ensureAuthProfileStore", () => {
it("merges main auth profiles into agent store and keeps agent overrides", () => {
const root = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-auth-merge-"));
const previousAgentDir = process.env.OPENCLAW_AGENT_DIR;
const previousPiAgentDir = process.env.PI_CODING_AGENT_DIR;
const previousStateDir = process.env.OPENCLAW_STATE_DIR;
try {
const mainDir = path.join(root, "main-agent");
process.env.OPENCLAW_STATE_DIR = root;
const mainDir = resolveDefaultMainAgentDir(root);
const agentDir = path.join(root, "agent-x");
fs.mkdirSync(mainDir, { recursive: true });
fs.mkdirSync(agentDir, { recursive: true });
process.env.OPENCLAW_AGENT_DIR = mainDir;
process.env.PI_CODING_AGENT_DIR = mainDir;
const mainStore = {
version: AUTH_STORE_VERSION,
profiles: {
@@ -227,24 +228,21 @@ describe("ensureAuthProfileStore", () => {
key: "agent-key",
});
} finally {
restoreAgentDirEnv({ previousAgentDir, previousPiAgentDir });
restoreEnvValue("OPENCLAW_STATE_DIR", previousStateDir);
fs.rmSync(root, { recursive: true, force: true });
}
});
it("uses the main agent's newer OAuth profile when an agent still has a stale default profile", () => {
const root = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-auth-drift-"));
const previousAgentDir = process.env.OPENCLAW_AGENT_DIR;
const previousPiAgentDir = process.env.PI_CODING_AGENT_DIR;
const previousStateDir = process.env.OPENCLAW_STATE_DIR;
try {
const mainDir = path.join(root, "main-agent");
process.env.OPENCLAW_STATE_DIR = root;
const mainDir = resolveDefaultMainAgentDir(root);
const agentDir = path.join(root, "agent-x");
fs.mkdirSync(mainDir, { recursive: true });
fs.mkdirSync(agentDir, { recursive: true });
process.env.OPENCLAW_AGENT_DIR = mainDir;
process.env.PI_CODING_AGENT_DIR = mainDir;
const freshProfileId = "openai-codex:user@example.com";
const staleProfileId = "openai-codex:default";
saveAuthProfileStore(
@@ -325,24 +323,21 @@ describe("ensureAuthProfileStore", () => {
) as { profiles: Record<string, unknown> };
expect(persistedAgentStore.profiles[staleProfileId]).toBeDefined();
} finally {
restoreAgentDirEnv({ previousAgentDir, previousPiAgentDir });
restoreEnvValue("OPENCLAW_STATE_DIR", previousStateDir);
fs.rmSync(root, { recursive: true, force: true });
}
});
it("keeps a newer agent replacement credential while repairing stale default references", () => {
const root = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-auth-drift-newer-agent-"));
const previousAgentDir = process.env.OPENCLAW_AGENT_DIR;
const previousPiAgentDir = process.env.PI_CODING_AGENT_DIR;
const previousStateDir = process.env.OPENCLAW_STATE_DIR;
try {
const mainDir = path.join(root, "main-agent");
process.env.OPENCLAW_STATE_DIR = root;
const mainDir = resolveDefaultMainAgentDir(root);
const agentDir = path.join(root, "agent-x");
fs.mkdirSync(mainDir, { recursive: true });
fs.mkdirSync(agentDir, { recursive: true });
process.env.OPENCLAW_AGENT_DIR = mainDir;
process.env.PI_CODING_AGENT_DIR = mainDir;
const freshProfileId = "openai-codex:user@example.com";
const staleProfileId = "openai-codex:default";
saveAuthProfileStore(
@@ -408,24 +403,21 @@ describe("ensureAuthProfileStore", () => {
expect(store.order?.["openai-codex"]).toEqual([freshProfileId]);
expect(store.lastGood?.["openai-codex"]).toBe(freshProfileId);
} finally {
restoreAgentDirEnv({ previousAgentDir, previousPiAgentDir });
restoreEnvValue("OPENCLAW_STATE_DIR", previousStateDir);
fs.rmSync(root, { recursive: true, force: true });
}
});
it("preserves a valid main default OAuth profile while replacing a stale agent override", () => {
const root = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-auth-drift-base-default-"));
const previousAgentDir = process.env.OPENCLAW_AGENT_DIR;
const previousPiAgentDir = process.env.PI_CODING_AGENT_DIR;
const previousStateDir = process.env.OPENCLAW_STATE_DIR;
try {
const mainDir = path.join(root, "main-agent");
process.env.OPENCLAW_STATE_DIR = root;
const mainDir = resolveDefaultMainAgentDir(root);
const agentDir = path.join(root, "agent-x");
fs.mkdirSync(mainDir, { recursive: true });
fs.mkdirSync(agentDir, { recursive: true });
process.env.OPENCLAW_AGENT_DIR = mainDir;
process.env.PI_CODING_AGENT_DIR = mainDir;
const freshProfileId = "openai-codex:user@example.com";
const defaultProfileId = "openai-codex:default";
saveAuthProfileStore(
@@ -497,24 +489,21 @@ describe("ensureAuthProfileStore", () => {
lastUsed: 123,
});
} finally {
restoreAgentDirEnv({ previousAgentDir, previousPiAgentDir });
restoreEnvValue("OPENCLAW_STATE_DIR", previousStateDir);
fs.rmSync(root, { recursive: true, force: true });
}
});
it("keeps a stale default OAuth profile when the main profile belongs to a different identity", () => {
const root = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-auth-drift-mismatch-"));
const previousAgentDir = process.env.OPENCLAW_AGENT_DIR;
const previousPiAgentDir = process.env.PI_CODING_AGENT_DIR;
const previousStateDir = process.env.OPENCLAW_STATE_DIR;
try {
const mainDir = path.join(root, "main-agent");
process.env.OPENCLAW_STATE_DIR = root;
const mainDir = resolveDefaultMainAgentDir(root);
const agentDir = path.join(root, "agent-x");
fs.mkdirSync(mainDir, { recursive: true });
fs.mkdirSync(agentDir, { recursive: true });
process.env.OPENCLAW_AGENT_DIR = mainDir;
process.env.PI_CODING_AGENT_DIR = mainDir;
const freshProfileId = "openai-codex:user@example.com";
const staleProfileId = "openai-codex:default";
saveAuthProfileStore(
@@ -568,24 +557,21 @@ describe("ensureAuthProfileStore", () => {
expect(store.order?.["openai-codex"]).toEqual([staleProfileId]);
expect(store.lastGood?.["openai-codex"]).toBe(staleProfileId);
} finally {
restoreAgentDirEnv({ previousAgentDir, previousPiAgentDir });
restoreEnvValue("OPENCLAW_STATE_DIR", previousStateDir);
fs.rmSync(root, { recursive: true, force: true });
}
});
it("rewrites invalidated per-agent Codex order to the main agent's healthy relogin profile", () => {
const root = fs.mkdtempSync(path.join(os.tmpdir(), "openclaw-auth-codex-relogin-"));
const previousAgentDir = process.env.OPENCLAW_AGENT_DIR;
const previousPiAgentDir = process.env.PI_CODING_AGENT_DIR;
const previousStateDir = process.env.OPENCLAW_STATE_DIR;
try {
const mainDir = path.join(root, "main-agent");
process.env.OPENCLAW_STATE_DIR = root;
const mainDir = resolveDefaultMainAgentDir(root);
const agentDir = path.join(root, "agent-x");
fs.mkdirSync(mainDir, { recursive: true });
fs.mkdirSync(agentDir, { recursive: true });
process.env.OPENCLAW_AGENT_DIR = mainDir;
process.env.PI_CODING_AGENT_DIR = mainDir;
const now = Date.now();
const healthyProfileId = "openai-codex:bunsthedev@gmail.com";
const staleProfileId = "openai-codex:val@viewdue.ai";
@@ -656,7 +642,7 @@ describe("ensureAuthProfileStore", () => {
expect(store.lastGood?.["openai-codex"]).toBe(healthyProfileId);
expect(store.usageStats?.[staleProfileId]).toBeUndefined();
} finally {
restoreAgentDirEnv({ previousAgentDir, previousPiAgentDir });
restoreEnvValue("OPENCLAW_STATE_DIR", previousStateDir);
fs.rmSync(root, { recursive: true, force: true });
}
});