| 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 "extensions/browser/api/power/power_api.h" | 5 #include "extensions/browser/api/power/power_api.h" |
| 6 | 6 |
| 7 #include <deque> | 7 #include <deque> |
| 8 #include <string> | 8 #include <string> |
| 9 | 9 |
| 10 #include "base/basictypes.h" | 10 #include "base/basictypes.h" |
| 11 #include "base/memory/ref_counted.h" | 11 #include "base/memory/ref_counted.h" |
| 12 #include "base/memory/scoped_ptr.h" | 12 #include "base/memory/scoped_ptr.h" |
| 13 #include "base/memory/weak_ptr.h" | 13 #include "base/memory/weak_ptr.h" |
| 14 #include "content/public/browser/power_save_blocker.h" | 14 #include "content/public/browser/power_save_blocker.h" |
| 15 #include "extensions/browser/api/power/power_api_manager.h" | |
| 16 #include "extensions/browser/api_test_utils.h" | 15 #include "extensions/browser/api_test_utils.h" |
| 17 #include "extensions/browser/api_unittest.h" | 16 #include "extensions/browser/api_unittest.h" |
| 18 #include "extensions/common/extension.h" | 17 #include "extensions/common/extension.h" |
| 19 #include "extensions/common/test_util.h" | 18 #include "extensions/common/test_util.h" |
| 20 | 19 |
| 21 namespace extensions { | 20 namespace extensions { |
| 22 | 21 |
| 23 namespace { | 22 namespace { |
| 24 | 23 |
| 25 // Args commonly passed to PowerSaveBlockerStubManager::CallFunction(). | 24 // Args commonly passed to PowerSaveBlockerStubManager::CallFunction(). |
| (...skipping 23 matching lines...) Expand all Loading... |
| 49 | 48 |
| 50 ~PowerSaveBlockerStub() override { unblock_callback_.Run(); } | 49 ~PowerSaveBlockerStub() override { unblock_callback_.Run(); } |
| 51 | 50 |
| 52 private: | 51 private: |
| 53 base::Closure unblock_callback_; | 52 base::Closure unblock_callback_; |
| 54 | 53 |
| 55 DISALLOW_COPY_AND_ASSIGN(PowerSaveBlockerStub); | 54 DISALLOW_COPY_AND_ASSIGN(PowerSaveBlockerStub); |
| 56 }; | 55 }; |
| 57 | 56 |
| 58 // Manages PowerSaveBlockerStub objects. Tests can instantiate this class | 57 // Manages PowerSaveBlockerStub objects. Tests can instantiate this class |
| 59 // to make PowerApiManager's calls to create PowerSaveBlockers record the | 58 // to make PowerAPI's calls to create PowerSaveBlockers record the |
| 60 // actions that would've been performed instead of actually blocking and | 59 // actions that would've been performed instead of actually blocking and |
| 61 // unblocking power management. | 60 // unblocking power management. |
| 62 class PowerSaveBlockerStubManager { | 61 class PowerSaveBlockerStubManager { |
| 63 public: | 62 public: |
| 64 explicit PowerSaveBlockerStubManager(content::BrowserContext* context) | 63 explicit PowerSaveBlockerStubManager(content::BrowserContext* context) |
| 65 : browser_context_(context), | 64 : browser_context_(context), |
| 66 weak_ptr_factory_(this) { | 65 weak_ptr_factory_(this) { |
| 67 // Use base::Unretained since callbacks with return values can't use | 66 // Use base::Unretained since callbacks with return values can't use |
| 68 // weak pointers. | 67 // weak pointers. |
| 69 PowerApiManager::Get(browser_context_)->SetCreateBlockerFunctionForTesting( | 68 PowerAPI::Get(browser_context_) |
| 70 base::Bind(&PowerSaveBlockerStubManager::CreateStub, | 69 ->SetCreateBlockerFunctionForTesting(base::Bind( |
| 71 base::Unretained(this))); | 70 &PowerSaveBlockerStubManager::CreateStub, base::Unretained(this))); |
| 72 } | 71 } |
| 73 | 72 |
| 74 ~PowerSaveBlockerStubManager() { | 73 ~PowerSaveBlockerStubManager() { |
| 75 PowerApiManager::Get(browser_context_)->SetCreateBlockerFunctionForTesting( | 74 PowerAPI::Get(browser_context_) |
| 76 PowerApiManager::CreateBlockerFunction()); | 75 ->SetCreateBlockerFunctionForTesting(PowerAPI::CreateBlockerFunction()); |
| 77 } | 76 } |
| 78 | 77 |
| 79 // Removes and returns the first item from |requests_|. Returns NONE if | 78 // Removes and returns the first item from |requests_|. Returns NONE if |
| 80 // |requests_| is empty. | 79 // |requests_| is empty. |
| 81 Request PopFirstRequest() { | 80 Request PopFirstRequest() { |
| 82 if (requests_.empty()) | 81 if (requests_.empty()) |
| 83 return NONE; | 82 return NONE; |
| 84 | 83 |
| 85 Request request = requests_.front(); | 84 Request request = requests_.front(); |
| 86 requests_.pop_front(); | 85 requests_.pop_front(); |
| (...skipping 33 matching lines...) Expand 10 before | Expand all | Expand 10 after Loading... |
| 120 // Requests in chronological order. | 119 // Requests in chronological order. |
| 121 std::deque<Request> requests_; | 120 std::deque<Request> requests_; |
| 122 | 121 |
| 123 base::WeakPtrFactory<PowerSaveBlockerStubManager> weak_ptr_factory_; | 122 base::WeakPtrFactory<PowerSaveBlockerStubManager> weak_ptr_factory_; |
| 124 | 123 |
| 125 DISALLOW_COPY_AND_ASSIGN(PowerSaveBlockerStubManager); | 124 DISALLOW_COPY_AND_ASSIGN(PowerSaveBlockerStubManager); |
| 126 }; | 125 }; |
| 127 | 126 |
| 128 } // namespace | 127 } // namespace |
| 129 | 128 |
| 130 class PowerApiTest : public ApiUnitTest { | 129 class PowerAPITest : public ApiUnitTest { |
| 131 public: | 130 public: |
| 132 void SetUp() override { | 131 void SetUp() override { |
| 133 ApiUnitTest::SetUp(); | 132 ApiUnitTest::SetUp(); |
| 134 manager_.reset(new PowerSaveBlockerStubManager(browser_context())); | 133 manager_.reset(new PowerSaveBlockerStubManager(browser_context())); |
| 135 } | 134 } |
| 136 | 135 |
| 137 void TearDown() override { | 136 void TearDown() override { |
| 138 manager_.reset(); | 137 manager_.reset(); |
| 139 ApiUnitTest::TearDown(); | 138 ApiUnitTest::TearDown(); |
| 140 } | 139 } |
| (...skipping 14 matching lines...) Expand all Loading... |
| 155 scoped_refptr<UIThreadExtensionFunction> function( | 154 scoped_refptr<UIThreadExtensionFunction> function( |
| 156 type == REQUEST ? | 155 type == REQUEST ? |
| 157 static_cast<UIThreadExtensionFunction*>( | 156 static_cast<UIThreadExtensionFunction*>( |
| 158 new PowerRequestKeepAwakeFunction) : | 157 new PowerRequestKeepAwakeFunction) : |
| 159 static_cast<UIThreadExtensionFunction*>( | 158 static_cast<UIThreadExtensionFunction*>( |
| 160 new PowerReleaseKeepAwakeFunction)); | 159 new PowerReleaseKeepAwakeFunction)); |
| 161 function->set_extension(extension); | 160 function->set_extension(extension); |
| 162 return api_test_utils::RunFunction(function.get(), args, browser_context()); | 161 return api_test_utils::RunFunction(function.get(), args, browser_context()); |
| 163 } | 162 } |
| 164 | 163 |
| 165 // Send a notification to PowerApiManager saying that |extension| has | 164 // Send a notification to PowerAPI saying that |extension| has |
| 166 // been unloaded. | 165 // been unloaded. |
| 167 void UnloadExtension(const extensions::Extension* extension) { | 166 void UnloadExtension(const extensions::Extension* extension) { |
| 168 PowerApiManager::Get(browser_context())->OnExtensionUnloaded( | 167 PowerAPI::Get(browser_context()) |
| 169 browser_context(), extension, UnloadedExtensionInfo::REASON_UNINSTALL); | 168 ->OnExtensionUnloaded(browser_context(), extension, |
| 169 UnloadedExtensionInfo::REASON_UNINSTALL); |
| 170 } | 170 } |
| 171 | 171 |
| 172 scoped_ptr<PowerSaveBlockerStubManager> manager_; | 172 scoped_ptr<PowerSaveBlockerStubManager> manager_; |
| 173 }; | 173 }; |
| 174 | 174 |
| 175 TEST_F(PowerApiTest, RequestAndRelease) { | 175 TEST_F(PowerAPITest, RequestAndRelease) { |
| 176 // Simulate an extension making and releasing a "display" request and a | 176 // Simulate an extension making and releasing a "display" request and a |
| 177 // "system" request. | 177 // "system" request. |
| 178 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); | 178 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); |
| 179 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 179 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 180 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 180 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 181 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); | 181 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); |
| 182 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 182 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 183 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 183 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 184 | 184 |
| 185 ASSERT_TRUE(CallFunction(REQUEST, kSystemArgs, extension())); | 185 ASSERT_TRUE(CallFunction(REQUEST, kSystemArgs, extension())); |
| 186 EXPECT_EQ(BLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); | 186 EXPECT_EQ(BLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); |
| 187 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 187 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 188 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); | 188 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); |
| 189 EXPECT_EQ(UNBLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); | 189 EXPECT_EQ(UNBLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); |
| 190 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 190 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 191 } | 191 } |
| 192 | 192 |
| 193 TEST_F(PowerApiTest, RequestWithoutRelease) { | 193 TEST_F(PowerAPITest, RequestWithoutRelease) { |
| 194 // Simulate an extension calling requestKeepAwake() without calling | 194 // Simulate an extension calling requestKeepAwake() without calling |
| 195 // releaseKeepAwake(). The override should be automatically removed when | 195 // releaseKeepAwake(). The override should be automatically removed when |
| 196 // the extension is unloaded. | 196 // the extension is unloaded. |
| 197 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); | 197 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); |
| 198 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 198 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 199 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 199 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 200 | 200 |
| 201 UnloadExtension(extension()); | 201 UnloadExtension(extension()); |
| 202 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 202 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 203 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 203 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 204 } | 204 } |
| 205 | 205 |
| 206 TEST_F(PowerApiTest, ReleaseWithoutRequest) { | 206 TEST_F(PowerAPITest, ReleaseWithoutRequest) { |
| 207 // Simulate an extension calling releaseKeepAwake() without having | 207 // Simulate an extension calling releaseKeepAwake() without having |
| 208 // calling requestKeepAwake() earlier. The call should be ignored. | 208 // calling requestKeepAwake() earlier. The call should be ignored. |
| 209 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); | 209 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); |
| 210 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 210 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 211 } | 211 } |
| 212 | 212 |
| 213 TEST_F(PowerApiTest, UpgradeRequest) { | 213 TEST_F(PowerAPITest, UpgradeRequest) { |
| 214 // Simulate an extension calling requestKeepAwake("system") and then | 214 // Simulate an extension calling requestKeepAwake("system") and then |
| 215 // requestKeepAwake("display"). When the second call is made, a | 215 // requestKeepAwake("display"). When the second call is made, a |
| 216 // display-sleep-blocking request should be made before the initial | 216 // display-sleep-blocking request should be made before the initial |
| 217 // app-suspension-blocking request is released. | 217 // app-suspension-blocking request is released. |
| 218 ASSERT_TRUE(CallFunction(REQUEST, kSystemArgs, extension())); | 218 ASSERT_TRUE(CallFunction(REQUEST, kSystemArgs, extension())); |
| 219 EXPECT_EQ(BLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); | 219 EXPECT_EQ(BLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); |
| 220 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 220 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 221 | 221 |
| 222 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); | 222 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); |
| 223 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 223 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 224 EXPECT_EQ(UNBLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); | 224 EXPECT_EQ(UNBLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); |
| 225 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 225 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 226 | 226 |
| 227 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); | 227 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); |
| 228 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 228 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 229 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 229 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 230 } | 230 } |
| 231 | 231 |
| 232 TEST_F(PowerApiTest, DowngradeRequest) { | 232 TEST_F(PowerAPITest, DowngradeRequest) { |
| 233 // Simulate an extension calling requestKeepAwake("display") and then | 233 // Simulate an extension calling requestKeepAwake("display") and then |
| 234 // requestKeepAwake("system"). When the second call is made, an | 234 // requestKeepAwake("system"). When the second call is made, an |
| 235 // app-suspension-blocking request should be made before the initial | 235 // app-suspension-blocking request should be made before the initial |
| 236 // display-sleep-blocking request is released. | 236 // display-sleep-blocking request is released. |
| 237 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); | 237 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); |
| 238 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 238 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 239 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 239 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 240 | 240 |
| 241 ASSERT_TRUE(CallFunction(REQUEST, kSystemArgs, extension())); | 241 ASSERT_TRUE(CallFunction(REQUEST, kSystemArgs, extension())); |
| 242 EXPECT_EQ(BLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); | 242 EXPECT_EQ(BLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); |
| 243 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 243 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 244 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 244 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 245 | 245 |
| 246 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); | 246 ASSERT_TRUE(CallFunction(RELEASE, kEmptyArgs, extension())); |
| 247 EXPECT_EQ(UNBLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); | 247 EXPECT_EQ(UNBLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); |
| 248 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 248 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 249 } | 249 } |
| 250 | 250 |
| 251 TEST_F(PowerApiTest, MultipleExtensions) { | 251 TEST_F(PowerAPITest, MultipleExtensions) { |
| 252 // Simulate an extension blocking the display from sleeping. | 252 // Simulate an extension blocking the display from sleeping. |
| 253 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); | 253 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); |
| 254 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 254 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 255 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 255 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 256 | 256 |
| 257 // Create a second extension that blocks system suspend. No additional | 257 // Create a second extension that blocks system suspend. No additional |
| 258 // PowerSaveBlocker is needed; the blocker from the first extension | 258 // PowerSaveBlocker is needed; the blocker from the first extension |
| 259 // already covers the behavior requested by the second extension. | 259 // already covers the behavior requested by the second extension. |
| 260 scoped_refptr<Extension> extension2(test_util::CreateEmptyExtension("id2")); | 260 scoped_refptr<Extension> extension2(test_util::CreateEmptyExtension("id2")); |
| 261 ASSERT_TRUE(CallFunction(REQUEST, kSystemArgs, extension2.get())); | 261 ASSERT_TRUE(CallFunction(REQUEST, kSystemArgs, extension2.get())); |
| 262 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 262 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 263 | 263 |
| 264 // When the first extension is unloaded, a new app-suspension blocker | 264 // When the first extension is unloaded, a new app-suspension blocker |
| 265 // should be created before the display-sleep blocker is destroyed. | 265 // should be created before the display-sleep blocker is destroyed. |
| 266 UnloadExtension(extension()); | 266 UnloadExtension(extension()); |
| 267 EXPECT_EQ(BLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); | 267 EXPECT_EQ(BLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); |
| 268 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 268 EXPECT_EQ(UNBLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 269 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 269 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 270 | 270 |
| 271 // Make the first extension block display-sleep again. | 271 // Make the first extension block display-sleep again. |
| 272 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); | 272 ASSERT_TRUE(CallFunction(REQUEST, kDisplayArgs, extension())); |
| 273 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); | 273 EXPECT_EQ(BLOCK_DISPLAY_SLEEP, manager_->PopFirstRequest()); |
| 274 EXPECT_EQ(UNBLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); | 274 EXPECT_EQ(UNBLOCK_APP_SUSPENSION, manager_->PopFirstRequest()); |
| 275 EXPECT_EQ(NONE, manager_->PopFirstRequest()); | 275 EXPECT_EQ(NONE, manager_->PopFirstRequest()); |
| 276 } | 276 } |
| 277 | 277 |
| 278 } // namespace extensions | 278 } // namespace extensions |
| OLD | NEW |