Add browser test for new user CrOS login flow

chrome/browser/chromeos/login/oobe_browsertest.cc

+// 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/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/google_apis/test_server/http_request.h"
+#include "chrome/browser/google_apis/test_server/http_response.h"
+#include "chrome/browser/google_apis/test_server/http_server.h"
+#include "chrome/browser/ui/browser.h"
+#include "chrome/common/chrome_notification_types.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 "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 "testing/gmock/include/gmock/gmock.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+using namespace google_apis;
+using namespace google_apis::test_server;
+
+namespace {
+
+// Used to add an observer to NotificationService after it's created.
+class TestBrowserMainExtraParts
+    : public ChromeBrowserMainExtraParts,
+      public content::NotificationObserver {
+ public:
+  TestBrowserMainExtraParts() {}
+  virtual ~TestBrowserMainExtraParts() {}
+
+  // ChromeBrowserMainExtraParts implementation.
+  virtual void PreEarlyInitialization() OVERRIDE {
+    registrar_.Add(this, chrome::NOTIFICATION_LOGIN_WEBUI_VISIBLE,
+                   content::NotificationService::AllSources());
+    registrar_.Add(this, chrome::NOTIFICATION_SESSION_STARTED,
+                   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_WEBUI_VISIBLE) {
+      LOG(INFO) << "NOTIFICATION_LOGIN_WEBUI_VISIBLE";
+      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");
+    } else if (type == chrome::NOTIFICATION_SESSION_STARTED) {
+      LOG(INFO) << "chrome::NOTIFICATION_SESSION_STARTED";
+      quit_task_.Run();
+    } else {
+      NOTREACHED();
+    }
+  }
+
+  content::NotificationRegistrar registrar_;
+  base::Closure quit_task_;
+
+  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(switches::kLoginManager);
+    command_line->AppendSwitch(switches::kForceLoginManagerInTests);
+    command_line->AppendSwitchASCII(switches::kLoginProfile, "user");
+    command_line->AppendSwitchASCII(switches::kAuthExtensionPath, "gaia_auth");
+  }
+
+  virtual void SetUpInProcessBrowserTestFixture() OVERRIDE {
+    // host_resolver()->AddRule("*", "127.0.0.1"); -- work in progress!
+    content_browser_client_.reset(new TestContentBrowserClient());
+    original_content_browser_client_ = content::GetContentClient()->browser();
+    content::GetContentClient()->set_browser_for_testing(
+        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_));
+  }
+
+  scoped_ptr<TestContentBrowserClient> content_browser_client_;
+  content::ContentBrowserClient* original_content_browser_client_;
+
+ public:
+  scoped_ptr<HttpResponse> HandleRequest(const HttpRequest& request) {
+    GURL url = test_server_->GetURL(request.relative_url);
+    LOG(WARNING) << "Http request: " << url.spec();
+
+    scoped_ptr<HttpResponse> http_response(new HttpResponse());
+    if (url.path() == "/ServiceLogin") {
+      http_response->set_code(test_server::SUCCESS);
+      http_response->set_content(service_login_response_);
+      http_response->set_content_type("text/html");
+    } else if (url.path() == "/ServiceLoginAuth") {
+      int continue_arg_begin = request.content.find("continue=") + 9;
+      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(test_server::SUCCESS);
+      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.Pass();
+  }
+
+  HttpServer* test_server_;
+  std::string service_login_response_;
+};
+
+IN_PROC_BROWSER_TEST_F(OobeTest, NewUser) {
+  test_server_ = new HttpServer();
+  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();
+
+  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();
+
+  LOG(INFO) << "Stopping the http server.";
+  content::BrowserThread::CurrentlyOn(content::BrowserThread::UI);
+  test_server_->ShutdownAndWaitUntilComplete();
+}
+
+}