Chromium Code Reviews
chromiumcodereview-hr@appspot.gserviceaccount.com (chromiumcodereview-hr) | Please choose your nickname with Settings | Help | Chromium Project | Gerrit Changes | Sign out
(6682)

Unified Diff: chrome/browser/chromeos/login/signed_settings_unittest.cc

Issue 8289021: chromeos: Change all clients of LoginLibrary to use SessionManagerClient. (Closed) Base URL: svn://svn.chromium.org/chrome/trunk/src
Patch Set: renamed Created 9 years, 2 months ago
Use n/p to move between diff chunks; N/P to move between comments. Draft comments are only viewable by you.
Jump to:
View side-by-side diff with in-line comments
Download patch
Index: chrome/browser/chromeos/login/signed_settings_unittest.cc
diff --git a/chrome/browser/chromeos/login/signed_settings_unittest.cc b/chrome/browser/chromeos/login/signed_settings_unittest.cc
index 495b1f5ada9aa51638a93818361075fb8a8f8e0b..9bebb7b88f70fbb076df0177330e1480b51f502d 100644
--- a/chrome/browser/chromeos/login/signed_settings_unittest.cc
+++ b/chrome/browser/chromeos/login/signed_settings_unittest.cc
@@ -10,8 +10,9 @@
#include "base/stringprintf.h"
#include "chrome/browser/chromeos/cros/cros_library.h"
#include "chrome/browser/chromeos/cros/mock_library_loader.h"
-#include "chrome/browser/chromeos/cros/mock_login_library.h"
#include "chrome/browser/chromeos/cros_settings_names.h"
+#include "chrome/browser/chromeos/dbus/dbus_thread_manager.h"
+#include "chrome/browser/chromeos/dbus/mock_session_manager_client.h"
#include "chrome/browser/chromeos/login/mock_owner_key_utils.h"
#include "chrome/browser/chromeos/login/mock_ownership_service.h"
#include "chrome/browser/chromeos/login/owner_manager_unittest.h"
@@ -24,7 +25,6 @@
using ::testing::A;
using ::testing::AnyNumber;
-using ::testing::InvokeArgument;
using ::testing::Return;
using ::testing::ReturnRef;
using ::testing::SaveArg;
@@ -116,10 +116,12 @@ class SignedSettingsTest : public testing::Test {
virtual void SetUp() {
file_thread_.Start();
+ DBusThreadManager::Initialize();
}
virtual void TearDown() {
OwnerKeyUtils::set_factory(NULL);
+ DBusThreadManager::Shutdown();
}
void mock_service(SignedSettings* s, MockOwnershipService* m) {
@@ -231,35 +233,6 @@ class SignedSettingsTest : public testing::Test {
message_loop_.RunAllPending();
}
- MockLoginLibrary* MockLoginLib() {
- chromeos::CrosLibrary::TestApi* test_api =
- chromeos::CrosLibrary::Get()->GetTestApi();
-
- // Mocks, ownership transferred to CrosLibrary class on creation.
- MockLoginLibrary* mock_library;
- MockLibraryLoader* loader;
-
- loader = new MockLibraryLoader();
- ON_CALL(*loader, Load(_))
- .WillByDefault(Return(true));
- EXPECT_CALL(*loader, Load(_))
- .Times(AnyNumber());
-
- test_api->SetLibraryLoader(loader, true);
-
- mock_library = new MockLoginLibrary();
- test_api->SetLoginLibrary(mock_library, true);
- return mock_library;
- }
-
- void UnMockLoginLib() {
- // Prevent bogus gMock leak check from firing.
- chromeos::CrosLibrary::TestApi* test_api =
- chromeos::CrosLibrary::Get()->GetTestApi();
- test_api->SetLibraryLoader(NULL, false);
- test_api->SetLoginLibrary(NULL, false);
- }
-
em::PolicyFetchResponse BuildProto(const std::string& data,
const std::string& sig,
std::string* out_serialized) {
@@ -540,7 +513,8 @@ TEST_F(SignedSettingsTest, RetrievePropertyNotFound) {
message_loop_.RunAllPending();
}
-ACTION_P(Retrieve, s) { (*arg0)((void*)arg1, s.c_str(), s.length()); }
+ACTION_P(Retrieve, policy_blob) { arg0.Run(policy_blob); }
+ACTION_P(Store, success) { arg1.Run(success); }
ACTION_P(FinishKeyOp, s) { arg2->OnKeyOpComplete(OwnerManager::SUCCESS, s); }
TEST_F(SignedSettingsTest, RetrievePolicyToRetrieveProperty) {
@@ -555,8 +529,9 @@ TEST_F(SignedSettingsTest, RetrievePolicyToRetrieveProperty) {
em::PolicyFetchResponse signed_policy = BuildProto(data,
fake_value_,
&signed_serialized);
- MockLoginLibrary* lib = MockLoginLib();
- EXPECT_CALL(*lib, RequestRetrievePolicy(_, _))
+ MockSessionManagerClient* client = new MockSessionManagerClient;
+ DBusThreadManager::Get()->set_session_manager_client_for_testing(client);
+ EXPECT_CALL(*client, RetrievePolicy(_))
.WillOnce(Retrieve(signed_serialized))
.RetiresOnSaturation();
@@ -582,7 +557,6 @@ TEST_F(SignedSettingsTest, RetrievePolicyToRetrieveProperty) {
s->Execute();
message_loop_.RunAllPending();
- UnMockLoginLib();
}
TEST_F(SignedSettingsTest, SignAndStorePolicy) {
@@ -617,13 +591,13 @@ TEST_F(SignedSettingsTest, SignAndStorePolicy) {
std::vector<uint8> fake_sig(fake_value_.c_str(),
fake_value_.c_str() + fake_value_.length());
- MockLoginLibrary* lib = MockLoginLib();
- EXPECT_CALL(*lib, RequestStorePolicy(StrEq(signed_serialized), _, s.get()))
- .WillOnce(InvokeArgument<1>(static_cast<void*>(s.get()), true))
+ MockSessionManagerClient* client = new MockSessionManagerClient;;
+ DBusThreadManager::Get()->set_session_manager_client_for_testing(client);
+ EXPECT_CALL(*client, StorePolicy(signed_serialized, _))
+ .WillOnce(Store(true))
.RetiresOnSaturation();
s->OnKeyOpComplete(OwnerManager::SUCCESS, fake_sig);
message_loop_.RunAllPending();
- UnMockLoginLib();
}
TEST_F(SignedSettingsTest, StoreSignedPolicy) {
@@ -638,9 +612,10 @@ TEST_F(SignedSettingsTest, StoreSignedPolicy) {
&signed_serialized);
scoped_refptr<SignedSettings> s(
SignedSettings::CreateStorePolicyOp(&signed_policy, &d));
- MockLoginLibrary* lib = MockLoginLib();
- EXPECT_CALL(*lib, RequestStorePolicy(StrEq(signed_serialized), _, s.get()))
- .WillOnce(InvokeArgument<1>(static_cast<void*>(s.get()), true))
+ MockSessionManagerClient* client = new MockSessionManagerClient;;
+ DBusThreadManager::Get()->set_session_manager_client_for_testing(client);
+ EXPECT_CALL(*client, StorePolicy(signed_serialized, _))
+ .WillOnce(Store(true))
.RetiresOnSaturation();
mock_service(s.get(), &m_);
@@ -650,7 +625,6 @@ TEST_F(SignedSettingsTest, StoreSignedPolicy) {
s->Execute();
message_loop_.RunAllPending();
- UnMockLoginLib();
}
TEST_F(SignedSettingsTest, StorePolicyNoKey) {
@@ -687,11 +661,10 @@ TEST_F(SignedSettingsTest, RetrievePolicy) {
d.expect_success();
scoped_refptr<SignedSettings> s(SignedSettings::CreateRetrievePolicyOp(&d));
- MockLoginLibrary* lib = MockLoginLib();
- EXPECT_CALL(*lib, RequestRetrievePolicy(_, s.get()))
- .WillOnce(InvokeArgument<0>(static_cast<void*>(s.get()),
- signed_serialized.c_str(),
- signed_serialized.length()))
+ MockSessionManagerClient* client = new MockSessionManagerClient;;
+ DBusThreadManager::Get()->set_session_manager_client_for_testing(client);
+ EXPECT_CALL(*client, RetrievePolicy(_))
+ .WillOnce(Retrieve(signed_serialized))
.RetiresOnSaturation();
mock_service(s.get(), &m_);
@@ -705,7 +678,6 @@ TEST_F(SignedSettingsTest, RetrievePolicy) {
s->Execute();
message_loop_.RunAllPending();
- UnMockLoginLib();
s->OnKeyOpComplete(OwnerManager::SUCCESS, std::vector<uint8>());
message_loop_.RunAllPending();
@@ -717,16 +689,14 @@ TEST_F(SignedSettingsTest, RetrieveNullPolicy) {
d.expect_failure(SignedSettings::NOT_FOUND);
scoped_refptr<SignedSettings> s(SignedSettings::CreateRetrievePolicyOp(&d));
- MockLoginLibrary* lib = MockLoginLib();
- EXPECT_CALL(*lib, RequestRetrievePolicy(_, s.get()))
- .WillOnce(InvokeArgument<0>(static_cast<void*>(s.get()),
- static_cast<const char*>(NULL),
- 0))
+ MockSessionManagerClient* client = new MockSessionManagerClient;;
+ DBusThreadManager::Get()->set_session_manager_client_for_testing(client);
+ EXPECT_CALL(*client, RetrievePolicy(_))
+ .WillOnce(Retrieve(""))
.RetiresOnSaturation();
s->Execute();
message_loop_.RunAllPending();
- UnMockLoginLib();
}
TEST_F(SignedSettingsTest, RetrieveEmptyPolicy) {
@@ -736,14 +706,14 @@ TEST_F(SignedSettingsTest, RetrieveEmptyPolicy) {
d.expect_failure(SignedSettings::NOT_FOUND);
scoped_refptr<SignedSettings> s(SignedSettings::CreateRetrievePolicyOp(&d));
- MockLoginLibrary* lib = MockLoginLib();
- EXPECT_CALL(*lib, RequestRetrievePolicy(_, s.get()))
- .WillOnce(InvokeArgument<0>(static_cast<void*>(s.get()), "", 0))
+ MockSessionManagerClient* client = new MockSessionManagerClient;;
+ DBusThreadManager::Get()->set_session_manager_client_for_testing(client);
+ EXPECT_CALL(*client, RetrievePolicy(_))
+ .WillOnce(Retrieve(""))
.RetiresOnSaturation();
s->Execute();
message_loop_.RunAllPending();
- UnMockLoginLib();
}
TEST_F(SignedSettingsTest, RetrieveUnsignedPolicy) {
@@ -755,16 +725,14 @@ TEST_F(SignedSettingsTest, RetrieveUnsignedPolicy) {
d.expect_failure(SignedSettings::BAD_SIGNATURE);
scoped_refptr<SignedSettings> s(SignedSettings::CreateRetrievePolicyOp(&d));
- MockLoginLibrary* lib = MockLoginLib();
- EXPECT_CALL(*lib, RequestRetrievePolicy(_, s.get()))
- .WillOnce(InvokeArgument<0>(static_cast<void*>(s.get()),
- serialized.c_str(),
- serialized.length()))
+ MockSessionManagerClient* client = new MockSessionManagerClient;;
+ DBusThreadManager::Get()->set_session_manager_client_for_testing(client);
+ EXPECT_CALL(*client, RetrievePolicy(_))
+ .WillOnce(Retrieve(serialized))
.RetiresOnSaturation();
s->Execute();
message_loop_.RunAllPending();
- UnMockLoginLib();
}
TEST_F(SignedSettingsTest, RetrieveMalsignedPolicy) {
@@ -776,11 +744,10 @@ TEST_F(SignedSettingsTest, RetrieveMalsignedPolicy) {
d.expect_failure(SignedSettings::BAD_SIGNATURE);
scoped_refptr<SignedSettings> s(SignedSettings::CreateRetrievePolicyOp(&d));
- MockLoginLibrary* lib = MockLoginLib();
- EXPECT_CALL(*lib, RequestRetrievePolicy(_, s.get()))
- .WillOnce(InvokeArgument<0>(static_cast<void*>(s.get()),
- signed_serialized.c_str(),
- signed_serialized.length()))
+ MockSessionManagerClient* client = new MockSessionManagerClient;;
+ DBusThreadManager::Get()->set_session_manager_client_for_testing(client);
+ EXPECT_CALL(*client, RetrievePolicy(_))
+ .WillOnce(Retrieve(signed_serialized))
.RetiresOnSaturation();
mock_service(s.get(), &m_);
@@ -791,7 +758,6 @@ TEST_F(SignedSettingsTest, RetrieveMalsignedPolicy) {
s->Execute();
message_loop_.RunAllPending();
- UnMockLoginLib();
s->OnKeyOpComplete(OwnerManager::OPERATION_FAILED, std::vector<uint8>());
message_loop_.RunAllPending();

Powered by Google App Engine
This is Rietveld 408576698