@skyramp/skyramp 1.2.32 → 1.2.34

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@skyramp/skyramp",
-  "version": "1.2.32",
+  "version": "1.2.34",
   "description": "module for leveraging skyramp cli functionality",
   "scripts": {
     "lint": "eslint 'src/**/*.js' 'src/**/*.ts' --fix",

package/src/classes/SmartPlaywright.d.ts

@@ -1,5 +1,5 @@
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
-export function newSkyrampPlaywrightPage(page: any, testInfo: any): any;
+export function newSkyrampPlaywrightPage(page: any, testInfo?: any): any;
 
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 export function expect(obj: any): any;

package/src/classes/SmartPlaywright.js

@@ -816,10 +816,11 @@ function newSkyrampPlaywrightPage(page, testInfo) {
  * Check if snapshot exists, create if missing, then compare. Enabling single-run screenshots
  */
 class SkyrampPageAssertions {
-    constructor(playwrightExpectation, actualObject, testInfo) {
+    constructor(playwrightExpectation, actualObject, testInfo, pageOrLocator) {
         this._playwrightExpectation = playwrightExpectation;
         this._actualObject = actualObject;
         this._testInfo = testInfo;
+        this._pageOrLocator = pageOrLocator || actualObject;
         this._autoBaseline = process.env.SKYRAMP_AUTO_BASELINE !== 'false';
 
         // Proxy to forward all methods except toHaveScreenshot
@@ -842,7 +843,7 @@ class SkyrampPageAssertions {
      */
     async toHaveScreenshot(nameOrOptions, options) {
         // wait for some time before taking snapshot
-        this._actualObject.waitForTimeout(2000);
+        await this._pageOrLocator.waitForTimeout(2000);
 
         // If auto-baseline disabled or no testInfo, use standard Playwright behavior
         if (!this._autoBaseline || !this._testInfo) {
@@ -895,8 +896,13 @@ function expect(obj, testInfo) {
     let extractedTestInfo = testInfo;
 
     if (obj instanceof SkyrampPlaywrightLocator) {
-        // locators do not have toHaveScreenshot
-        return playwrightExpect(obj.unwrap());
+        actualObject = obj.unwrap();
+        // Get testInfo from the locator's parent page
+        if (!extractedTestInfo && obj._skyrampPage && obj._skyrampPage._testInfo) {
+            extractedTestInfo = obj._skyrampPage._testInfo;
+        }
+        const playwrightExpectation = playwrightExpect(actualObject);
+        return new SkyrampPageAssertions(playwrightExpectation, actualObject, extractedTestInfo, actualObject.page());
     }
     if (obj instanceof SkyrampPlaywrightPage) {
         actualObject = obj.unwrap();