web/public/test/fakes/test_web_client.mm

// Copyright 2014 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.

#import "ios/web/public/test/fakes/test_web_client.h"

#include "base/logging.h"
#include "ios/web/test/test_url_constants.h"
#include "ui/base/resource/resource_bundle.h"
#include "url/gurl.h"

#if !defined(__has_feature) || !__has_feature(objc_arc)
#error "This file requires ARC support."
#endif

namespace web {

TestWebClient::TestWebClient()
    : last_cert_error_code_(0),
      last_cert_error_overridable_(true),
      is_slim_navigation_manager_enabled_(false) {}

TestWebClient::~TestWebClient() {}

void TestWebClient::AddAdditionalSchemes(
    std::vector<url::SchemeWithType>* additional_standard_schemes) const {
  url::SchemeWithType web_ui_scheme = {kTestWebUIScheme,
                                       url::SCHEME_WITHOUT_PORT};
  additional_standard_schemes->push_back(web_ui_scheme);

  url::SchemeWithType native_scheme = {kTestNativeContentScheme,
                                       url::SCHEME_WITHOUT_PORT};
  additional_standard_schemes->push_back(native_scheme);
}

bool TestWebClient::IsAppSpecificURL(const GURL& url) const {
  return url.SchemeIs(kTestWebUIScheme) ||
         url.SchemeIs(kTestNativeContentScheme);
}

base::RefCountedMemory* TestWebClient::GetDataResourceBytes(
    int resource_id) const {
  if (!ResourceBundle::HasSharedInstance())
    return nullptr;
  return ResourceBundle::GetSharedInstance().LoadDataResourceBytes(resource_id);
}

NSString* TestWebClient::GetEarlyPageScript(BrowserState* browser_state) const {
  return early_page_script_ ? early_page_script_ : @"";
}

void TestWebClient::SetEarlyPageScript(NSString* page_script) {
  early_page_script_ = [page_script copy];
}

void TestWebClient::AllowCertificateError(
    WebState* web_state,
    int cert_error,
    const net::SSLInfo& ssl_info,
    const GURL& request_url,
    bool overridable,
    const base::Callback<void(bool)>& callback) {
  last_cert_error_code_ = cert_error;
  last_cert_error_ssl_info_ = ssl_info;
  last_cert_error_request_url_ = request_url;
  last_cert_error_overridable_ = overridable;

  callback.Run(false);
}

bool TestWebClient::IsSlimNavigationManagerEnabled() const {
  return is_slim_navigation_manager_enabled_;
}

void TestWebClient::SetIsSlimNavigationManager(bool flag) {
  is_slim_navigation_manager_enabled_ = flag;
}

}  // namespace web