| OLD | NEW |
| (Empty) |
| 1 // Copyright 2016 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 "ash/common/system/tiles/tray_tiles.h" | |
| 6 | |
| 7 #include "ash/common/session/session_state_delegate.h" | |
| 8 #include "ash/common/system/tiles/tiles_default_view.h" | |
| 9 #include "ash/common/wm_shell.h" | |
| 10 #include "ash/test/ash_test_base.h" | |
| 11 #include "ui/views/controls/button/custom_button.h" | |
| 12 #include "ui/views/view.h" | |
| 13 | |
| 14 namespace ash { | |
| 15 | |
| 16 class TrayTilesTest : public test::AshTestBase { | |
| 17 public: | |
| 18 TrayTilesTest() {} | |
| 19 ~TrayTilesTest() override {} | |
| 20 | |
| 21 void SetUp() override { | |
| 22 test::AshTestBase::SetUp(); | |
| 23 tray_tiles_.reset(new TrayTiles(GetPrimarySystemTray())); | |
| 24 } | |
| 25 | |
| 26 void TearDown() override { | |
| 27 tray_tiles_.reset(); | |
| 28 test::AshTestBase::TearDown(); | |
| 29 } | |
| 30 | |
| 31 views::CustomButton* GetSettingsButton() { | |
| 32 return tray_tiles()->default_view_->settings_button_; | |
| 33 } | |
| 34 | |
| 35 views::CustomButton* GetHelpButton() { | |
| 36 return tray_tiles()->default_view_->help_button_; | |
| 37 } | |
| 38 | |
| 39 views::CustomButton* GetLockButton() { | |
| 40 return tray_tiles()->default_view_->lock_button_; | |
| 41 } | |
| 42 | |
| 43 views::CustomButton* GetPowerButton() { | |
| 44 return tray_tiles()->default_view_->power_button_; | |
| 45 } | |
| 46 | |
| 47 TrayTiles* tray_tiles() { return tray_tiles_.get(); } | |
| 48 | |
| 49 private: | |
| 50 std::unique_ptr<TrayTiles> tray_tiles_; | |
| 51 | |
| 52 DISALLOW_COPY_AND_ASSIGN(TrayTilesTest); | |
| 53 }; | |
| 54 | |
| 55 TEST_F(TrayTilesTest, ButtonStatesWithAddingUser) { | |
| 56 SetUserAddingScreenRunning(true); | |
| 57 std::unique_ptr<views::View> default_view( | |
| 58 tray_tiles()->CreateDefaultViewForTesting(LoginStatus::USER)); | |
| 59 EXPECT_EQ(GetSettingsButton()->state(), views::Button::STATE_DISABLED); | |
| 60 EXPECT_EQ(GetHelpButton()->state(), views::Button::STATE_DISABLED); | |
| 61 EXPECT_EQ(GetLockButton()->state(), views::Button::STATE_DISABLED); | |
| 62 EXPECT_EQ(GetPowerButton()->state(), views::Button::STATE_NORMAL); | |
| 63 } | |
| 64 | |
| 65 TEST_F(TrayTilesTest, ButtonStatesWithLoginStatusNotLoggedIn) { | |
| 66 std::unique_ptr<views::View> default_view( | |
| 67 tray_tiles()->CreateDefaultViewForTesting(LoginStatus::NOT_LOGGED_IN)); | |
| 68 EXPECT_EQ(GetSettingsButton()->state(), views::Button::STATE_DISABLED); | |
| 69 EXPECT_EQ(GetHelpButton()->state(), views::Button::STATE_DISABLED); | |
| 70 EXPECT_EQ(GetLockButton()->state(), views::Button::STATE_DISABLED); | |
| 71 EXPECT_EQ(GetPowerButton()->state(), views::Button::STATE_NORMAL); | |
| 72 } | |
| 73 | |
| 74 TEST_F(TrayTilesTest, ButtonStatesWithLoginStatusLocked) { | |
| 75 std::unique_ptr<views::View> default_view( | |
| 76 tray_tiles()->CreateDefaultViewForTesting(LoginStatus::LOCKED)); | |
| 77 EXPECT_EQ(GetSettingsButton()->state(), views::Button::STATE_DISABLED); | |
| 78 EXPECT_EQ(GetHelpButton()->state(), views::Button::STATE_DISABLED); | |
| 79 EXPECT_EQ(GetLockButton()->state(), views::Button::STATE_DISABLED); | |
| 80 EXPECT_EQ(GetPowerButton()->state(), views::Button::STATE_NORMAL); | |
| 81 } | |
| 82 | |
| 83 TEST_F(TrayTilesTest, ButtonStatesWithLoginStatusUser) { | |
| 84 std::unique_ptr<views::View> default_view( | |
| 85 tray_tiles()->CreateDefaultViewForTesting(LoginStatus::USER)); | |
| 86 EXPECT_EQ(GetSettingsButton()->state(), views::Button::STATE_NORMAL); | |
| 87 EXPECT_EQ(GetHelpButton()->state(), views::Button::STATE_NORMAL); | |
| 88 EXPECT_EQ(GetLockButton()->state(), views::Button::STATE_NORMAL); | |
| 89 EXPECT_EQ(GetPowerButton()->state(), views::Button::STATE_NORMAL); | |
| 90 } | |
| 91 | |
| 92 } // namespace ash | |
| OLD | NEW |