chrome/browser/chromeos/login/captive_portal_window_browsertest.cc

   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/webui_login_view.h"
  12 #include "chrome/test/base/in_process_browser_test.h"
  13 #include "chromeos/chromeos_switches.h"
  14
  15 namespace chromeos {
  16
  17 namespace {
  18
  19 // Stub implementation of CaptivePortalWindowProxyDelegate, does
  20 // nothing and used to instantiate CaptivePortalWindowProxy.
  21 class CaptivePortalWindowProxyStubDelegate
  22     : public CaptivePortalWindowProxyDelegate {
  23  public:
  24   CaptivePortalWindowProxyStubDelegate(): num_portal_notifications_(0) {
  25   }
  26
  27   virtual ~CaptivePortalWindowProxyStubDelegate() {
  28   }
  29
  30   virtual void OnPortalDetected() OVERRIDE {
  31     ++num_portal_notifications_;
  32   }
  33
  34   int num_portal_notifications() const { return num_portal_notifications_; }
  35
  36  private:
  37   int num_portal_notifications_;
  38 };
  39
  40 }  // namespace
  41
  42 class CaptivePortalWindowTest : public InProcessBrowserTest {
  43  protected:
  44   void ShowIfRedirected() {
  45     captive_portal_window_proxy_->ShowIfRedirected();
  46   }
  47
  48   void Show() {
  49     captive_portal_window_proxy_->Show();
  50   }
  51
  52   void Close() {
  53     captive_portal_window_proxy_->Close();
  54   }
  55
  56   void OnRedirected() {
  57     captive_portal_window_proxy_->OnRedirected();
  58   }
  59
  60   void OnOriginalURLLoaded() {
  61     captive_portal_window_proxy_->OnOriginalURLLoaded();
  62   }
  63
  64   void CheckState(bool is_shown, int num_portal_notifications) {
  65     bool actual_is_shown = (CaptivePortalWindowProxy::STATE_DISPLAYED ==
  66                             captive_portal_window_proxy_->GetState());
  67     ASSERT_EQ(is_shown, actual_is_shown);
  68     ASSERT_EQ(num_portal_notifications, delegate_.num_portal_notifications());
  69   }
  70
  71   virtual void SetUpCommandLine(CommandLine* command_line) OVERRIDE {
  72     command_line->AppendSwitch(chromeos::switches::kForceLoginManagerInTests);
  73     command_line->AppendSwitch(chromeos::switches::kLoginManager);
  74   }
  75
  76   virtual void SetUpOnMainThread() OVERRIDE {
  77     CHECK(LoginDisplayHostImpl::default_host());
  78     content::WebContents* web_contents =
  79         LoginDisplayHostImpl::default_host()->GetWebUILoginView()->
  80             GetWebContents();
  81     captive_portal_window_proxy_.reset(
  82         new CaptivePortalWindowProxy(&delegate_, web_contents));
  83   }
  84
  85  private:
  86   scoped_ptr<CaptivePortalWindowProxy> captive_portal_window_proxy_;
  87   CaptivePortalWindowProxyStubDelegate delegate_;
  88 };
  89
  90 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, ShowClose) {
  91   CheckState(false, 0);
  92
  93   Show();
  94   CheckState(true, 0);
  95
  96   Close();
  97   CheckState(false, 0);
  98 }
  99
 100 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, OnRedirected) {
 101   CheckState(false, 0);
 102
 103   ShowIfRedirected();
 104   CheckState(false, 0);
 105
 106   OnRedirected();
 107   CheckState(true, 1);
 108
 109   Close();
 110   CheckState(false, 1);
 111 }
 112
 113 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, OnOriginalURLLoaded) {
 114   CheckState(false, 0);
 115
 116   ShowIfRedirected();
 117   CheckState(false, 0);
 118
 119   OnRedirected();
 120   CheckState(true, 1);
 121
 122   OnOriginalURLLoaded();
 123   CheckState(false, 1);
 124 }
 125
 126 IN_PROC_BROWSER_TEST_F(CaptivePortalWindowTest, MultipleCalls) {
 127   CheckState(false, 0);
 128
 129   ShowIfRedirected();
 130   CheckState(false, 0);
 131
 132   Show();
 133   CheckState(true, 0);
 134
 135   Close();
 136   CheckState(false, 0);
 137
 138   OnRedirected();
 139   CheckState(false, 1);
 140
 141   OnOriginalURLLoaded();
 142   CheckState(false, 1);
 143
 144   Show();
 145   CheckState(true, 1);
 146
 147   OnRedirected();
 148   CheckState(true, 2);
 149
 150   Close();
 151   CheckState(false, 2);
 152
 153   OnOriginalURLLoaded();
 154   CheckState(false, 2);
 155 }
 156
 157 }  // namespace chromeos