@@ -102,7 +102,7 @@ const mockQuerySelectorAll = vi.fn().mockImplementation((selector) => {
102102 return mockElements ; // For onlyHighlighted=true case
103103 } else if (
104104 selector ===
105- ".visual-builder__disabled-variant-field, .visual-builder__variant-field, .visual-builder__base-field"
105+ ".visual-builder__disabled-variant-field, .visual-builder__variant-field, .visual-builder__base-field, .visual-builder__lower-order-variant-field "
106106 ) {
107107 return mockElements ; // For onlyHighlighted=false case
108108 }
@@ -304,7 +304,7 @@ describe("useVariantFieldsPostMessageEvent", () => {
304304
305305 // Verify querySelectorAll was called with the correct selector
306306 expect ( mockQuerySelectorAll ) . toHaveBeenCalledWith (
307- ".visual-builder__disabled-variant-field, .visual-builder__variant-field, .visual-builder__base-field"
307+ ".visual-builder__disabled-variant-field, .visual-builder__variant-field, .visual-builder__base-field, .visual-builder__lower-order-variant-field "
308308 ) ;
309309
310310 // Verify that classes were removed from elements correctly
@@ -313,7 +313,8 @@ describe("useVariantFieldsPostMessageEvent", () => {
313313 "visual-builder__disabled-variant-field" ,
314314 "visual-builder__variant-field" ,
315315 visualBuilderStyles ( ) [ "visual-builder__variant-field" ] ,
316- "visual-builder__base-field"
316+ "visual-builder__base-field" ,
317+ "visual-builder__lower-order-variant-field"
317318 ) ;
318319 } ) ;
319320 } ) ;
@@ -439,7 +440,7 @@ describe("removeVariantFieldClass", () => {
439440
440441 // Verify querySelectorAll was called with the correct selector
441442 expect ( mockQuerySelectorAll ) . toHaveBeenCalledWith (
442- ".visual-builder__disabled-variant-field, .visual-builder__variant-field, .visual-builder__base-field"
443+ ".visual-builder__disabled-variant-field, .visual-builder__variant-field, .visual-builder__base-field, .visual-builder__lower-order-variant-field "
443444 ) ;
444445
445446 // Verify classes were removed
@@ -448,7 +449,8 @@ describe("removeVariantFieldClass", () => {
448449 "visual-builder__disabled-variant-field" ,
449450 "visual-builder__variant-field" ,
450451 visualBuilderStyles ( ) [ "visual-builder__variant-field" ] ,
451- "visual-builder__base-field"
452+ "visual-builder__base-field" ,
453+ "visual-builder__lower-order-variant-field"
452454 ) ;
453455 } ) ;
454456 } ) ;
@@ -458,7 +460,7 @@ describe("removeVariantFieldClass", () => {
458460
459461 // Verify querySelectorAll was called with the correct selector
460462 expect ( mockQuerySelectorAll ) . toHaveBeenCalledWith (
461- ".visual-builder__disabled-variant-field, .visual-builder__variant-field, .visual-builder__base-field"
463+ ".visual-builder__disabled-variant-field, .visual-builder__variant-field, .visual-builder__base-field, .visual-builder__lower-order-variant-field "
462464 ) ;
463465 } ) ;
464466} ) ;
0 commit comments