Home | History | Annotate | Download | only in login 
      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 #ifndef CHROME_BROWSER_CHROMEOS_LOGIN_LOGIN_MANAGER_TEST_H_
      6 #define CHROME_BROWSER_CHROMEOS_LOGIN_LOGIN_MANAGER_TEST_H_
      7 
      8 #include "chrome/browser/chromeos/login/mock_login_utils.h"
      9 #include "chrome/browser/chromeos/login/test/js_checker.h"
     10 #include "chrome/test/base/in_process_browser_test.h"
     11 
     12 namespace content {
     13 class WebContents;
     14 }  // namespace content
     15 
     16 namespace chromeos {
     17 
     18 // Base class for Chrome OS out-of-box/login WebUI tests.
     19 // If no special configuration is done launches out-of-box WebUI.
     20 // To launch login UI use PRE_* test that will register user(s) and mark
     21 // out-of-box as completed.
     22 // Guarantees that WebUI has been initialized by waiting for
     23 // NOTIFICATION_LOGIN_OR_LOCK_WEBUI_VISIBLE notification.
     24 class LoginManagerTest : public InProcessBrowserTest {
     25  public:
     26   explicit LoginManagerTest(bool should_launch_browser);
     27 
     28   // Overriden from InProcessBrowserTest.
     29   virtual void CleanUpOnMainThread() OVERRIDE;
     30   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE;
     31   virtual void SetUpInProcessBrowserTestFixture() OVERRIDE;
     32   virtual void SetUpOnMainThread() OVERRIDE;
     33 
     34   // Registers user with given |username| on device.
     35   // Should be called in PRE_* test.
     36   // TODO(dzhioev): add ability to register users without PRE_* test.
     37   void RegisterUser(const std::string& username);
     38 
     39   // Set expected credentials for next login attempt.
     40   void SetExpectedCredentials(const std::string& username,
     41                               const std::string& password);
     42 
     43   // Tries to login with |username| and |password|. Returns false if attempt
     44   // has failed.
     45   bool TryToLogin(const std::string& username, const std::string& password);
     46 
     47   // Tries to add user to session with |username| and |password|. Returns false
     48   // if attempt has failed. this function does the same as TryToLogin but
     49   // doesn't check that new user become active user.
     50   bool AddUserTosession(const std::string& username,
     51                         const std::string& password);
     52 
     53   // Login user with |username|. User should be registered using RegisterUser().
     54   void LoginUser(const std::string& username);
     55 
     56   // Add user with |username| to session.
     57   void AddUser(const std::string& username);
     58 
     59   // Executes given JS |expression| in |web_contents_| and checks
     60   // that it is true.
     61   void JSExpect(const std::string& expression);
     62 
     63   MockLoginUtils& login_utils() { return *mock_login_utils_; }
     64 
     65   content::WebContents* web_contents() { return web_contents_; }
     66 
     67  private:
     68   void InitializeWebContents();
     69 
     70   void set_web_contents(content::WebContents* web_contents) {
     71     web_contents_ = web_contents;
     72   }
     73 
     74   MockLoginUtils* mock_login_utils_;
     75   bool should_launch_browser_;
     76   content::WebContents* web_contents_;
     77   test::JSChecker js_checker_;
     78 
     79   DISALLOW_COPY_AND_ASSIGN(LoginManagerTest);
     80 };
     81 
     82 }  // namespace chromeos
     83 
     84 #endif  // CHROME_BROWSER_CHROMEOS_LOGIN_LOGIN_MANAGER_TEST_H_
     85