wip

tests/integration/infrastructure/CheckoutPriceExtractor.test.ts

@@ -28,6 +28,8 @@ describe('CheckoutPriceExtractor Integration', () => {
     // Create nested locator mock for span.label-pill
     mockPillLocator = {
       textContent: vi.fn().mockResolvedValue('$0.50'),
+      first: vi.fn().mockReturnThis(),
+      locator: vi.fn().mockReturnThis(),
     };
 
     mockLocator = {
@@ -35,10 +37,16 @@ describe('CheckoutPriceExtractor Integration', () => {
       innerHTML: vi.fn(),
       textContent: vi.fn(),
       locator: vi.fn(() => mockPillLocator),
+      first: vi.fn().mockReturnThis(),
     };
 
     mockPage = {
-      locator: vi.fn(() => mockLocator),
+      locator: vi.fn((selector) => {
+        if (selector === '.label-pill, .label-inverse') {
+          return mockPillLocator;
+        }
+        return mockLocator;
+      }),
     };
   });