src/third_party/WebKit/Source/modules/navigatorcontentutils/testing/NavigatorContentUtilsClientMock.h

   1 // Copyright 2014 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #ifndef NavigatorContentUtilsClientMock_h
   6 #define NavigatorContentUtilsClientMock_h
   7
   8 #include "modules/navigatorcontentutils/NavigatorContentUtilsClient.h"
   9 #include "wtf/HashSet.h"
  10 #include "wtf/text/WTFString.h"
  11
  12 namespace blink {
  13
  14 class KURL;
  15
  16 // Provides a mock object for the navigatorcontentutils client.
  17 class NavigatorContentUtilsClientMock : public NavigatorContentUtilsClient {
  18 public:
  19     NavigatorContentUtilsClientMock() { }
  20     virtual ~NavigatorContentUtilsClientMock() { }
  21
  22     virtual void registerProtocolHandler(const String& scheme, const KURL& baseURL, const KURL&, const String& title);
  23
  24     virtual CustomHandlersState isProtocolHandlerRegistered(const String& scheme, const KURL& baseURL, const KURL&);
  25     virtual void unregisterProtocolHandler(const String& scheme, const KURL& baseURL, const KURL&);
  26
  27 private:
  28     typedef struct {
  29         String scheme;
  30         KURL baseURL;
  31         KURL url;
  32         String title;
  33     } ProtocolInfo;
  34
  35     typedef HashMap<String, ProtocolInfo> RegisteredProtocolMap;
  36     RegisteredProtocolMap m_protocolMap;
  37 };
  38
  39 } // namespace blink
  40
  41 #endif // NavigatorContentUtilsClientMock_h