content/test/mock_ssl_host_state_delegate.h

// Copyright (c) 2015 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 CONTENT_TEST_MOCK_SSL_HOST_STATE_DELEGATE_H_
#define CONTENT_TEST_MOCK_SSL_HOST_STATE_DELEGATE_H_

#include "content/public/browser/ssl_host_state_delegate.h"

namespace content {

class MockSSLHostStateDelegate : public SSLHostStateDelegate {
 public:
  MockSSLHostStateDelegate();
  ~MockSSLHostStateDelegate() override;

  void AllowCert(const std::string& host,
                 const net::X509Certificate& cert,
                 int error,
                 WebContents* web_contents) override;

  void Clear(
      base::RepeatingCallback<bool(const std::string&)> host_filter) override;

  CertJudgment QueryPolicy(const std::string& host,
                           const net::X509Certificate& cert,
                           int error,
                           WebContents* web_contents) override;

  void HostRanInsecureContent(const std::string& host,
                              int child_id,
                              InsecureContentType content_type) override;

  bool DidHostRunInsecureContent(const std::string& host,
                                 int child_id,
                                 InsecureContentType content_type) override;

  void AllowHttpForHost(const std::string& host) override;

  bool IsHttpAllowedForHost(const std::string& host) override;

  void RevokeUserAllowExceptions(const std::string& host) override;

  bool HasAllowException(const std::string& host,
                         WebContents* web_contents) override;

 private:
  std::set<std::string> exceptions_;
  std::set<std::string> hosts_ran_insecure_content_;
  std::set<std::string> allow_http_hosts_;
};

}  // namespace content

#endif  // CONTENT_TEST_MOCK_SSL_HOST_STATE_DELEGATE_H_