chromeos/cryptohome/system_salt_getter_unittest.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 "chromeos/cryptohome/system_salt_getter.h"
   6
   7 #include "base/bind.h"
   8 #include "base/message_loop/message_loop.h"
   9 #include "base/run_loop.h"
  10 #include "chromeos/dbus/fake_cryptohome_client.h"
  11 #include "chromeos/dbus/fake_dbus_thread_manager.h"
  12 #include "testing/gtest/include/gtest/gtest.h"
  13
  14 namespace chromeos {
  15 namespace {
  16
  17 // Used as a GetSystemSaltCallback.
  18 void CopySystemSalt(std::string* out_system_salt,
  19                     const std::string& system_salt) {
  20   *out_system_salt = system_salt;
  21 }
  22
  23 class SystemSaltGetterTest : public testing::Test {
  24  protected:
  25   SystemSaltGetterTest() : fake_cryptohome_client_(NULL) {}
  26
  27   virtual void SetUp() OVERRIDE {
  28     FakeDBusThreadManager* dbus_thread_manager = new FakeDBusThreadManager;
  29     fake_cryptohome_client_ = new FakeCryptohomeClient;
  30     dbus_thread_manager->SetCryptohomeClient(
  31         scoped_ptr<CryptohomeClient>(fake_cryptohome_client_));
  32     DBusThreadManager::InitializeForTesting(dbus_thread_manager);
  33
  34     EXPECT_FALSE(SystemSaltGetter::IsInitialized());
  35     SystemSaltGetter::Initialize();
  36     ASSERT_TRUE(SystemSaltGetter::IsInitialized());
  37     ASSERT_TRUE(SystemSaltGetter::Get());
  38   }
  39
  40   virtual void TearDown() OVERRIDE {
  41     SystemSaltGetter::Shutdown();
  42     DBusThreadManager::Shutdown();
  43   }
  44
  45   base::MessageLoopForUI message_loop_;
  46   FakeCryptohomeClient* fake_cryptohome_client_;
  47 };
  48
  49 TEST_F(SystemSaltGetterTest, GetSystemSalt) {
  50   // Try to get system salt before the service becomes available.
  51   fake_cryptohome_client_->SetServiceIsAvailable(false);
  52   std::string system_salt;
  53   SystemSaltGetter::Get()->GetSystemSalt(
  54       base::Bind(&CopySystemSalt, &system_salt));
  55   base::RunLoop().RunUntilIdle();
  56   EXPECT_TRUE(system_salt.empty());  // System salt is not returned yet.
  57
  58   // Service becomes available.
  59   fake_cryptohome_client_->SetServiceIsAvailable(true);
  60   base::RunLoop().RunUntilIdle();
  61   const std::string expected_system_salt =
  62       SystemSaltGetter::ConvertRawSaltToHexString(
  63           FakeCryptohomeClient::GetStubSystemSalt());
  64   EXPECT_EQ(expected_system_salt, system_salt);  // System salt is returned.
  65 }
  66
  67 }  // namespace
  68 }  // namespace chromeos