Implement ephemeral users

chrome/browser/chromeos/login/user_manager_unittest.cc

+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
+// Use of this source code is governed by a BSD-style license that can be
+// found in the LICENSE file.
+
+#include <cstdlib>
+#include <cstring>
+
+#include "base/memory/scoped_ptr.h"
+#include "base/message_loop.h"
+#include "base/values.h"
+#include "chrome/browser/browser_process.h"
+#include "chrome/browser/chromeos/cros_settings.h"
+#include "chrome/browser/chromeos/cros_settings_names.h"
+#include "chrome/browser/chromeos/cros_settings_provider.h"
+#include "chrome/browser/chromeos/login/user.h"
+#include "chrome/browser/chromeos/login/user_manager.h"
+#include "chrome/browser/chromeos/login/user_manager_impl.h"
+#include "chrome/browser/chromeos/stub_cros_settings_provider.h"
+#include "chrome/browser/prefs/pref_service.h"
+#include "chrome/test/base/testing_browser_process.h"
+#include "chrome/test/base/testing_pref_service.h"
+#include "content/test/test_browser_thread.h"
+#include "testing/gtest/include/gtest/gtest.h"
+
+namespace chromeos {
+
+class UserManagerTest : public testing::Test {
+ public:
+  UserManagerTest()
+      : message_loop_(MessageLoop::TYPE_UI),
+        ui_thread_(content::BrowserThread::UI, &message_loop_),
+        file_thread_(content::BrowserThread::FILE, &message_loop_) {
+  }
+
+  virtual void SetUp() {
+    cros_settings_ = CrosSettings::Get();
+
+    // Replace the real DeviceSettingsProvider with a stub.
+    device_settings_provider_ =
+        cros_settings_->GetProvider(chromeos::kReportDeviceVersionInfo);
+    EXPECT_TRUE(device_settings_provider_ != NULL);
+    EXPECT_TRUE(
+        cros_settings_->RemoveSettingsProvider(device_settings_provider_));
+    cros_settings_->AddSettingsProvider(&stub_settings_provider_);
+
+    // Populate the stub DeviceSettingsProvider with valid values.
+    SetDeviceSettings(false, "");
+
+    // Register an in-memory local settings instance.
+    local_state_.reset(new TestingPrefService);
+    reinterpret_cast<TestingBrowserProcess*>(g_browser_process)
+        ->SetLocalState(local_state_.get());
+    UserManager::RegisterPrefs(local_state_.get());
+
+    // A stub user is automatically logged in by UserManager. Reset this.
+    ResetUserManager();
+  }
+
+  virtual void TearDown() {
+    // Unregister the in-memory local settings instance.
+    reinterpret_cast<TestingBrowserProcess*>(g_browser_process)
+        ->SetLocalState(0);
+
+    // Restore the real DeviceSettingsProvider.
+    EXPECT_TRUE(
+      cros_settings_->RemoveSettingsProvider(&stub_settings_provider_));
+    cros_settings_->AddSettingsProvider(device_settings_provider_);
+  }
+
+  bool GetUserManagerEphemeralUsersEnabled() const {
+    return reinterpret_cast<UserManagerImpl*>(UserManager::Get())->
+        ephemeral_users_enabled_;
+  }
+
+  void SetUserManagerEphemeralUsersEnabled(bool ephemeral_users_enabled) {
+    reinterpret_cast<UserManagerImpl*>(UserManager::Get())->
+        ephemeral_users_enabled_ = ephemeral_users_enabled;
+  }
+
+  const std::string& GetUserManagerOwnerEmail() const {
+    return reinterpret_cast<UserManagerImpl*>(UserManager::Get())->
+        owner_email_;
+  }
+
+  void SetUserManagerOwnerEmail(const std::string& owner_email) {
+    reinterpret_cast<UserManagerImpl*>(UserManager::Get())->
+        owner_email_ = owner_email;
+  }
+
+  void ResetUserManager() {
+    UserManagerImpl* user_manager_impl(new UserManagerImpl());
+    user_manager_impl->logged_in_user_ = NULL;
+    UserManager::Set(user_manager_impl);
+  }
+
+  void SetDeviceSettings(bool ephemeral_users_enabled,
+                         const std::string &owner) {
+    stub_settings_provider_.Set(kAccountsPrefEphemeralUsersEnabled,
+        base::FundamentalValue(ephemeral_users_enabled));
+    base::StringValue owner_value(owner);
+    stub_settings_provider_.Set(kDeviceOwner, owner_value);
+  }
+
+  void RetrieveTrustedDevicePolicies() {
+    reinterpret_cast<UserManagerImpl*>(UserManager::Get())->
+        RetrieveTrustedDevicePolicies();
+  }
+
+ protected:
+  MessageLoop message_loop_;
+  content::TestBrowserThread ui_thread_;
+  content::TestBrowserThread file_thread_;
+
+  CrosSettings* cros_settings_;
+  CrosSettingsProvider* device_settings_provider_;
+  StubCrosSettingsProvider stub_settings_provider_;
+  scoped_ptr<TestingPrefService> local_state_;
+};
+
+TEST_F(UserManagerTest, RetrieveTrustedDevicePolicies) {
+  SetUserManagerEphemeralUsersEnabled(true);
+  SetUserManagerOwnerEmail("");
+
+  SetDeviceSettings(false, "owner@invalid.domain");
+  RetrieveTrustedDevicePolicies();
+
+  EXPECT_FALSE(GetUserManagerEphemeralUsersEnabled());
+  EXPECT_EQ(GetUserManagerOwnerEmail(), "owner@invalid.domain");
+}
+
+TEST_F(UserManagerTest, RemoveAllExceptOwnerFromList) {
+  UserManager::Get()->UserLoggedIn("owner@invalid.domain");
+  ResetUserManager();
+  UserManager::Get()->UserLoggedIn("user0@invalid.domain");
+  ResetUserManager();
+  UserManager::Get()->UserLoggedIn("user1@invalid.domain");
+  ResetUserManager();
+
+  const UserList* users = &UserManager::Get()->GetUsers();
+  EXPECT_TRUE(users->size() == 3);
+  EXPECT_EQ((*users)[0]->email(), "user1@invalid.domain");
+  EXPECT_EQ((*users)[1]->email(), "user0@invalid.domain");
+  EXPECT_EQ((*users)[2]->email(), "owner@invalid.domain");
+
+  SetDeviceSettings(true, "owner@invalid.domain");
+  RetrieveTrustedDevicePolicies();
+
+  users = &UserManager::Get()->GetUsers();
+  EXPECT_TRUE(users->size() == 1);
+  EXPECT_EQ((*users)[0]->email(), "owner@invalid.domain");
+}
+
+TEST_F(UserManagerTest, EphemeralUserLoggedIn) {
+  SetDeviceSettings(true, "owner@invalid.domain");
+  RetrieveTrustedDevicePolicies();
+
+  UserManager::Get()->UserLoggedIn("owner@invalid.domain");
+  ResetUserManager();
+  UserManager::Get()->UserLoggedIn("user0@invalid.domain");
+  ResetUserManager();
+
+  const UserList* users = &UserManager::Get()->GetUsers();
+  EXPECT_TRUE(users->size() == 1);
+  EXPECT_EQ((*users)[0]->email(), "owner@invalid.domain");
+}
+
+}  // namespace chromeos