diff --git a/lib/watir-get-image-content/image.rb b/lib/watir-get-image-content/image.rb index ebff8dc..9c597e6 100644 --- a/lib/watir-get-image-content/image.rb +++ b/lib/watir-get-image-content/image.rb @@ -7,13 +7,13 @@ class Image def to_x_base64(target) assert_exists js = %Q{var canvas = document.createElement("canvas"); - canvas.width = arguments[0].width; - canvas.height = arguments[0].height; + canvas.width = arguments[1]; + canvas.height = arguments[2]; var ctx = canvas.getContext("2d"); ctx.drawImage(arguments[0], 0, 0); var dataURL = canvas.toDataURL("image/#{target}"); return dataURL.replace(/^data:image\\/(png|jpg|gif);base64,/, "");} - driver.execute_script js, @element + driver.execute_script js, @element, natural_width, natural_height end def to_png_base64