Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(179)

Side by Side Diff: headless/lib/headless_web_contents_browsertest.cc

Issue 2641073004: Revert of [headless] Fix screenshots with gpu enabled. (Closed)
Patch Set: Created 3 years, 11 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View unified diff | Download patch
« no previous file with comments | « headless/lib/DEPS ('k') | headless/test/headless_browser_test.h » ('j') | no next file with comments »
Toggle Intra-line Diffs ('i') | Expand Comments ('e') | Collapse Comments ('c') | Show Comments Hide Comments ('s')
OLDNEW
1 // Copyright 2016 The Chromium Authors. All rights reserved. 1 // Copyright 2016 The Chromium Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be 2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file. 3 // found in the LICENSE file.
4 4
5 #include <memory> 5 #include <memory>
6 #include <string> 6 #include <string>
7 #include <vector> 7 #include <vector>
8 8
9 #include "base/base64.h"
10 #include "content/public/test/browser_test.h" 9 #include "content/public/test/browser_test.h"
11 #include "headless/public/devtools/domains/page.h" 10 #include "headless/public/devtools/domains/page.h"
12 #include "headless/public/devtools/domains/runtime.h"
13 #include "headless/public/devtools/domains/security.h" 11 #include "headless/public/devtools/domains/security.h"
14 #include "headless/public/headless_browser.h" 12 #include "headless/public/headless_browser.h"
15 #include "headless/public/headless_devtools_client.h" 13 #include "headless/public/headless_devtools_client.h"
16 #include "headless/public/headless_web_contents.h" 14 #include "headless/public/headless_web_contents.h"
17 #include "headless/test/headless_browser_test.h" 15 #include "headless/test/headless_browser_test.h"
18 #include "testing/gmock/include/gmock/gmock.h" 16 #include "testing/gmock/include/gmock/gmock.h"
19 #include "testing/gtest/include/gtest/gtest.h" 17 #include "testing/gtest/include/gtest/gtest.h"
20 #include "third_party/skia/include/core/SkBitmap.h"
21 #include "third_party/skia/include/core/SkColor.h"
22 #include "ui/gfx/codec/png_codec.h"
23 #include "ui/gfx/geometry/size.h" 18 #include "ui/gfx/geometry/size.h"
24 #include "url/gurl.h" 19 #include "url/gurl.h"
25 20
26 using testing::UnorderedElementsAre; 21 using testing::UnorderedElementsAre;
27 22
28 namespace headless { 23 namespace headless {
29 24
30 class HeadlessWebContentsTest : public HeadlessBrowserTest {}; 25 class HeadlessWebContentsTest : public HeadlessBrowserTest {};
31 26
32 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) { 27 IN_PROC_BROWSER_TEST_F(HeadlessWebContentsTest, Navigation) {
(...skipping 21 matching lines...) Expand all
54 HeadlessWebContents* web_contents = 49 HeadlessWebContents* web_contents =
55 browser_context->CreateWebContentsBuilder() 50 browser_context->CreateWebContentsBuilder()
56 .SetInitialURL(embedded_test_server()->GetURL("/window_open.html")) 51 .SetInitialURL(embedded_test_server()->GetURL("/window_open.html"))
57 .Build(); 52 .Build();
58 EXPECT_TRUE(WaitForLoad(web_contents)); 53 EXPECT_TRUE(WaitForLoad(web_contents));
59 54
60 EXPECT_EQ(static_cast<size_t>(2), 55 EXPECT_EQ(static_cast<size_t>(2),
61 browser_context->GetAllWebContents().size()); 56 browser_context->GetAllWebContents().size());
62 } 57 }
63 58
64 namespace {
65 bool DecodePNG(std::string base64_data, SkBitmap* bitmap) {
66 std::string png_data;
67 if (!base::Base64Decode(base64_data, &png_data))
68 return false;
69 return gfx::PNGCodec::Decode(
70 reinterpret_cast<unsigned const char*>(png_data.data()), png_data.size(),
71 bitmap);
72 }
73 } // namespace
74
75 // Parameter specifies whether --disable-gpu should be used.
76 class HeadlessWebContentsScreenshotTest 59 class HeadlessWebContentsScreenshotTest
77 : public HeadlessAsyncDevTooledBrowserTest, 60 : public HeadlessAsyncDevTooledBrowserTest {
78 public ::testing::WithParamInterface<bool> {
79 public: 61 public:
80 void SetUp() override {
81 EnablePixelOutput();
82 if (GetParam())
83 UseSoftwareCompositing();
84 HeadlessAsyncDevTooledBrowserTest::SetUp();
85 }
86
87 void RunDevTooledTest() override { 62 void RunDevTooledTest() override {
88 std::unique_ptr<runtime::EvaluateParams> params =
89 runtime::EvaluateParams::Builder()
90 .SetExpression("document.body.style.background = '#0000ff'")
91 .Build();
92 devtools_client_->GetRuntime()->Evaluate(
93 std::move(params),
94 base::Bind(&HeadlessWebContentsScreenshotTest::OnPageSetupCompleted,
95 base::Unretained(this)));
96 }
97
98 void OnPageSetupCompleted(std::unique_ptr<runtime::EvaluateResult> result) {
99 devtools_client_->GetPage()->GetExperimental()->CaptureScreenshot( 63 devtools_client_->GetPage()->GetExperimental()->CaptureScreenshot(
100 page::CaptureScreenshotParams::Builder().Build(), 64 page::CaptureScreenshotParams::Builder().Build(),
101 base::Bind(&HeadlessWebContentsScreenshotTest::OnScreenshotCaptured, 65 base::Bind(&HeadlessWebContentsScreenshotTest::OnScreenshotCaptured,
102 base::Unretained(this))); 66 base::Unretained(this)));
103 } 67 }
104 68
105 void OnScreenshotCaptured( 69 void OnScreenshotCaptured(
106 std::unique_ptr<page::CaptureScreenshotResult> result) { 70 std::unique_ptr<page::CaptureScreenshotResult> result) {
107 std::string base64 = result->GetData(); 71 EXPECT_LT(0U, result->GetData().length());
108 EXPECT_LT(0U, base64.length());
109 SkBitmap result_bitmap;
110 EXPECT_TRUE(DecodePNG(base64, &result_bitmap));
111
112 EXPECT_EQ(800, result_bitmap.width());
113 EXPECT_EQ(600, result_bitmap.height());
114 SkColor actual_color = result_bitmap.getColor(400, 300);
115 SkColor expected_color = SkColorSetRGB(0x00, 0x00, 0xff);
116 EXPECT_EQ(expected_color, actual_color);
117 FinishAsynchronousTest(); 72 FinishAsynchronousTest();
118 } 73 }
119 }; 74 };
120 75
121 HEADLESS_ASYNC_DEVTOOLED_TEST_P(HeadlessWebContentsScreenshotTest); 76 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsScreenshotTest);
122
123 // Instantiate test case for both software and gpu compositing modes.
124 INSTANTIATE_TEST_CASE_P(HeadlessWebContentsScreenshotTests,
125 HeadlessWebContentsScreenshotTest,
126 ::testing::Bool());
127 77
128 class HeadlessWebContentsSecurityTest 78 class HeadlessWebContentsSecurityTest
129 : public HeadlessAsyncDevTooledBrowserTest, 79 : public HeadlessAsyncDevTooledBrowserTest,
130 public security::ExperimentalObserver { 80 public security::ExperimentalObserver {
131 public: 81 public:
132 void RunDevTooledTest() override { 82 void RunDevTooledTest() override {
133 devtools_client_->GetSecurity()->GetExperimental()->AddObserver(this); 83 devtools_client_->GetSecurity()->GetExperimental()->AddObserver(this);
134 devtools_client_->GetSecurity()->GetExperimental()->Enable( 84 devtools_client_->GetSecurity()->GetExperimental()->Enable(
135 security::EnableParams::Builder().Build()); 85 security::EnableParams::Builder().Build());
136 } 86 }
137 87
138 void OnSecurityStateChanged( 88 void OnSecurityStateChanged(
139 const security::SecurityStateChangedParams& params) override { 89 const security::SecurityStateChangedParams& params) override {
140 EXPECT_EQ(security::SecurityState::NEUTRAL, params.GetSecurityState()); 90 EXPECT_EQ(security::SecurityState::NEUTRAL, params.GetSecurityState());
141 91
142 FinishAsynchronousTest(); 92 FinishAsynchronousTest();
143 } 93 }
144 }; 94 };
145 95
146 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsSecurityTest); 96 HEADLESS_ASYNC_DEVTOOLED_TEST_F(HeadlessWebContentsSecurityTest);
147 97
148 } // namespace headless 98 } // namespace headless
OLDNEW
« no previous file with comments | « headless/lib/DEPS ('k') | headless/test/headless_browser_test.h » ('j') | no next file with comments »

Powered by Google App Engine
This is Rietveld 408576698