| OLD | NEW |
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 The Chromium Authors. All rights reserved. |
| 2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
| 3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
| 4 | 4 |
| 5 #include "chrome/browser/chromeos/policy/server_backed_state_keys_broker.h" | 5 #include "chrome/browser/chromeos/policy/server_backed_state_keys_broker.h" |
| 6 | 6 |
| 7 #include "base/bind.h" | 7 #include "base/bind.h" |
| 8 #include "base/bind_helpers.h" | 8 #include "base/bind_helpers.h" |
| 9 #include "base/macros.h" | 9 #include "base/macros.h" |
| 10 #include "base/message_loop/message_loop.h" | 10 #include "base/message_loop/message_loop.h" |
| (...skipping 95 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 106 base::Unretained(this))); | 106 base::Unretained(this))); |
| 107 base::RunLoop().RunUntilIdle(); | 107 base::RunLoop().RunUntilIdle(); |
| 108 EXPECT_TRUE(updated_); | 108 EXPECT_TRUE(updated_); |
| 109 ExpectGood(); | 109 ExpectGood(); |
| 110 | 110 |
| 111 // Update callbacks get fired if the keys change. | 111 // Update callbacks get fired if the keys change. |
| 112 state_keys_.erase(state_keys_.begin()); | 112 state_keys_.erase(state_keys_.begin()); |
| 113 state_keys_.push_back("4"); | 113 state_keys_.push_back("4"); |
| 114 fake_session_manager_client_.set_server_backed_state_keys(state_keys_); | 114 fake_session_manager_client_.set_server_backed_state_keys(state_keys_); |
| 115 updated_ = false; | 115 updated_ = false; |
| 116 task_runner_->RunPendingTasks(); | 116 task_runner_->RunUntilIdle(); |
| 117 base::RunLoop().RunUntilIdle(); | |
| 118 EXPECT_TRUE(updated_); | 117 EXPECT_TRUE(updated_); |
| 119 ExpectGood(); | 118 ExpectGood(); |
| 120 | 119 |
| 121 // No update callback if the keys are unchanged. | 120 // No update callback if the keys are unchanged. |
| 122 updated_ = false; | 121 updated_ = false; |
| 123 task_runner_->RunPendingTasks(); | 122 task_runner_->RunUntilIdle(); |
| 124 base::RunLoop().RunUntilIdle(); | |
| 125 EXPECT_FALSE(updated_); | 123 EXPECT_FALSE(updated_); |
| 126 ExpectGood(); | 124 ExpectGood(); |
| 127 } | 125 } |
| 128 | 126 |
| 129 TEST_F(ServerBackedStateKeysBrokerTest, Request) { | 127 TEST_F(ServerBackedStateKeysBrokerTest, Request) { |
| 130 broker_.RequestStateKeys( | 128 broker_.RequestStateKeys( |
| 131 base::Bind(&ServerBackedStateKeysBrokerTest::HandleStateKeysCallback, | 129 base::Bind(&ServerBackedStateKeysBrokerTest::HandleStateKeysCallback, |
| 132 base::Unretained(this))); | 130 base::Unretained(this))); |
| 133 base::RunLoop().RunUntilIdle(); | 131 base::RunLoop().RunUntilIdle(); |
| 134 ExpectGood(); | 132 ExpectGood(); |
| 135 EXPECT_TRUE(callback_invoked_); | 133 EXPECT_TRUE(callback_invoked_); |
| 136 EXPECT_EQ(state_keys_, callback_state_keys_); | 134 EXPECT_EQ(state_keys_, callback_state_keys_); |
| 137 } | 135 } |
| 138 | 136 |
| 139 TEST_F(ServerBackedStateKeysBrokerTest, RequestFailure) { | 137 TEST_F(ServerBackedStateKeysBrokerTest, RequestFailure) { |
| 140 fake_session_manager_client_.set_server_backed_state_keys( | 138 fake_session_manager_client_.set_server_backed_state_keys( |
| 141 std::vector<std::string>()); | 139 std::vector<std::string>()); |
| 142 | 140 |
| 143 broker_.RequestStateKeys( | 141 broker_.RequestStateKeys( |
| 144 base::Bind(&ServerBackedStateKeysBrokerTest::HandleStateKeysCallback, | 142 base::Bind(&ServerBackedStateKeysBrokerTest::HandleStateKeysCallback, |
| 145 base::Unretained(this))); | 143 base::Unretained(this))); |
| 146 base::RunLoop().RunUntilIdle(); | 144 base::RunLoop().RunUntilIdle(); |
| 147 EXPECT_TRUE(callback_invoked_); | 145 EXPECT_TRUE(callback_invoked_); |
| 148 EXPECT_TRUE(callback_state_keys_.empty()); | 146 EXPECT_TRUE(callback_state_keys_.empty()); |
| 149 } | 147 } |
| 150 | 148 |
| 151 } // namespace policy | 149 } // namespace policy |
| OLD | NEW |