@allwright.dev/vitest 0.0.25 → 0.0.26

package/dist/index.d.ts

@@ -1,4 +1,4 @@
-import { Page, type Browser, type LaunchOptions, type WaitForSelectorOptions } from "@allwright.dev/core";
+import { Locator, Page, type Browser, type LaunchOptions, type WaitForSelectorOptions } from "@allwright.dev/core";
 import { expect as vitestExpect } from "vitest";
 export interface AllwrightVitestOptions {
     launchOptions?: LaunchOptions;
@@ -26,6 +26,12 @@ export interface PageExpectMatchers {
     toHaveCount(selector: string, expected: number, options?: RetryExpectationOptions): Promise<void>;
     toBeVisible(selector: string, options?: VisibleExpectationOptions): Promise<void>;
 }
+export interface LocatorExpectMatchers {
+    toHaveText(expected: string | RegExp, options?: TextExpectationOptions): Promise<void>;
+    toContainText(expected: string | RegExp, options?: TextExpectationOptions): Promise<void>;
+    toHaveCount(expected: number, options?: RetryExpectationOptions): Promise<void>;
+    toBeVisible(options?: VisibleExpectationOptions): Promise<void>;
+}
 export declare const test: import("vitest").TestAPI<{
     browser: Browser;
     page: Page;
@@ -35,8 +41,9 @@ type VitestExpect = typeof vitestExpect;
 type VitestMatcherReturn = ReturnType<VitestExpect>;
 interface AllwrightExpect extends VitestExpect {
     (actual: Page): PageExpectMatchers;
+    (actual: Locator): LocatorExpectMatchers;
     <T>(actual: T): VitestMatcherReturn;
 }
 export declare const expect: AllwrightExpect;
 export { vitestExpect };
-export type { Browser, LaunchOptions, Page };
+export type { Browser, LaunchOptions, Locator, Page };

package/dist/index.js

@@ -1,4 +1,4 @@
-import { chromium, setServerAddr, shutdown, Page, } from "@allwright.dev/core";
+import { chromium, setServerAddr, shutdown, Locator, Page, } from "@allwright.dev/core";
 import { expect as vitestExpect, test as base } from "vitest";
 export const test = base.extend({
     allwright: async ({}, use) => {
@@ -26,6 +26,9 @@ const DEFAULT_EXPECT_INTERVAL_MS = 100;
 function isPage(value) {
     return value instanceof Page;
 }
+function isLocator(value) {
+    return value instanceof Locator;
+}
 async function retryExpectation(callback, options = {}) {
     const timeoutMs = options.timeoutMs ?? DEFAULT_EXPECT_TIMEOUT_MS;
     const intervalMs = options.intervalMs ?? DEFAULT_EXPECT_INTERVAL_MS;
@@ -85,7 +88,49 @@ function createPageExpect(page) {
         },
     };
 }
+function createLocatorExpect(locator) {
+    return {
+        async toHaveText(expected, options = {}) {
+            await retryExpectation(async () => {
+                const result = await locator.textContent(options.command ?? {});
+                if (expected instanceof RegExp) {
+                    vitestExpect(result.text).toMatch(expected);
+                    return;
+                }
+                vitestExpect(result.text).toBe(expected);
+            }, options);
+        },
+        async toContainText(expected, options = {}) {
+            await retryExpectation(async () => {
+                const result = await locator.textContent(options.command ?? {});
+                if (expected instanceof RegExp) {
+                    vitestExpect(result.text).toMatch(expected);
+                    return;
+                }
+                vitestExpect(result.text).toContain(expected);
+            }, options);
+        },
+        async toHaveCount(expected, options = {}) {
+            await retryExpectation(async () => {
+                const result = await locator.count({});
+                vitestExpect(result.count).toBe(expected);
+            }, options);
+        },
+        async toBeVisible(options = {}) {
+            await retryExpectation(async () => {
+                const result = await locator.waitFor({
+                    visible: true,
+                    ...(options.command ?? {}),
+                });
+                vitestExpect(result.visible).toBe(true);
+            }, options);
+        },
+    };
+}
 const expectImpl = ((actual) => {
+    if (isLocator(actual)) {
+        return createLocatorExpect(actual);
+    }
     if (isPage(actual)) {
         return createPageExpect(actual);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@allwright.dev/vitest",
-  "version": "0.0.25",
+  "version": "0.0.26",
   "description": "Vitest fixtures for the allwright TypeScript client.",
   "license": "MIT",
   "type": "module",
@@ -39,7 +39,7 @@
     "prepublishOnly": "npm run build"
   },
   "dependencies": {
-    "@allwright.dev/core": "0.0.25"
+    "@allwright.dev/core": "0.0.26"
   },
   "devDependencies": {
     "typescript": "^5.9.2",