components/custom_handlers/test_protocol_handler_registry_delegate.h

// Copyright 2019 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.

#ifndef COMPONENTS_CUSTOM_HANDLERS_TEST_PROTOCOL_HANDLER_REGISTRY_DELEGATE_H_
#define COMPONENTS_CUSTOM_HANDLERS_TEST_PROTOCOL_HANDLER_REGISTRY_DELEGATE_H_

#include <set>
#include <string>

#include "components/custom_handlers/protocol_handler_registry.h"

namespace custom_handlers {

// A test ProtocolHandlerRegistry::Delegate implementation that keeps track of
// registered protocols and doesn't change any OS settings.
class TestProtocolHandlerRegistryDelegate
    : public custom_handlers::ProtocolHandlerRegistry::Delegate {
 public:
  TestProtocolHandlerRegistryDelegate();
  ~TestProtocolHandlerRegistryDelegate() override;

  TestProtocolHandlerRegistryDelegate(
      const TestProtocolHandlerRegistryDelegate& other) = delete;
  TestProtocolHandlerRegistryDelegate& operator=(
      const TestProtocolHandlerRegistryDelegate& other) = delete;

  // ProtocolHandlerRegistry::Delegate:
  void RegisterExternalHandler(const std::string& protocol) override;
  void DeregisterExternalHandler(const std::string& protocol) override;
  bool IsExternalHandlerRegistered(const std::string& protocol) override;
  void RegisterWithOSAsDefaultClient(const std::string& protocol,
                                     DefaultClientCallback callback) override;
  void CheckDefaultClientWithOS(const std::string& protocol,
                                DefaultClientCallback callback) override;
  bool ShouldRemoveHandlersNotInOS() override;

  bool IsFakeRegistered(const std::string& protocol);
  bool IsFakeRegisteredWithOS(const std::string& protocol);

  void set_force_os_failure(bool force) { force_os_failure_ = force; }
  bool force_os_failure() { return force_os_failure_; }

  void Reset();

 private:
  // Holds registered protocols.
  std::set<std::string> registered_protocols_;
  std::set<std::string> os_registered_protocols_;
  bool force_os_failure_{false};
};

}  // namespace custom_handlers

#endif  // COMPONENTS_CUSTOM_HANDLERS_TEST_PROTOCOL_HANDLER_REGISTRY_DELEGATE_H_