Skip to content

Commit 532c99b

Browse files
committed
fix: outcome logger test
1 parent 6162ca1 commit 532c99b

File tree

1 file changed

+33
-3
lines changed

1 file changed

+33
-3
lines changed

gui/src/util/editOutcomeLogger.test.ts

Lines changed: 33 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,10 @@
1-
import { ApplyState, ToolCallState } from "core";
1+
import {
2+
ApplyState,
3+
BrowserSerializedContinueConfig,
4+
ToolCallState,
5+
} from "core";
26
import { describe, expect, it, vi } from "vitest";
7+
import { EMPTY_CONFIG } from "../redux/slices/configSlice";
38
import { assembleEditOutcomeData } from "./editOutcomeLogger";
49

510
vi.mock("../redux/store", () => ({
@@ -17,6 +22,19 @@ vi.mock("../redux/store", () => ({
1722
},
1823
}));
1924

25+
const EMPTY_CONFIG_WITH_TEST_MODEL_SELECTED: BrowserSerializedContinueConfig = {
26+
...EMPTY_CONFIG,
27+
selectedModelByRole: {
28+
...EMPTY_CONFIG.selectedModelByRole,
29+
chat: {
30+
provider: "test",
31+
model: "test-model",
32+
title: "Test model",
33+
underlyingProviderName: "provider",
34+
},
35+
},
36+
};
37+
2038
describe("assembleEditOutcomeData", () => {
2139
it("should assemble complete edit outcome data correctly", () => {
2240
const toolCallState: ToolCallState = {
@@ -37,7 +55,13 @@ describe("assembleEditOutcomeData", () => {
3755
fileContent: "const new = 2;\nconst another = 3;",
3856
};
3957

40-
const result = assembleEditOutcomeData(toolCallState, applyState, true);
58+
const result = assembleEditOutcomeData(
59+
[],
60+
EMPTY_CONFIG_WITH_TEST_MODEL_SELECTED,
61+
toolCallState,
62+
applyState,
63+
true,
64+
);
4165
console.log(result);
4266

4367
expect(result).toMatchObject({
@@ -71,7 +95,13 @@ describe("assembleEditOutcomeData", () => {
7195
fileContent: "",
7296
};
7397

74-
const result = assembleEditOutcomeData(toolCallState, applyState, false);
98+
const result = assembleEditOutcomeData(
99+
[],
100+
EMPTY_CONFIG_WITH_TEST_MODEL_SELECTED,
101+
toolCallState,
102+
applyState,
103+
false,
104+
);
75105

76106
expect(result.previousCodeLines).toBe(0);
77107
expect(result.newCodeLines).toBe(0);

0 commit comments

Comments
 (0)