| // Copyright (c) 2013 The Chromium Authors. All rights reserved. |
| // Use of this source code is governed by a BSD-style license that can be |
| // found in the LICENSE file. |
| |
| #include "base/command_line.h" |
| #include "base/path_service.h" |
| #include "chrome/browser/chrome_browser_main.h" |
| #include "chrome/browser/chrome_browser_main_extra_parts.h" |
| #include "chrome/browser/chrome_content_browser_client.h" |
| #include "chrome/browser/chrome_notification_types.h" |
| #include "chrome/browser/chromeos/cros/cros_in_process_browser_test.h" |
| #include "chrome/browser/chromeos/login/existing_user_controller.h" |
| #include "chrome/browser/chromeos/login/webui_login_display.h" |
| #include "chrome/browser/chromeos/login/wizard_controller.h" |
| #include "chrome/browser/ui/browser.h" |
| #include "chrome/common/chrome_paths.h" |
| #include "chrome/common/chrome_switches.h" |
| #include "chrome/test/base/in_process_browser_test.h" |
| #include "chrome/test/base/interactive_test_utils.h" |
| #include "chrome/test/base/ui_test_utils.h" |
| #include "chromeos/chromeos_switches.h" |
| #include "content/public/browser/notification_observer.h" |
| #include "content/public/browser/notification_registrar.h" |
| #include "content/public/browser/notification_service.h" |
| #include "content/public/test/test_utils.h" |
| #include "google_apis/gaia/gaia_switches.h" |
| #include "net/test/embedded_test_server/embedded_test_server.h" |
| #include "net/test/embedded_test_server/http_request.h" |
| #include "net/test/embedded_test_server/http_response.h" |
| #include "testing/gmock/include/gmock/gmock.h" |
| #include "testing/gtest/include/gtest/gtest.h" |
| |
| using namespace net::test_server; |
| |
| namespace { |
| |
| // Used to add an observer to NotificationService after it's created. |
| class TestBrowserMainExtraParts |
| : public ChromeBrowserMainExtraParts, |
| public content::NotificationObserver { |
| public: |
| TestBrowserMainExtraParts() |
| : webui_visible_(false), |
| browsing_data_removed_(false), |
| signin_screen_shown_(false) {} |
| virtual ~TestBrowserMainExtraParts() {} |
| |
| // ChromeBrowserMainExtraParts implementation. |
| virtual void PreEarlyInitialization() OVERRIDE { |
| registrar_.Add(this, chrome::NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE, |
| content::NotificationService::AllSources()); |
| registrar_.Add(this, chrome::NOTIFICATION_SESSION_STARTED, |
| content::NotificationService::AllSources()); |
| registrar_.Add(this, chrome::NOTIFICATION_BROWSING_DATA_REMOVED, |
| content::NotificationService::AllSources()); |
| } |
| |
| void set_quit_task(const base::Closure& quit_task) { quit_task_ = quit_task; } |
| |
| private: |
| // Overridden from content::NotificationObserver: |
| virtual void Observe(int type, |
| const content::NotificationSource& source, |
| const content::NotificationDetails& details) OVERRIDE { |
| if (type == chrome::NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE) { |
| LOG(INFO) << "NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE"; |
| webui_visible_ = true; |
| if (browsing_data_removed_ && !signin_screen_shown_) { |
| signin_screen_shown_ = true; |
| ShowSigninScreen(); |
| } |
| } else if (type == chrome::NOTIFICATION_BROWSING_DATA_REMOVED) { |
| LOG(INFO) << "chrome::NOTIFICATION_BROWSING_DATA_REMOVED"; |
| browsing_data_removed_ = true; |
| if (webui_visible_ && !signin_screen_shown_) { |
| signin_screen_shown_ = true; |
| ShowSigninScreen(); |
| } |
| } else if (type == chrome::NOTIFICATION_SESSION_STARTED) { |
| LOG(INFO) << "chrome::NOTIFICATION_SESSION_STARTED"; |
| quit_task_.Run(); |
| } else { |
| NOTREACHED(); |
| } |
| } |
| |
| void ShowSigninScreen() { |
| chromeos::ExistingUserController* controller = |
| chromeos::ExistingUserController::current_controller(); |
| CHECK(controller); |
| chromeos::WebUILoginDisplay* webui_login_display = |
| static_cast<chromeos::WebUILoginDisplay*>( |
| controller->login_display()); |
| CHECK(webui_login_display); |
| webui_login_display->ShowSigninScreenForCreds("username", "password"); |
| // TODO(glotov): mock GAIA server (test_server_) should support |
| // username/password configuration. |
| } |
| |
| bool webui_visible_, browsing_data_removed_, signin_screen_shown_; |
| content::NotificationRegistrar registrar_; |
| base::Closure quit_task_; |
| GURL gaia_url_; |
| |
| DISALLOW_COPY_AND_ASSIGN(TestBrowserMainExtraParts); |
| }; |
| |
| class TestContentBrowserClient : public chrome::ChromeContentBrowserClient { |
| public: |
| TestContentBrowserClient() {} |
| virtual ~TestContentBrowserClient() {} |
| |
| virtual content::BrowserMainParts* CreateBrowserMainParts( |
| const content::MainFunctionParams& parameters) OVERRIDE { |
| ChromeBrowserMainParts* main_parts = static_cast<ChromeBrowserMainParts*>( |
| ChromeContentBrowserClient::CreateBrowserMainParts(parameters)); |
| |
| browser_main_extra_parts_ = new TestBrowserMainExtraParts(); |
| main_parts->AddParts(browser_main_extra_parts_); |
| return main_parts; |
| } |
| |
| TestBrowserMainExtraParts* browser_main_extra_parts_; |
| |
| private: |
| DISALLOW_COPY_AND_ASSIGN(TestContentBrowserClient); |
| }; |
| |
| const base::FilePath kServiceLogin("chromeos/service_login.html"); |
| |
| class OobeTest : public chromeos::CrosInProcessBrowserTest { |
| protected: |
| virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE { |
| command_line->AppendSwitch(chromeos::switches::kLoginManager); |
| command_line->AppendSwitch(chromeos::switches::kForceLoginManagerInTests); |
| command_line->AppendSwitch( |
| chromeos::switches::kDisableChromeCaptivePortalDetector); |
| command_line->AppendSwitchASCII(chromeos::switches::kLoginProfile, "user"); |
| command_line->AppendSwitchASCII( |
| chromeos::switches::kAuthExtensionPath, "gaia_auth"); |
| } |
| |
| virtual void SetUpInProcessBrowserTestFixture() OVERRIDE { |
| content_browser_client_.reset(new TestContentBrowserClient()); |
| original_content_browser_client_ = content::SetBrowserClientForTesting( |
| content_browser_client_.get()); |
| base::FilePath test_data_dir; |
| PathService::Get(chrome::DIR_TEST_DATA, &test_data_dir); |
| CHECK(file_util::ReadFileToString(test_data_dir.Append(kServiceLogin), |
| &service_login_response_)); |
| } |
| |
| virtual void SetUpOnMainThread() OVERRIDE { |
| test_server_ = new EmbeddedTestServer( |
| content::BrowserThread::GetMessageLoopProxyForThread( |
| content::BrowserThread::IO)); |
| CHECK(test_server_->InitializeAndWaitUntilReady()); |
| test_server_->RegisterRequestHandler( |
| base::Bind(&OobeTest::HandleRequest, base::Unretained(this))); |
| LOG(INFO) << "Set up http server at " << test_server_->base_url(); |
| |
| const GURL gaia_url("http://localhost:" + test_server_->base_url().port()); |
| CommandLine::ForCurrentProcess()->AppendSwitchASCII( |
| ::switches::kGaiaUrl, gaia_url.spec()); |
| } |
| |
| virtual void CleanUpOnMainThread() OVERRIDE { |
| LOG(INFO) << "Stopping the http server."; |
| EXPECT_TRUE(test_server_->ShutdownAndWaitUntilComplete()); |
| delete test_server_; // Destructor wants UI thread. |
| } |
| |
| scoped_ptr<HttpResponse> HandleRequest(const HttpRequest& request) { |
| GURL url = test_server_->GetURL(request.relative_url); |
| LOG(INFO) << "Http request: " << url.spec(); |
| |
| scoped_ptr<BasicHttpResponse> http_response(new BasicHttpResponse()); |
| if (url.path() == "/ServiceLogin") { |
| http_response->set_code(net::HTTP_OK); |
| http_response->set_content(service_login_response_); |
| http_response->set_content_type("text/html"); |
| } else if (url.path() == "/ServiceLoginAuth") { |
| LOG(INFO) << "Params: " << request.content; |
| static const char kContinueParam[] = "continue="; |
| int continue_arg_begin = request.content.find(kContinueParam) + |
| arraysize(kContinueParam) - 1; |
| int continue_arg_end = request.content.find("&", continue_arg_begin); |
| const std::string continue_url = request.content.substr( |
| continue_arg_begin, continue_arg_end - continue_arg_begin); |
| http_response->set_code(net::HTTP_OK); |
| const std::string redirect_js = |
| "document.location.href = unescape('" + continue_url + "');"; |
| http_response->set_content( |
| "<HTML><HEAD><SCRIPT>\n" + redirect_js + "\n</SCRIPT></HEAD></HTML>"); |
| http_response->set_content_type("text/html"); |
| } else { |
| NOTREACHED() << url.path(); |
| } |
| return http_response.PassAs<HttpResponse>(); |
| } |
| |
| scoped_ptr<TestContentBrowserClient> content_browser_client_; |
| content::ContentBrowserClient* original_content_browser_client_; |
| std::string service_login_response_; |
| EmbeddedTestServer* test_server_; // cant use scoped_ptr because destructor |
| // needs UI thread. |
| }; |
| |
| IN_PROC_BROWSER_TEST_F(OobeTest, NewUser) { |
| chromeos::WizardController::SkipPostLoginScreensForTesting(); |
| chromeos::WizardController* wizard_controller = |
| chromeos::WizardController::default_controller(); |
| CHECK(wizard_controller); |
| wizard_controller->SkipToLoginForTesting(); |
| |
| scoped_refptr<content::MessageLoopRunner> runner = |
| new content::MessageLoopRunner; |
| content_browser_client_->browser_main_extra_parts_->set_quit_task( |
| runner->QuitClosure()); |
| runner->Run(); |
| } |
| |
| } // namespace |