Home | History | Annotate | Download | only in global_error 
      1 // Copyright (c) 2012 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 "chrome/browser/ui/global_error/global_error_service.h"
      6 
      7 #include "base/memory/scoped_ptr.h"
      8 #include "chrome/browser/ui/browser.h"
      9 #include "chrome/browser/ui/global_error/global_error.h"
     10 #include "chrome/browser/ui/global_error/global_error_bubble_view_base.h"
     11 #include "chrome/browser/ui/global_error/global_error_service_factory.h"
     12 #include "chrome/test/base/in_process_browser_test.h"
     13 #include "chrome/test/base/ui_test_utils.h"
     14 
     15 namespace {
     16 
     17 // An error that has a bubble view.
     18 class BubbleViewError : public GlobalErrorWithStandardBubble {
     19  public:
     20   BubbleViewError() : bubble_view_close_count_(0) { }
     21 
     22   int bubble_view_close_count() { return bubble_view_close_count_; }
     23 
     24   virtual bool HasMenuItem() OVERRIDE { return false; }
     25   virtual int MenuItemCommandID() OVERRIDE {
     26     ADD_FAILURE();
     27     return 0;
     28   }
     29   virtual base::string16 MenuItemLabel() OVERRIDE {
     30     ADD_FAILURE();
     31     return base::string16();
     32   }
     33   virtual void ExecuteMenuItem(Browser* browser) OVERRIDE { ADD_FAILURE(); }
     34 
     35   virtual bool HasBubbleView() OVERRIDE { return true; }
     36   virtual base::string16 GetBubbleViewTitle() OVERRIDE {
     37     return base::string16();
     38   }
     39   virtual std::vector<base::string16> GetBubbleViewMessages() OVERRIDE {
     40     return std::vector<base::string16>();
     41   }
     42   virtual base::string16 GetBubbleViewAcceptButtonLabel() OVERRIDE {
     43     return base::string16();
     44   }
     45   virtual base::string16 GetBubbleViewCancelButtonLabel() OVERRIDE {
     46     return base::string16();
     47   }
     48   virtual void OnBubbleViewDidClose(Browser* browser) OVERRIDE {
     49     EXPECT_TRUE(browser);
     50     ++bubble_view_close_count_;
     51   }
     52   virtual void BubbleViewAcceptButtonPressed(Browser* browser) OVERRIDE {}
     53   virtual void BubbleViewCancelButtonPressed(Browser* browser) OVERRIDE {}
     54 
     55  private:
     56   int bubble_view_close_count_;
     57 
     58   DISALLOW_COPY_AND_ASSIGN(BubbleViewError);
     59 };
     60 
     61 } // namespace
     62 
     63 class GlobalErrorServiceBrowserTest : public InProcessBrowserTest {
     64 };
     65 
     66 // Test that showing a error with a bubble view works.
     67 IN_PROC_BROWSER_TEST_F(GlobalErrorServiceBrowserTest, ShowBubbleView) {
     68   // This will be deleted by the GlobalErrorService.
     69   BubbleViewError* error = new BubbleViewError;
     70 
     71   GlobalErrorService* service =
     72       GlobalErrorServiceFactory::GetForProfile(browser()->profile());
     73   service->AddGlobalError(error);
     74 
     75   EXPECT_EQ(error, service->GetFirstGlobalErrorWithBubbleView());
     76   EXPECT_FALSE(error->HasShownBubbleView());
     77   EXPECT_EQ(0, error->bubble_view_close_count());
     78 
     79   // Creating a second browser window should show the bubble view.
     80   CreateBrowser(browser()->profile());
     81   EXPECT_EQ(NULL, service->GetFirstGlobalErrorWithBubbleView());
     82   EXPECT_TRUE(error->HasShownBubbleView());
     83   EXPECT_EQ(0, error->bubble_view_close_count());
     84 }
     85 
     86 // Test that GlobalErrorBubbleViewBase::CloseBubbleView correctly closes the
     87 // bubble view.
     88 IN_PROC_BROWSER_TEST_F(GlobalErrorServiceBrowserTest, CloseBubbleView) {
     89   // This will be deleted by the GlobalErrorService.
     90   BubbleViewError* error = new BubbleViewError;
     91 
     92   GlobalErrorService* service =
     93       GlobalErrorServiceFactory::GetForProfile(browser()->profile());
     94   service->AddGlobalError(error);
     95 
     96   EXPECT_EQ(error, service->GetFirstGlobalErrorWithBubbleView());
     97   EXPECT_FALSE(error->HasShownBubbleView());
     98   EXPECT_EQ(0, error->bubble_view_close_count());
     99 
    100   // Creating a second browser window should show the bubble view.
    101   CreateBrowser(browser()->profile());
    102   EXPECT_EQ(NULL, service->GetFirstGlobalErrorWithBubbleView());
    103   EXPECT_TRUE(error->HasShownBubbleView());
    104   EXPECT_EQ(0, error->bubble_view_close_count());
    105 
    106   // Explicitly close the bubble view.
    107   EXPECT_TRUE(error->GetBubbleView());
    108   error->GetBubbleView()->CloseBubbleView();
    109   content::RunAllPendingInMessageLoop();
    110   EXPECT_EQ(1, error->bubble_view_close_count());
    111 }
    112 
    113 // Test that bubble is silently dismissed if it is showing when the GlobalError
    114 // instance is removed from the profile.
    115 #if defined(OS_WIN)
    116 #define MAYBE_BubbleViewDismissedOnRemove DISABLED_BubbleViewDismissedOnRemove
    117 #else
    118 #define MAYBE_BubbleViewDismissedOnRemove BubbleViewDismissedOnRemove
    119 #endif
    120 IN_PROC_BROWSER_TEST_F(GlobalErrorServiceBrowserTest,
    121                        MAYBE_BubbleViewDismissedOnRemove) {
    122   scoped_ptr<BubbleViewError> error(new BubbleViewError);
    123 
    124   GlobalErrorService* service =
    125       GlobalErrorServiceFactory::GetForProfile(browser()->profile());
    126   service->AddGlobalError(error.get());
    127 
    128   EXPECT_EQ(error.get(), service->GetFirstGlobalErrorWithBubbleView());
    129   error->ShowBubbleView(browser());
    130   content::RunAllPendingInMessageLoop();
    131   EXPECT_TRUE(error->HasShownBubbleView());
    132   EXPECT_EQ(0, error->bubble_view_close_count());
    133 
    134   // Removing |error| from profile should dismiss the bubble view without
    135   // calling |error->BubbleViewDidClose|.
    136   service->RemoveGlobalError(error.get());
    137   content::RunAllPendingInMessageLoop();
    138   EXPECT_EQ(1, error->bubble_view_close_count());
    139   // |error| is no longer owned by service and will be deleted.
    140 }
    141