Index: chrome/test/chromedriver/window_commands.cc |
diff --git a/chrome/test/chromedriver/window_commands.cc b/chrome/test/chromedriver/window_commands.cc |
index c1971098c77cc7fd8c20ddc3a9ee3395ccae2185..997290e842edec00b49861ae58435d400c4ee5e0 100644 |
--- a/chrome/test/chromedriver/window_commands.cc |
+++ b/chrome/test/chromedriver/window_commands.cc |
@@ -739,16 +739,16 @@ Status ExecuteScreenshot( |
if (session->chrome->GetType() == Chrome::DESKTOP && |
!session->force_devtools_screenshot) { |
AutomationExtension* extension = NULL; |
- Status status = session->chrome->GetAutomationExtension(&extension); |
+ status = session->chrome->GetAutomationExtension(&extension); |
if (status.IsError()) |
return status; |
status = extension->CaptureScreenshot(&screenshot); |
// If the screenshot was forbidden, fallback to DevTools. |
- if (status.code() != kForbidden) |
- return status; |
+ if (status.code() == kForbidden) |
+ status = web_view->CaptureScreenshot(&screenshot); |
+ } else { |
+ status = web_view->CaptureScreenshot(&screenshot); |
} |
- |
- status = web_view->CaptureScreenshot(&screenshot); |
if (status.IsError()) |
return status; |