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_; |