1- import { ApplyState , ToolCallState } from "core" ;
1+ import {
2+ ApplyState ,
3+ BrowserSerializedContinueConfig ,
4+ ToolCallState ,
5+ } from "core" ;
26import { describe , expect , it , vi } from "vitest" ;
7+ import { EMPTY_CONFIG } from "../redux/slices/configSlice" ;
38import { assembleEditOutcomeData } from "./editOutcomeLogger" ;
49
510vi . 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+
2038describe ( "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