[java] Fixing format of element screenshot command

SeleniumHQ · Dec 10, 2017 · f3e776f · f3e776f
1 parent 0ebad53
commit f3e776f
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/java/client/src/org/openqa/selenium/remote/http/AbstractHttpCommandCodec.java b/java/client/src/org/openqa/selenium/remote/http/AbstractHttpCommandCodec.java
@@ -153,7 +153,7 @@ public AbstractHttpCommandCodec() {
 
     defineCommand(UPLOAD_FILE, post("/session/:sessionId/file"));
     defineCommand(SCREENSHOT, get("/session/:sessionId/screenshot"));
-    defineCommand(ELEMENT_SCREENSHOT, get("/session/:sessionId/screenshot/:id"));
+    defineCommand(ELEMENT_SCREENSHOT, get("/session/:sessionId/element/:id/screenshot"));
     defineCommand(GET_TITLE, get("/session/:sessionId/title"));
 
     defineCommand(FIND_ELEMENT, post("/session/:sessionId/element"));

diff --git a/java/client/test/org/openqa/selenium/TakesScreenshotTest.java b/java/client/test/org/openqa/selenium/TakesScreenshotTest.java
@@ -17,6 +17,7 @@
 
 package org.openqa.selenium;
 
+import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
 import static org.junit.Assert.assertTrue;
 import static org.junit.Assert.fail;
@@ -141,6 +142,29 @@ && getEffectivePlatform(driver).is(LINUX)
     compareColors(expectedColors, actualColors);
   }
 
+  @Test
+  @Ignore(value = CHROME)
+  @Ignore(value = FIREFOX)
+  @Ignore(value = IE)
+  @Ignore(value = SAFARI)
+  public void testShouldCaptureScreenshotOfAnElement() throws Exception {
+    driver.get(appServer.whereIs("screen/screen.html"));
+    WebElement element = driver.findElement(By.id("cell11"));
+
+    byte[] imageData = element.getScreenshotAs(OutputType.BYTES);
+    assertTrue(imageData != null);
+    assertTrue(imageData.length > 0);
+    BufferedImage image = ImageIO.read(new ByteArrayInputStream(imageData));
+    assertTrue(image != null);
+
+    Raster raster = image.getRaster();
+    String hex = String.format("#%02x%02x%02x",
+                               (raster.getSample(1, 1, 0)),
+                               (raster.getSample(1, 1, 1)),
+                               (raster.getSample(1, 1, 2)));
+    assertEquals("#0f12f7", hex);
+  }
+
   @Test
   @Ignore(value = SAFARI, reason = "takes only visible viewport")
   @Ignore(value = CHROME, reason = "takes only visible viewport")