Support element screenshot (#128)

tebeka · Oct 6, 2018 · 3013caa · 3013caa
1 parent 31a4f14
commit 3013caa
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 0 deletions.
diff --git a/remote.go b/remote.go
@@ -1438,3 +1438,15 @@ func (elem *remoteWE) MarshalJSON() ([]byte, error) {
 		webElementIdentifier: elem.id,
 	})
 }
+
+func (elem *remoteWE) Screenshot(scroll bool) ([]byte, error) {
+	data, err := elem.parent.stringCommand(fmt.Sprintf("/session/%%s/element/%s/screenshot", elem.id))
+	if err != nil {
+		return nil, err
+	}
+
+	// Selenium returns a base64 encoded image.
+	buf := []byte(data)
+	decoder := base64.NewDecoder(base64.StdEncoding, bytes.NewBuffer(buf))
+	return ioutil.ReadAll(decoder)
+}
diff --git a/selenium.go b/selenium.go
@@ -408,4 +408,6 @@ type WebElement interface {
 	// CSSProperty returns the value of the specified CSS property of the
 	// element.
 	CSSProperty(name string) (string, error)
+	// Screenshot takes a screenshot of the attribute scroll'ing if necessary.
+	Screenshot(scroll bool) ([]byte, error)
 }