fix(locators): Style attribute in ScreenshotAsync

microsoft · May 1, 2024 · b786d35 · b786d35
1 parent c14ebe5
commit b786d35
Show file tree

Hide file tree

Showing 5 changed files with 16 additions and 0 deletions.
diff --git a/src/Playwright.Tests/ElementHandleScreenshotTests.cs b/src/Playwright.Tests/ElementHandleScreenshotTests.cs
@@ -401,4 +401,19 @@ public async Task PathOptionShouldCreateSubdirectories()
         await elementHandle.ScreenshotAsync(new() { Path = outputPath });
         Assert.True(ScreenshotHelper.PixelMatch("screenshot-element-bounding-box.png", outputPath));
     }
+
+    [PlaywrightTest("locator-screenshot.spec.ts", "should hide elements based on attr")]
+    public async Task ShouldHideElementsBasedOnAttr()
+    {
+        await Page.GotoAsync(Server.Prefix + "/grid.html");
+        var locator = Page.Locator("div").Nth(5);
+        await locator.EvaluateAsync("element => element.setAttribute('data-test-screenshot', 'red-background')");
+        var screenshot = await locator.ScreenshotAsync(new()
+        {
+            Style = @"[data-test-screenshot=""red-background""] {
+                background-color: red !important;
+            }",
+        });
+        Assert.True(ScreenshotHelper.PixelMatch("grid-cell-5-red.png", screenshot));
+    }
 }
diff --git a/src/Playwright.Tests/Screenshots/chromium/grid-cell-5-red.png b/src/Playwright.Tests/Screenshots/chromium/grid-cell-5-red.png
diff --git a/src/Playwright.Tests/Screenshots/firefox/grid-cell-5-red.png b/src/Playwright.Tests/Screenshots/firefox/grid-cell-5-red.png
diff --git a/src/Playwright.Tests/Screenshots/webkit/grid-cell-5-red.png b/src/Playwright.Tests/Screenshots/webkit/grid-cell-5-red.png
diff --git a/src/Playwright/Core/ElementHandle.cs b/src/Playwright/Core/ElementHandle.cs
@@ -105,6 +105,7 @@ public async Task<byte[]> ScreenshotAsync(ElementHandleScreenshotOptions options
             ["scale"] = options.Scale,
             ["quality"] = options.Quality,
             ["maskColor"] = options.MaskColor,
+            ["style"] = options.Style,
         };
         if (options.Mask != null)
         {