Upstream most of ios/web/net

ios/web/public/test/test_web_client.h

+// 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.
+
+#ifndef WEB_TEST_WEB_PUBLIC_TEST_TEST_WEB_CLIENT_H_
+#define WEB_TEST_WEB_PUBLIC_TEST_TEST_WEB_CLIENT_H_
+
+#import <Foundation/Foundation.h>
+
+#include "base/compiler_specific.h"
+#include "base/mac/scoped_nsobject.h"
+#include "base/memory/scoped_ptr.h"
+#include "ios/web/public/web_client.h"
+
+namespace web {
+
+class WebViewFactory;
+
+// A WebClient used for testing purposes.
+class TestWebClient : public web::WebClient {
+ public:
+  TestWebClient();
+  ~TestWebClient() override;
+  // WebClient implementation.
+  std::string GetUserAgent(bool is_desktop_user_agent) const override;
+  NSString* GetEarlyPageScript(web::WebViewType web_view_type) const override;
+
+  // Changes the user agent for testing purposes. Passing true
+  // for |is_desktop_user_agent| affects the result of GetUserAgent(true) call.
+  // Passing false affects the result of GetUserAgent(false).
+  void SetUserAgent(const std::string& user_agent, bool is_desktop_user_agent);
+
+  // Changes Early Page Script for testing purposes.
+  void SetEarlyPageScript(NSString* page_script,
+                          web::WebViewType web_view_type);
+
+ private:
+  std::string user_agent_;
+  std::string desktop_user_agent_;
+  base::scoped_nsobject<NSMutableDictionary> early_page_scripts_;
+};
+
+}  // namespace web
+
+#endif // WEB_TEST_WEB_PUBLIC_TEST_TEST_WEB_CLIENT_H_