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:
BubbleViewError()20 BubbleViewError() : bubble_view_close_count_(0) { }
21
bubble_view_close_count()22 int bubble_view_close_count() { return bubble_view_close_count_; }
23
HasMenuItem()24 virtual bool HasMenuItem() OVERRIDE { return false; }
MenuItemCommandID()25 virtual int MenuItemCommandID() OVERRIDE {
26 ADD_FAILURE();
27 return 0;
28 }
MenuItemLabel()29 virtual base::string16 MenuItemLabel() OVERRIDE {
30 ADD_FAILURE();
31 return base::string16();
32 }
ExecuteMenuItem(Browser * browser)33 virtual void ExecuteMenuItem(Browser* browser) OVERRIDE { ADD_FAILURE(); }
34
HasBubbleView()35 virtual bool HasBubbleView() OVERRIDE { return true; }
GetBubbleViewTitle()36 virtual base::string16 GetBubbleViewTitle() OVERRIDE {
37 return base::string16();
38 }
GetBubbleViewMessages()39 virtual std::vector<base::string16> GetBubbleViewMessages() OVERRIDE {
40 return std::vector<base::string16>();
41 }
GetBubbleViewAcceptButtonLabel()42 virtual base::string16 GetBubbleViewAcceptButtonLabel() OVERRIDE {
43 return base::string16();
44 }
GetBubbleViewCancelButtonLabel()45 virtual base::string16 GetBubbleViewCancelButtonLabel() OVERRIDE {
46 return base::string16();
47 }
OnBubbleViewDidClose(Browser * browser)48 virtual void OnBubbleViewDidClose(Browser* browser) OVERRIDE {
49 EXPECT_TRUE(browser);
50 ++bubble_view_close_count_;
51 }
BubbleViewAcceptButtonPressed(Browser * browser)52 virtual void BubbleViewAcceptButtonPressed(Browser* browser) OVERRIDE {}
BubbleViewCancelButtonPressed(Browser * browser)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.
IN_PROC_BROWSER_TEST_F(GlobalErrorServiceBrowserTest,ShowBubbleView)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.
IN_PROC_BROWSER_TEST_F(GlobalErrorServiceBrowserTest,CloseBubbleView)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
IN_PROC_BROWSER_TEST_F(GlobalErrorServiceBrowserTest,MAYBE_BubbleViewDismissedOnRemove)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