github/openclaw
1
1
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-02-19 18:39:20 -05:00
Code Issues Packages Projects Releases Wiki Activity

Lint: formatting and unused vars in test

Browse Source
This commit is contained in:
Muhammed Mukhthar CM
2026-01-14 07:10:59 +00:00
committed by Peter Steinberger
parent e41a8c8af1
commit f99e2eaebb
1 changed files with 47 additions and 42 deletions
Download Patch File Download Diff File Expand all files Collapse all files

89
src/agents/pi-embedded-runner.sanitize-session-history.test.ts
View File

@@ -1,104 +1,109 @@
import { describe, it, expect, vi, beforeEach } from 'vitest';
import { sanitizeSessionHistory } from './pi-embedded-runner/google.js';
import * as helpers from './pi-embedded-helpers.js';
import type { AgentMessage } from '@mariozechner/pi-agent-core';
import type { SessionManager } from '@mariozechner/pi-coding-agent';
import type { AgentMessage } from "@mariozechner/pi-agent-core";
import type { SessionManager } from "@mariozechner/pi-coding-agent";
import { beforeEach, describe, expect, it, vi } from "vitest";
import * as helpers from "./pi-embedded-helpers.js";
import { sanitizeSessionHistory } from "./pi-embedded-runner/google.js";
// Mock dependencies
vi.mock('./pi-embedded-helpers.js', async () => {
const actual = await vi.importActual('./pi-embedded-helpers.js');
vi.mock("./pi-embedded-helpers.js", async () => {
const actual = await vi.importActual("./pi-embedded-helpers.js");
return {
...actual,
isGoogleModelApi: vi.fn(),
downgradeGeminiHistory: vi.fn(),
sanitizeSessionMessagesImages: vi.fn().mockImplementation(async (msgs) => msgs),
sanitizeSessionMessagesImages: vi
.fn()
.mockImplementation(async (msgs) => msgs),
};
});
// We don't mock session-transcript-repair.js as it is a pure function and complicates mocking.
// We rely on the real implementation which should pass through our simple messages.
describe('sanitizeSessionHistory', () => {
describe("sanitizeSessionHistory", () => {
const mockSessionManager = {
getEntries: vi.fn().mockReturnValue([]),
appendCustomEntry: vi.fn(),
} as unknown as SessionManager;
const mockMessages: AgentMessage[] = [
{ role: 'user', content: 'hello' }
];
const mockMessages: AgentMessage[] = [{ role: "user", content: "hello" }];
beforeEach(() => {
vi.resetAllMocks();
vi.mocked(helpers.sanitizeSessionMessagesImages).mockImplementation(async (msgs) => msgs);
vi.mocked(helpers.sanitizeSessionMessagesImages).mockImplementation(
async (msgs) => msgs,
);
// Default mock implementation
vi.mocked(helpers.downgradeGeminiHistory).mockImplementation((msgs) => {
if (!msgs) return [];
return [...msgs, { role: 'system', content: 'downgraded' }];
if (!msgs) return [];
return [...msgs, { role: "system", content: "downgraded" }];
});
});
it('should downgrade history for Google models if provider is not google-antigravity', async () => {
it("should downgrade history for Google models if provider is not google-antigravity", async () => {
vi.mocked(helpers.isGoogleModelApi).mockReturnValue(true);
const result = await sanitizeSessionHistory({
messages: mockMessages,
modelApi: 'google-gemini',
provider: 'google-vertex',
modelApi: "google-gemini",
provider: "google-vertex",
sessionManager: mockSessionManager,
sessionId: 'test-session',
sessionId: "test-session",
});
expect(helpers.isGoogleModelApi).toHaveBeenCalledWith('google-gemini');
expect(helpers.isGoogleModelApi).toHaveBeenCalledWith("google-gemini");
expect(helpers.downgradeGeminiHistory).toHaveBeenCalled();
// Check if the result contains the downgraded message
expect(result).toContainEqual({ role: 'system', content: 'downgraded' });
expect(result).toContainEqual({ role: "system", content: "downgraded" });
});
it('should NOT downgrade history for google-antigravity provider', async () => {
it("should NOT downgrade history for google-antigravity provider", async () => {
vi.mocked(helpers.isGoogleModelApi).mockReturnValue(true);
const result = await sanitizeSessionHistory({
messages: mockMessages,
modelApi: 'google-gemini',
provider: 'google-antigravity',
modelApi: "google-gemini",
provider: "google-antigravity",
sessionManager: mockSessionManager,
sessionId: 'test-session',
sessionId: "test-session",
});
expect(helpers.isGoogleModelApi).toHaveBeenCalledWith('google-gemini');
expect(helpers.isGoogleModelApi).toHaveBeenCalledWith("google-gemini");
expect(helpers.downgradeGeminiHistory).not.toHaveBeenCalled();
// Result should not contain the downgraded message
expect(result).not.toContainEqual({ role: 'system', content: 'downgraded' });
expect(result).not.toContainEqual({
role: "system",
content: "downgraded",
});
});
it('should NOT downgrade history for non-Google models', async () => {
it("should NOT downgrade history for non-Google models", async () => {
vi.mocked(helpers.isGoogleModelApi).mockReturnValue(false);
const _result = await sanitizeSessionHistory({
messages: mockMessages,
modelApi: 'anthropic-messages',
provider: 'anthropic',
modelApi: "anthropic-messages",
provider: "anthropic",
sessionManager: mockSessionManager,
sessionId: 'test-session',
sessionId: "test-session",
});
expect(helpers.isGoogleModelApi).toHaveBeenCalledWith('anthropic-messages');
expect(helpers.isGoogleModelApi).toHaveBeenCalledWith("anthropic-messages");
expect(helpers.downgradeGeminiHistory).not.toHaveBeenCalled();
});
it('should downgrade history if provider is undefined but model is Google', async () => {
vi.mocked(helpers.isGoogleModelApi).mockReturnValue(true);
it("should downgrade history if provider is undefined but model is Google", async () => {
vi.mocked(helpers.isGoogleModelApi).mockReturnValue(true);
const _result = await sanitizeSessionHistory({
messages: mockMessages,
modelApi: 'google-gemini',
modelApi: "google-gemini",
provider: undefined,
sessionManager: mockSessionManager,
sessionId: 'test-session',
sessionId: "test-session",
});
expect(helpers.isGoogleModelApi).toHaveBeenCalledWith('google-gemini');
expect(helpers.isGoogleModelApi).toHaveBeenCalledWith("google-gemini");
expect(helpers.downgradeGeminiHistory).toHaveBeenCalled();
});
});