Home | History | Annotate | Download | only in signin 
      1 // Copyright (c) 2012 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 "chrome/browser/ui/webui/signin/login_ui_service.h"
      6 
      7 #include "base/compiler_specific.h"
      8 #include "base/logging.h"
      9 #include "testing/gtest/include/gtest/gtest.h"
     10 
     11 class TestLoginUI : public LoginUIService::LoginUI {
     12  public:
     13   TestLoginUI() { }
     14   virtual ~TestLoginUI() { }
     15   virtual void FocusUI() OVERRIDE { }
     16   virtual void CloseUI() OVERRIDE { }
     17 
     18  private:
     19   DISALLOW_COPY_AND_ASSIGN(TestLoginUI);
     20 };
     21 
     22 class TestObserver : public LoginUIService::Observer {
     23  public:
     24   TestObserver() : ui_shown_count_(0),
     25                    ui_closed_count_(0) {
     26   }
     27 
     28   int ui_shown_count() const { return ui_shown_count_; }
     29   int ui_closed_count() const { return ui_closed_count_; }
     30 
     31  private:
     32   virtual void OnLoginUIShown(LoginUIService::LoginUI* ui) OVERRIDE {
     33     ++ui_shown_count_;
     34   }
     35 
     36   virtual void OnLoginUIClosed(LoginUIService::LoginUI* ui) OVERRIDE {
     37     ++ui_closed_count_;
     38   }
     39 
     40   int ui_shown_count_;
     41   int ui_closed_count_;
     42 
     43   DISALLOW_COPY_AND_ASSIGN(TestObserver);
     44 };
     45 
     46 class LoginUIServiceTest : public testing::Test {
     47  public:
     48   LoginUIServiceTest() : service_(NULL) { }
     49   virtual ~LoginUIServiceTest() { }
     50 
     51  protected:
     52   LoginUIService service_;
     53   TestObserver observer_;
     54 
     55   DISALLOW_COPY_AND_ASSIGN(LoginUIServiceTest);
     56 };
     57 
     58 // Test that the observer is called back when login UI is shown
     59 // or closed.
     60 TEST_F(LoginUIServiceTest, LoginUIServiceObserver) {
     61   service_.AddObserver(&observer_);
     62   EXPECT_EQ(NULL, service_.current_login_ui());
     63   TestLoginUI ui;
     64   service_.SetLoginUI(&ui);
     65   EXPECT_EQ(1, observer_.ui_shown_count());
     66 
     67   // If a different UI is closed than the one that was shown, no
     68   // notification should be fired.
     69   TestLoginUI other_ui;
     70   service_.LoginUIClosed(&other_ui);
     71   EXPECT_EQ(1, observer_.ui_shown_count());
     72 
     73   // If the right UI is closed, notification should be fired.
     74   service_.LoginUIClosed(&ui);
     75   EXPECT_EQ(1, observer_.ui_closed_count());
     76   service_.RemoveObserver(&observer_);
     77 }
     78