// Copyright 2019 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. #ifndef COMPONENTS_CUSTOM_HANDLERS_TEST_PROTOCOL_HANDLER_REGISTRY_DELEGATE_H_ #define COMPONENTS_CUSTOM_HANDLERS_TEST_PROTOCOL_HANDLER_REGISTRY_DELEGATE_H_ #include #include #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 registered_protocols_; std::set os_registered_protocols_; bool force_os_failure_{false}; }; } // namespace custom_handlers #endif // COMPONENTS_CUSTOM_HANDLERS_TEST_PROTOCOL_HANDLER_REGISTRY_DELEGATE_H_