Prototype version of the first-run dialog for Windows 8

chrome/browser/custom_handlers/protocol_handler_registry_unittest.cc

 // Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/custom_handlers/protocol_handler_registry.h"
 
 #include <set>
 
 #include "base/memory/scoped_ptr.h"
 #include "base/message_loop.h"
@@ skipping 73 matching lines @@
 
 class FakeClientObserver
     : public ProtocolHandlerRegistry::DefaultClientObserver {
  public:
   FakeClientObserver(ProtocolHandlerRegistry* registry,
                      FakeDelegate* registry_delegate)
       : ProtocolHandlerRegistry::DefaultClientObserver(registry),
         delegate_(registry_delegate) {}
 
   virtual void SetDefaultWebClientUIState(
-      ShellIntegration::DefaultWebClientUIState state) {
+      ShellIntegration::DefaultWebClientUIState state,
+      ShellIntegration::SetDefaultWebClientResult call_result) {
     ProtocolHandlerRegistry::DefaultClientObserver::SetDefaultWebClientUIState(
-        state);
+        state, call_result);
     if (state == ShellIntegration::STATE_IS_DEFAULT) {
       delegate_->FakeRegisterWithOS(worker_->protocol());
     }
     if (state != ShellIntegration::STATE_PROCESSING) {
       MessageLoop::current()->Quit();
     }
   }
 
  private:
   FakeDelegate* delegate_;
@@ skipping 12 matching lines @@
   virtual ~FakeProtocolClientWorker() {}
 
   virtual ShellIntegration::DefaultWebClientState CheckIsDefault() {
     if (force_failure_) {
       return ShellIntegration::NOT_DEFAULT_WEB_CLIENT;
     } else {
       return ShellIntegration::IS_DEFAULT_WEB_CLIENT;
     }
   }
 
-  virtual void SetAsDefault(bool interactive_permitted) {}
+  virtual ShellIntegration::SetDefaultWebClientResult
+      SetAsDefault(bool interactive_permitted) {
+    return ShellIntegration::RESULT_SET_DEFAULT_OK;
+  }
 
  private:
   bool force_failure_;
 };
 
 ProtocolHandlerRegistry::DefaultClientObserver*
     FakeDelegate::CreateShellObserver(ProtocolHandlerRegistry* registry) {
   return new FakeClientObserver(registry, this);
 }
 
@@ skipping 693 matching lines @@
   ASSERT_EQ(ph3.url().GetOrigin() == ph2.url().GetOrigin(),
       ph3.IsSameOrigin(ph2));
 }
 
 TEST_F(ProtocolHandlerRegistryTest, TestInstallDefaultHandler) {
   ReloadProtocolHandlerRegistryAndInstallDefaultHandler();
   std::vector<std::string> protocols;
   registry()->GetRegisteredProtocols(&protocols);
   ASSERT_EQ(static_cast<size_t>(1), protocols.size());
 }