| OLD | NEW |
| 1 // Copyright (c) 2013 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2013 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 "base/path_service.h" | 5 #include "base/path_service.h" |
| 6 #include "base/run_loop.h" | 6 #include "base/run_loop.h" |
| 7 #include "content/public/browser/render_view_host.h" | 7 #include "content/public/browser/render_view_host.h" |
| 8 #include "content/public/browser/web_contents.h" | 8 #include "content/public/browser/web_contents.h" |
| 9 #include "content/public/common/content_paths.h" | 9 #include "content/public/common/content_paths.h" |
| 10 #include "content/public/test/content_browser_test.h" | 10 #include "content/public/test/content_browser_test.h" |
| 11 #include "content/public/test/content_browser_test_utils.h" | 11 #include "content/public/test/content_browser_test_utils.h" |
| 12 #include "content/shell/browser/shell.h" | 12 #include "content/shell/browser/shell.h" |
| 13 #include "net/base/filename_util.h" | 13 #include "net/base/filename_util.h" |
| 14 #include "third_party/skia/include/core/SkBitmap.h" | 14 #include "third_party/skia/include/core/SkBitmap.h" |
| 15 | 15 |
| 16 namespace content { | 16 namespace content { |
| 17 | 17 |
| 18 class RenderWidgetHostBrowserTest : public ContentBrowserTest { | 18 class RenderWidgetHostBrowserTest : public ContentBrowserTest { |
| 19 public: | 19 public: |
| 20 RenderWidgetHostBrowserTest() {} | 20 RenderWidgetHostBrowserTest() {} |
| 21 | 21 |
| 22 virtual void SetUpOnMainThread() OVERRIDE { | 22 virtual void SetUpOnMainThread() OVERRIDE { |
| 23 ASSERT_TRUE(PathService::Get(DIR_TEST_DATA, &test_dir_)); | 23 ASSERT_TRUE(PathService::Get(DIR_TEST_DATA, &test_dir_)); |
| 24 } | 24 } |
| 25 | 25 |
| 26 void GetSnapshotFromRendererCallback(const base::Closure& quit_closure, | |
| 27 bool* snapshot_valid, | |
| 28 bool success, | |
| 29 const SkBitmap& bitmap) { | |
| 30 quit_closure.Run(); | |
| 31 EXPECT_EQ(success, true); | |
| 32 | |
| 33 const int row_bytes = bitmap.rowBytesAsPixels(); | |
| 34 SkColor* pixels = reinterpret_cast<SkColor*>(bitmap.getPixels()); | |
| 35 for (int i = 0; i < bitmap.width(); ++i) { | |
| 36 for (int j = 0; j < bitmap.height(); ++j) { | |
| 37 if (pixels[j * row_bytes + i] != SK_ColorRED) { | |
| 38 return; | |
| 39 } | |
| 40 } | |
| 41 } | |
| 42 *snapshot_valid = true; | |
| 43 } | |
| 44 | |
| 45 protected: | 26 protected: |
| 46 base::FilePath test_dir_; | 27 base::FilePath test_dir_; |
| 47 }; | 28 }; |
| 48 | 29 |
| 49 // Disabled on Aura since this is not possible with ubercomp. | |
| 50 #if defined(USE_AURA) | |
| 51 #define MAYBE_GetSnapshotFromRendererTest DISABLED_GetSnapshotFromRendererTest | |
| 52 #else | |
| 53 #define MAYBE_GetSnapshotFromRendererTest GetSnapshotFromRendererTest | |
| 54 #endif | |
| 55 IN_PROC_BROWSER_TEST_F(RenderWidgetHostBrowserTest, | |
| 56 MAYBE_GetSnapshotFromRendererTest) { | |
| 57 base::RunLoop run_loop; | |
| 58 | |
| 59 NavigateToURL(shell(), GURL(net::FilePathToFileURL( | |
| 60 test_dir_.AppendASCII("rwh_simple.html")))); | |
| 61 | |
| 62 bool snapshot_valid = false; | |
| 63 RenderViewHost* const rwh = shell()->web_contents()->GetRenderViewHost(); | |
| 64 rwh->GetSnapshotFromRenderer(gfx::Rect(), base::Bind( | |
| 65 &RenderWidgetHostBrowserTest::GetSnapshotFromRendererCallback, | |
| 66 base::Unretained(this), | |
| 67 run_loop.QuitClosure(), | |
| 68 &snapshot_valid)); | |
| 69 run_loop.Run(); | |
| 70 | |
| 71 EXPECT_EQ(snapshot_valid, true); | |
| 72 } | |
| 73 | |
| 74 } // namespace content | 30 } // namespace content |
| OLD | NEW |