| OLD | NEW |
| (Empty) |
| 1 // Copyright 2013 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 "base/command_line.h" | |
| 6 #include "base/compiler_specific.h" | |
| 7 #include "base/logging.h" | |
| 8 #include "base/memory/scoped_ptr.h" | |
| 9 #include "chrome/browser/chromeos/login/captive_portal_window_proxy.h" | |
| 10 #include "chrome/browser/chromeos/login/login_display_host_impl.h" | |
| 11 #include "chrome/browser/chromeos/login/login_manager_test.h" | |
| 12 #include "chrome/browser/chromeos/login/startup_utils.h" | |
| 13 #include "chrome/browser/chromeos/login/test/oobe_screen_waiter.h" | |
| 14 #include "chrome/browser/chromeos/login/webui_login_view.h" | |
| 15 #include "chrome/browser/chromeos/net/network_portal_detector.h" | |
| 16 #include "chrome/browser/chromeos/net/network_portal_detector_test_impl.h" | |
| 17 #include "chrome/test/base/in_process_browser_test.h" | |
| 18 #include "chromeos/chromeos_switches.h" | |
| 19 | |
| 20 namespace chromeos { | |
| 21 | |
| 22 namespace { | |
| 23 | |
| 24 const char kStubEthernetServicePath[] = "eth1"; | |
| 25 | |
| 26 // Stub implementation of CaptivePortalWindowProxyDelegate, does | |
| 27 // nothing and used to instantiate CaptivePortalWindowProxy. | |
| 28 class CaptivePortalWindowProxyStubDelegate | |
| 29 : public CaptivePortalWindowProxyDelegate { | |
| 30 public: | |
| 31 CaptivePortalWindowProxyStubDelegate(): num_portal_notifications_(0) { | |
| 32 } | |
| 33 | |
| 34 virtual ~CaptivePortalWindowProxyStubDelegate() { | |
| 35 } | |
| 36 | |
| 37 virtual void OnPortalDetected() OVERRIDE { | |
| 38 ++num_portal_notifications_; | |
| 39 } | |
| 40 | |
| 41 int num_portal_notifications() const { return num_portal_notifications_; } | |
| 42 | |
| 43 private: | |
| 44 int num_portal_notifications_; | |
| 45 }; | |
| 46 | |
| 47 } // namespace | |
| 48 | |
| 49 class CaptivePortalWindowTest : public InProcessBrowserTest { | |
| 50 protected: | |
| 51 void ShowIfRedirected() { | |
| 52 captive_portal_window_proxy_->ShowIfRedirected(); | |
| 53 } | |
| 54 | |
| 55 void Show() { | |
| 56 captive_portal_window_proxy_->Show(); | |
| 57 } | |
| 58 | |
| 59 void Close() { | |
| 60 captive_portal_window_proxy_->Close(); | |
| 61 } | |
| 62 | |
| 63 void OnRedirected() { | |
| 64 captive_portal_window_proxy_->OnRedirected(); | |
| 65 } | |
| 66 | |
| 67 void OnOriginalURLLoaded() { | |
| 68 captive_portal_window_proxy_->OnOriginalURLLoaded(); | |
| 69 } | |
| 70 | |
| 71 void CheckState(bool is_shown, int num_portal_notifications) { | |
| 72 bool actual_is_shown = (CaptivePortalWindowProxy::STATE_DISPLAYED == | |
| 73 captive_portal_window_proxy_->GetState()); | |
| 74 ASSERT_EQ(is_shown, actual_is_shown); | |
| 75 ASSERT_EQ(num_portal_notifications, delegate_.num_portal_notifications()); | |
| 76 } | |
| 77 | |
| 78 virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { | |
| 79 command_line->AppendSwitch(chromeos::switches::kForceLoginManagerInTests); | |
| 80 command_line->AppendSwitch(chromeos::switches::kLoginManager); | |
| 81 } | |
| 82 | |
| 83 virtual void SetUpOnMainThread() OVERRIDE { | |
| 84 host_ = LoginDisplayHostImpl::default_host(); | |
| 85 CHECK(host_); | |
| 86 content::WebContents* web_contents = | |
| 87 LoginDisplayHostImpl::default_host()->GetWebUILoginView()-> | |
| 88 GetWebContents(); | |
| 89 captive_portal_window_proxy_.reset( | |
| 90 new CaptivePortalWindowProxy(&delegate_, web_contents)); | |
| 91 } | |
| 92 | |
| 93 virtual void CleanUpOnMainThread() OVERRIDE { | |
| 94 captive_portal_window_proxy_.reset(); | |
| 95 base::MessageLoopForUI::current()->DeleteSoon(FROM_HERE, host_); | |
| 96 base::MessageLoopForUI::current()->RunUntilIdle(); | |
| 97 } | |
| 98 | |
| 99 private: | |
| 100 scoped_ptr<CaptivePortalWindowProxy> captive_portal_window_proxy_; | |
| 101 CaptivePortalWindowProxyStubDelegate delegate_; | |
| 102 | |
| 103 LoginDisplayHost* host_; | |
| 104 }; | |
| 105 | |
| 106 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, Show) { | |
| 107 Show(); | |
| 108 } | |
| 109 | |
| 110 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, ShowClose) { | |
| 111 CheckState(false, 0); | |
| 112 | |
| 113 Show(); | |
| 114 CheckState(true, 0); | |
| 115 | |
| 116 Close(); | |
| 117 CheckState(false, 0); | |
| 118 } | |
| 119 | |
| 120 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, OnRedirected) { | |
| 121 CheckState(false, 0); | |
| 122 | |
| 123 ShowIfRedirected(); | |
| 124 CheckState(false, 0); | |
| 125 | |
| 126 OnRedirected(); | |
| 127 CheckState(true, 1); | |
| 128 | |
| 129 Close(); | |
| 130 CheckState(false, 1); | |
| 131 } | |
| 132 | |
| 133 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, OnOriginalURLLoaded) { | |
| 134 CheckState(false, 0); | |
| 135 | |
| 136 ShowIfRedirected(); | |
| 137 CheckState(false, 0); | |
| 138 | |
| 139 OnRedirected(); | |
| 140 CheckState(true, 1); | |
| 141 | |
| 142 OnOriginalURLLoaded(); | |
| 143 CheckState(false, 1); | |
| 144 } | |
| 145 | |
| 146 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, MultipleCalls) { | |
| 147 CheckState(false, 0); | |
| 148 | |
| 149 ShowIfRedirected(); | |
| 150 CheckState(false, 0); | |
| 151 | |
| 152 Show(); | |
| 153 CheckState(true, 0); | |
| 154 | |
| 155 Close(); | |
| 156 CheckState(false, 0); | |
| 157 | |
| 158 OnRedirected(); | |
| 159 CheckState(false, 1); | |
| 160 | |
| 161 OnOriginalURLLoaded(); | |
| 162 CheckState(false, 1); | |
| 163 | |
| 164 Show(); | |
| 165 CheckState(true, 1); | |
| 166 | |
| 167 OnRedirected(); | |
| 168 CheckState(true, 2); | |
| 169 | |
| 170 Close(); | |
| 171 CheckState(false, 2); | |
| 172 | |
| 173 OnOriginalURLLoaded(); | |
| 174 CheckState(false, 2); | |
| 175 } | |
| 176 | |
| 177 class CaptivePortalWindowCtorDtorTest : public LoginManagerTest { | |
| 178 public: | |
| 179 CaptivePortalWindowCtorDtorTest() | |
| 180 : LoginManagerTest(false) {} | |
| 181 virtual ~CaptivePortalWindowCtorDtorTest() {} | |
| 182 | |
| 183 virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { | |
| 184 LoginManagerTest::SetUpInProcessBrowserTestFixture(); | |
| 185 | |
| 186 network_portal_detector_ = new NetworkPortalDetectorTestImpl(); | |
| 187 NetworkPortalDetector::InitializeForTesting(network_portal_detector_); | |
| 188 NetworkPortalDetector::CaptivePortalState portal_state; | |
| 189 portal_state.status = NetworkPortalDetector::CAPTIVE_PORTAL_STATUS_PORTAL; | |
| 190 portal_state.response_code = 200; | |
| 191 network_portal_detector_->SetDefaultNetworkPathForTesting( | |
| 192 kStubEthernetServicePath); | |
| 193 network_portal_detector_->SetDetectionResultsForTesting( | |
| 194 kStubEthernetServicePath, portal_state); | |
| 195 } | |
| 196 | |
| 197 protected: | |
| 198 NetworkPortalDetectorTestImpl* network_portal_detector() { | |
| 199 return network_portal_detector_; | |
| 200 } | |
| 201 | |
| 202 PortalDetectorStrategy::StrategyId strategy_id() { | |
| 203 return network_portal_detector_->strategy_id(); | |
| 204 } | |
| 205 | |
| 206 private: | |
| 207 NetworkPortalDetectorTestImpl* network_portal_detector_; | |
| 208 | |
| 209 DISALLOW_COPY_AND_ASSIGN(CaptivePortalWindowCtorDtorTest); | |
| 210 }; | |
| 211 | |
| 212 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowCtorDtorTest, PRE_OpenPortalDialog) { | |
| 213 StartupUtils::MarkOobeCompleted(); | |
| 214 } | |
| 215 | |
| 216 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowCtorDtorTest, OpenPortalDialog) { | |
| 217 LoginDisplayHostImpl* host = | |
| 218 static_cast<LoginDisplayHostImpl*>(LoginDisplayHostImpl::default_host()); | |
| 219 ASSERT_TRUE(host); | |
| 220 OobeUI* oobe = host->GetOobeUI(); | |
| 221 ASSERT_TRUE(oobe); | |
| 222 ErrorScreenActor* actor = oobe->GetErrorScreenActor(); | |
| 223 ASSERT_TRUE(actor); | |
| 224 | |
| 225 // Error screen asks portal detector to change detection strategy. | |
| 226 ErrorScreen error_screen(NULL, actor); | |
| 227 | |
| 228 ASSERT_EQ(PortalDetectorStrategy::STRATEGY_ID_LOGIN_SCREEN, strategy_id()); | |
| 229 network_portal_detector()->NotifyObserversForTesting(); | |
| 230 OobeScreenWaiter(OobeDisplay::SCREEN_ERROR_MESSAGE).Wait(); | |
| 231 ASSERT_EQ(PortalDetectorStrategy::STRATEGY_ID_ERROR_SCREEN, strategy_id()); | |
| 232 | |
| 233 actor->ShowCaptivePortal(); | |
| 234 } | |
| 235 | |
| 236 } // namespace chromeos | |
| OLD | NEW |