@@ -246,7 +246,7 @@ describe("ClineProvider", () => {
246246 // Mock CustomModesManager
247247 const mockCustomModesManager = {
248248 updateCustomMode : jest . fn ( ) . mockResolvedValue ( undefined ) ,
249- getCustomModes : jest . fn ( ) . mockResolvedValue ( { } ) ,
249+ getCustomModes : jest . fn ( ) . mockResolvedValue ( { customModes : [ ] } ) ,
250250 dispose : jest . fn ( ) ,
251251 }
252252
@@ -1049,7 +1049,7 @@ describe("ClineProvider", () => {
10491049 "900x600" , // browserViewportSize
10501050 "code" , // mode
10511051 { } , // customModePrompts
1052- { } , // customModes
1052+ { customModes : [ ] } , // customModes
10531053 undefined , // effectiveInstructions
10541054 undefined , // preferredLanguage
10551055 true , // diffEnabled
@@ -1102,7 +1102,7 @@ describe("ClineProvider", () => {
11021102 "900x600" , // browserViewportSize
11031103 "code" , // mode
11041104 { } , // customModePrompts
1105- { } , // customModes
1105+ { customModes : [ ] } , // customModes
11061106 undefined , // effectiveInstructions
11071107 undefined , // preferredLanguage
11081108 false , // diffEnabled
@@ -1220,12 +1220,14 @@ describe("ClineProvider", () => {
12201220 provider . customModesManager = {
12211221 updateCustomMode : jest . fn ( ) . mockResolvedValue ( undefined ) ,
12221222 getCustomModes : jest . fn ( ) . mockResolvedValue ( {
1223- "test-mode" : {
1224- slug : "test-mode" ,
1225- name : "Test Mode" ,
1226- roleDefinition : "Updated role definition" ,
1227- groups : [ "read" ] as const ,
1228- } ,
1223+ customModes : [
1224+ {
1225+ slug : "test-mode" ,
1226+ name : "Test Mode" ,
1227+ roleDefinition : "Updated role definition" ,
1228+ groups : [ "read" ] as const ,
1229+ } ,
1230+ ] ,
12291231 } ) ,
12301232 dispose : jest . fn ( ) ,
12311233 } as any
@@ -1251,27 +1253,29 @@ describe("ClineProvider", () => {
12511253 )
12521254
12531255 // Verify state was updated
1254- expect ( mockContext . globalState . update ) . toHaveBeenCalledWith (
1255- "customModes" ,
1256- expect . objectContaining ( {
1257- "test-mode" : expect . objectContaining ( {
1256+ expect ( mockContext . globalState . update ) . toHaveBeenCalledWith ( "customModes" , {
1257+ customModes : [
1258+ expect . objectContaining ( {
12581259 slug : "test-mode" ,
12591260 roleDefinition : "Updated role definition" ,
12601261 } ) ,
1261- } ) ,
1262- )
1262+ ] ,
1263+ } )
12631264
12641265 // Verify state was posted to webview
1266+ // Verify state was posted to webview with correct format
12651267 expect ( mockPostMessage ) . toHaveBeenCalledWith (
12661268 expect . objectContaining ( {
12671269 type : "state" ,
12681270 state : expect . objectContaining ( {
1269- customModes : expect . objectContaining ( {
1270- "test-mode" : expect . objectContaining ( {
1271- slug : "test-mode" ,
1272- roleDefinition : "Updated role definition" ,
1273- } ) ,
1274- } ) ,
1271+ customModes : {
1272+ customModes : [
1273+ expect . objectContaining ( {
1274+ slug : "test-mode" ,
1275+ roleDefinition : "Updated role definition" ,
1276+ } ) ,
1277+ ] ,
1278+ } ,
12751279 } ) ,
12761280 } ) ,
12771281 )
0 commit comments