src/content/browser/web_contents/aura/overscroll_navigation_overlay_unittest.cc

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "content/browser/web_contents/aura/overscroll_navigation_overlay.h"
   6
   7 #include "content/browser/frame_host/navigation_entry_impl.h"
   8 #include "content/browser/web_contents/aura/image_window_delegate.h"
   9 #include "content/browser/web_contents/web_contents_view.h"
  10 #include "content/common/frame_messages.h"
  11 #include "content/common/view_messages.h"
  12 #include "content/public/test/mock_render_process_host.h"
  13 #include "content/test/test_render_frame_host.h"
  14 #include "content/test/test_render_view_host.h"
  15 #include "content/test/test_web_contents.h"
  16 #include "ui/aura/test/test_windows.h"
  17 #include "ui/aura/window.h"
  18 #include "ui/gfx/codec/png_codec.h"
  19
  20 namespace content {
  21
  22 class OverscrollNavigationOverlayTest : public RenderViewHostImplTestHarness {
  23  public:
  24   OverscrollNavigationOverlayTest() {}
  25   virtual ~OverscrollNavigationOverlayTest() {}
  26
  27   gfx::Image CreateDummyScreenshot() {
  28     SkBitmap bitmap;
  29     bitmap.allocN32Pixels(1, 1);
  30     bitmap.eraseColor(SK_ColorWHITE);
  31     return gfx::Image::CreateFrom1xBitmap(bitmap);
  32   }
  33
  34   void SetDummyScreenshotOnNavEntry(NavigationEntry* entry) {
  35     SkBitmap bitmap;
  36     bitmap.allocN32Pixels(1, 1);
  37     bitmap.eraseColor(SK_ColorWHITE);
  38     std::vector<unsigned char> png_data;
  39     gfx::PNGCodec::EncodeBGRASkBitmap(bitmap, true, &png_data);
  40     scoped_refptr<base::RefCountedBytes> png_bytes =
  41         base::RefCountedBytes::TakeVector(&png_data);
  42     NavigationEntryImpl* entry_impl =
  43         NavigationEntryImpl::FromNavigationEntry(entry);
  44     entry_impl->SetScreenshotPNGData(png_bytes);
  45   }
  46
  47   void ReceivePaintUpdate() {
  48     FrameHostMsg_DidFirstVisuallyNonEmptyPaint msg(
  49         main_test_rfh()->GetRoutingID());
  50     RenderViewHostTester::TestOnMessageReceived(test_rvh(), msg);
  51   }
  52
  53   void PerformBackNavigationViaSliderCallbacks() {
  54     // Sets slide direction to SLIDE_BACK, sets screenshot from NavEntry at
  55     // offset -1  on layer_delegate_.
  56     delete GetOverlay()->CreateBackLayer();
  57     // Performs BACK navigation, sets image from layer_delegate_ on
  58     // image_delegate_.
  59     GetOverlay()->OnWindowSlideCompleting();
  60     GetOverlay()->OnWindowSlideCompleted(scoped_ptr<ui::Layer>());
  61   }
  62
  63  protected:
  64   // RenderViewHostImplTestHarness:
  65   virtual void SetUp() OVERRIDE {
  66     RenderViewHostImplTestHarness::SetUp();
  67
  68     const GURL first("https://www.google.com");
  69     contents()->NavigateAndCommit(first);
  70     EXPECT_TRUE(controller().GetVisibleEntry());
  71     EXPECT_FALSE(controller().CanGoBack());
  72
  73     const GURL second("http://www.chromium.org");
  74     contents()->NavigateAndCommit(second);
  75     EXPECT_TRUE(controller().CanGoBack());
  76
  77     // Receive a paint update. This is necessary to make sure the size is set
  78     // correctly in RenderWidgetHostImpl.
  79     ViewHostMsg_UpdateRect_Params params;
  80     memset(&params, 0, sizeof(params));
  81     params.view_size = gfx::Size(10, 10);
  82     ViewHostMsg_UpdateRect rect(test_rvh()->GetRoutingID(), params);
  83     RenderViewHostTester::TestOnMessageReceived(test_rvh(), rect);
  84
  85     // Reset pending flags for size/paint.
  86     test_rvh()->ResetSizeAndRepaintPendingFlags();
  87
  88     // Create the overlay, and set the contents of the overlay window.
  89     overlay_.reset(new OverscrollNavigationOverlay(contents()));
  90     ImageWindowDelegate* image_delegate = new ImageWindowDelegate();
  91     scoped_ptr<aura::Window> overlay_window(
  92       aura::test::CreateTestWindowWithDelegate(
  93           image_delegate,
  94           0,
  95           gfx::Rect(root_window()->bounds().size()),
  96           root_window()));
  97
  98     overlay_->SetOverlayWindow(overlay_window.Pass(), image_delegate);
  99     overlay_->StartObserving();
 100
 101     EXPECT_TRUE(overlay_->web_contents());
 102     EXPECT_FALSE(overlay_->loading_complete_);
 103     EXPECT_FALSE(overlay_->received_paint_update_);
 104   }
 105
 106   virtual void TearDown() OVERRIDE {
 107     overlay_.reset();
 108     RenderViewHostImplTestHarness::TearDown();
 109   }
 110
 111   OverscrollNavigationOverlay* GetOverlay() {
 112     return overlay_.get();
 113   }
 114
 115  private:
 116   scoped_ptr<OverscrollNavigationOverlay> overlay_;
 117
 118   DISALLOW_COPY_AND_ASSIGN(OverscrollNavigationOverlayTest);
 119 };
 120
 121 TEST_F(OverscrollNavigationOverlayTest, FirstVisuallyNonEmptyPaint_NoImage) {
 122   ReceivePaintUpdate();
 123   EXPECT_TRUE(GetOverlay()->received_paint_update_);
 124   EXPECT_FALSE(GetOverlay()->loading_complete_);
 125   // The paint update will hide the overlay.
 126   EXPECT_FALSE(GetOverlay()->web_contents());
 127 }
 128
 129 TEST_F(OverscrollNavigationOverlayTest, FirstVisuallyNonEmptyPaint_WithImage) {
 130   GetOverlay()->image_delegate_->SetImage(CreateDummyScreenshot());
 131
 132   ReceivePaintUpdate();
 133   EXPECT_TRUE(GetOverlay()->received_paint_update_);
 134   EXPECT_FALSE(GetOverlay()->loading_complete_);
 135   // The paint update will hide the overlay.
 136   EXPECT_FALSE(GetOverlay()->web_contents());
 137 }
 138
 139 TEST_F(OverscrollNavigationOverlayTest, LoadUpdateWithoutNonEmptyPaint) {
 140   GetOverlay()->image_delegate_->SetImage(CreateDummyScreenshot());
 141   process()->sink().ClearMessages();
 142
 143   contents()->TestSetIsLoading(false);
 144   EXPECT_TRUE(GetOverlay()->loading_complete_);
 145   EXPECT_FALSE(GetOverlay()->received_paint_update_);
 146   // The page load should hide the overlay.
 147   EXPECT_FALSE(GetOverlay()->web_contents());
 148 }
 149
 150 TEST_F(OverscrollNavigationOverlayTest, MultiNavigation_PaintUpdate) {
 151   GetOverlay()->image_delegate_->SetImage(CreateDummyScreenshot());
 152   SetDummyScreenshotOnNavEntry(controller().GetEntryAtOffset(-1));
 153
 154   PerformBackNavigationViaSliderCallbacks();
 155   // Screenshot was set on NavEntry at offset -1.
 156   EXPECT_TRUE(GetOverlay()->image_delegate_->has_image());
 157   EXPECT_FALSE(GetOverlay()->received_paint_update_);
 158
 159   ReceivePaintUpdate();
 160   // Paint updates until the navigation is committed represent updates
 161   // for the previous page, so they shouldn't affect the flag.
 162   EXPECT_FALSE(GetOverlay()->received_paint_update_);
 163
 164   contents()->CommitPendingNavigation();
 165   ReceivePaintUpdate();
 166   // Navigation was committed and the paint update was received - the flag
 167   // should now be updated.
 168   EXPECT_TRUE(GetOverlay()->received_paint_update_);
 169
 170   EXPECT_FALSE(GetOverlay()->web_contents());
 171 }
 172
 173 TEST_F(OverscrollNavigationOverlayTest, MultiNavigation_LoadingUpdate) {
 174   GetOverlay()->image_delegate_->SetImage(CreateDummyScreenshot());
 175
 176   PerformBackNavigationViaSliderCallbacks();
 177   // No screenshot was set on NavEntry at offset -1.
 178   EXPECT_FALSE(GetOverlay()->image_delegate_->has_image());
 179   // Navigation was started, so the loading status flag should be reset.
 180   EXPECT_FALSE(GetOverlay()->loading_complete_);
 181
 182   // Load updates until the navigation is committed represent updates for the
 183   // previous page, so they shouldn't affect the flag.
 184   contents()->TestSetIsLoading(true);
 185   contents()->TestSetIsLoading(false);
 186   EXPECT_FALSE(GetOverlay()->loading_complete_);
 187
 188   contents()->CommitPendingNavigation();
 189   contents()->TestSetIsLoading(true);
 190   contents()->TestSetIsLoading(false);
 191   // Navigation was committed and the load update was received - the flag
 192   // should now be updated.
 193   EXPECT_TRUE(GetOverlay()->loading_complete_);
 194
 195   EXPECT_FALSE(GetOverlay()->web_contents());
 196 }
 197
 198 }  // namespace content