fix(locators): Style attribute in ScreenshotAsync (#2926)

microsoft · May 1, 2024 · 3edc14f · 3edc14f
1 parent 9ef9165
commit 3edc14f
Show file tree

Hide file tree

Showing 2 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 });
         PlaywrightAssert.ToMatchSnapshot("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;
+            }",
+        });
+        PlaywrightAssert.ToMatchSnapshot("grid-cell-5-red.png", screenshot);
+    }
 }
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)
         {