blob: d4b1e5c333ca6cbeef260c4cb0d9c3ed236f7717 [file] [log] [blame]
// 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/test_web_client.h"
#include "base/strings/sys_string_conversions.h"
namespace web {
TestWebClient::TestWebClient()
: early_page_scripts_([[NSMutableDictionary alloc] init]) {
}
TestWebClient::~TestWebClient() {
}
std::string TestWebClient::GetUserAgent(bool desktop_user_agent) const {
return desktop_user_agent ? desktop_user_agent_ : user_agent_;
}
void TestWebClient::SetUserAgent(const std::string& user_agent,
bool is_desktop_user_agent) {
if (is_desktop_user_agent)
desktop_user_agent_ = user_agent;
else
user_agent_ = user_agent;
}
NSString* TestWebClient::GetEarlyPageScript(
web::WebViewType web_view_type) const {
NSString* result = [early_page_scripts_ objectForKey:@(web_view_type)];
return result ? result : @"";
}
void TestWebClient::SetEarlyPageScript(NSString* page_script,
web::WebViewType web_view_type) {
[early_page_scripts_ setObject:page_script forKey:@(web_view_type)];
}
} // namespace web