| Index: chrome/browser/custom_handlers/protocol_handler_registry_unittest.cc
|
| diff --git a/chrome/browser/custom_handlers/protocol_handler_registry_unittest.cc b/chrome/browser/custom_handlers/protocol_handler_registry_unittest.cc
|
| index f246291f46b1c3ce59aa09842efc656f7e161598..7d89c81790a0cfd5ab699d6bfae70972ae5c7b07 100644
|
| --- a/chrome/browser/custom_handlers/protocol_handler_registry_unittest.cc
|
| +++ b/chrome/browser/custom_handlers/protocol_handler_registry_unittest.cc
|
| @@ -226,15 +226,24 @@ class FakeProtocolClientWorker
|
| private:
|
| ~FakeProtocolClientWorker() override {}
|
|
|
| - ShellIntegration::DefaultWebClientState CheckIsDefault() override {
|
| - if (force_failure_) {
|
| - return ShellIntegration::NOT_DEFAULT;
|
| - } else {
|
| - return ShellIntegration::IS_DEFAULT;
|
| - }
|
| + void CheckIsDefault() override {
|
| + ShellIntegration::DefaultWebClientState state =
|
| + ShellIntegration::IS_DEFAULT;
|
| + if (force_failure_)
|
| + state = ShellIntegration::NOT_DEFAULT;
|
| +
|
| + BrowserThread::PostTask(
|
| + BrowserThread::UI, FROM_HERE,
|
| + base::Bind(&FakeProtocolClientWorker::OnCheckIsDefaultComplete, this,
|
| + state));
|
| }
|
|
|
| - bool SetAsDefault(bool interactive_permitted) override { return true; }
|
| + void SetAsDefault(bool interactive_permitted) override {
|
| + BrowserThread::PostTask(
|
| + BrowserThread::UI, FROM_HERE,
|
| + base::Bind(&FakeProtocolClientWorker::OnSetAsDefaultAttemptComplete,
|
| + this, AttemptResult::SUCCESS));
|
| + }
|
|
|
| private:
|
| bool force_failure_;
|
|
|