This source file includes following definitions.
- window
- multi_user_window_manager
- SetNumberOfUsers
- SetUp
- TearDown
- TEST_F
- TEST_F
#include "ash/shell.h"
#include "ash/test/ash_test_base.h"
#include "ash/test/test_session_state_delegate.h"
#include "ash/test/test_shell_delegate.h"
#include "base/command_line.h"
#include "base/compiler_specific.h"
#include "base/logging.h"
#include "chrome/browser/ui/ash/multi_user/multi_user_context_menu.h"
#include "chrome/browser/ui/ash/multi_user/multi_user_window_manager.h"
#include "chrome/browser/ui/ash/multi_user/multi_user_window_manager_chromeos.h"
#include "chrome/common/chrome_switches.h"
#include "ui/base/models/menu_model.h"
#include "ui/base/ui_base_types.h"
namespace ash {
namespace test {
class MultiUserContextMenuChromeOSTest : public AshTestBase {
public:
MultiUserContextMenuChromeOSTest() : multi_user_window_manager_(NULL) {}
virtual void SetUp() OVERRIDE;
virtual void TearDown() OVERRIDE;
protected:
void SetUpForThisManyWindows(int windows);
aura::Window* window() { return window_; }
chrome::MultiUserWindowManagerChromeOS* multi_user_window_manager() {
return multi_user_window_manager_;
}
void SetNumberOfUsers(int users) {
ash::test::TestSessionStateDelegate* delegate =
static_cast<ash::test::TestSessionStateDelegate*>(
ash::Shell::GetInstance()->session_state_delegate());
delegate->set_logged_in_users(users);
}
private:
aura::Window* window_;
chrome::MultiUserWindowManagerChromeOS* multi_user_window_manager_;
DISALLOW_COPY_AND_ASSIGN(MultiUserContextMenuChromeOSTest);
};
void MultiUserContextMenuChromeOSTest::SetUp() {
CommandLine::ForCurrentProcess()->AppendSwitch(switches::kMultiProfiles);
AshTestBase::SetUp();
window_ = CreateTestWindowInShellWithId(0);
window_->Show();
multi_user_window_manager_ = new chrome::MultiUserWindowManagerChromeOS("A");
chrome::MultiUserWindowManager::SetInstanceForTest(multi_user_window_manager_,
chrome::MultiUserWindowManager::MULTI_PROFILE_MODE_SEPARATED);
EXPECT_TRUE(multi_user_window_manager_);
}
void MultiUserContextMenuChromeOSTest::TearDown() {
delete window_;
chrome::MultiUserWindowManager::DeleteInstance();
AshTestBase::TearDown();
}
TEST_F(MultiUserContextMenuChromeOSTest, UnownedWindow) {
EXPECT_EQ(NULL, CreateMultiUserContextMenu(window()).get());
SetNumberOfUsers(2);
EXPECT_EQ(NULL, CreateMultiUserContextMenu(window()).get());
}
TEST_F(MultiUserContextMenuChromeOSTest, OwnedWindow) {
multi_user_window_manager()->SetWindowOwner(window(), "A");
EXPECT_EQ(NULL, CreateMultiUserContextMenu(window()).get());
{
SetNumberOfUsers(2);
scoped_ptr<ui::MenuModel> menu = CreateMultiUserContextMenu(window());
ASSERT_TRUE(menu.get());
EXPECT_EQ(1, menu.get()->GetItemCount());
}
{
SetNumberOfUsers(3);
scoped_ptr<ui::MenuModel> menu = CreateMultiUserContextMenu(window());
ASSERT_TRUE(menu.get());
EXPECT_EQ(2, menu.get()->GetItemCount());
}
}
}
}