ios/web/webui/crw_web_ui_manager_unittest.mm - chromium/src - Git at Google

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

 #import "ios/web/webui/crw_web_ui_manager.h"

 #include <memory>

 #include "base/files/file_path.h"
 #include "base/files/file_util.h"
 #include "base/logging.h"
 #import "base/mac/scoped_nsobject.h"
 #include "base/memory/ptr_util.h"
 #include "base/message_loop/message_loop.h"
 #include "base/path_service.h"
 #include "base/strings/stringprintf.h"
 #import "base/strings/sys_string_conversions.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/values.h"
 #include "ios/web/public/test/scoped_testing_web_client.h"
 #include "ios/web/public/test/test_browser_state.h"
 #import "ios/web/public/test/test_web_client.h"
 #include "ios/web/public/test/web_test.h"
 #include "ios/web/web_state/web_state_impl.h"
 #import "ios/web/webui/crw_web_ui_page_builder.h"
 #include "ios/web/webui/url_fetcher_block_adapter.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/gtest_mac.h"

 namespace web {

 // Path for test favicon file.
 const char kFaviconPath[] = "ios/web/test/data/testfavicon.png";
 // URL for mock WebUI page.
 const char kTestWebUIUrl[] = "testwebui://test/";
 // URL for mock favicon.
 const char kFaviconUrl[] = "testwebui://favicon/";
 // Name of test Mojo module.
 const char kMojoModuleName[] = "test-mojo-module";
 // HTML for mock WebUI page.
 NSString* kHtml = @"<html>Hello World</html>";
 // Mojo module for WebUI page.
 NSString* kMojoModule = @"service_provider.connect('Test');";

 // Mock of WebStateImpl to check that LoadHtml and ExecuteJavaScript are called
 // as expected.
 class MockWebStateImpl : public WebStateImpl {
  public:
   MockWebStateImpl(BrowserState* browser_state)
       : WebStateImpl(browser_state), last_committed_url_(kTestWebUIUrl) {}
   MOCK_METHOD2(LoadWebUIHtml,
                void(const base::string16& html, const GURL& url));
   MOCK_METHOD1(ExecuteJavaScript, void(const base::string16& javascript));
   const GURL& GetLastCommittedURL() const override {
     return last_committed_url_;
   }

  private:
   GURL last_committed_url_;
 };

 // Mock of URLFetcherBlockAdapter to provide mock resources.
 class MockURLFetcherBlockAdapter : public URLFetcherBlockAdapter {
  public:
   MockURLFetcherBlockAdapter(
       const GURL& url,
       net::URLRequestContextGetter* request_context,
       URLFetcherBlockAdapterCompletion completion_handler)
       : URLFetcherBlockAdapter(url, request_context, completion_handler),
         url_(url),
         completion_handler_(completion_handler) {}

   void Start() override {
     if (url_.spec() == kFaviconUrl) {
       base::FilePath favicon_path;
       ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &favicon_path));
       favicon_path = favicon_path.AppendASCII(kFaviconPath);
       NSData* favicon = [NSData
           dataWithContentsOfFile:base::SysUTF8ToNSString(favicon_path.value())];
       completion_handler_.get()(favicon, this);
     } else if (url_.path().find(kMojoModuleName) != std::string::npos) {
       completion_handler_.get()(
           [kMojoModule dataUsingEncoding:NSUTF8StringEncoding], this);

     } else if (url_.scheme().find("test") != std::string::npos) {
       completion_handler_.get()([kHtml dataUsingEncoding:NSUTF8StringEncoding],
                                 this);
     } else {
       NOTREACHED();
     }
   }

  private:
   // The URL to fetch.
   const GURL url_;
   // Callback for resource load.
   base::mac::ScopedBlock<URLFetcherBlockAdapterCompletion> completion_handler_;
 };

 }  // namespace web

 // Subclass of CRWWebUIManager for testing.
 @interface CRWTestWebUIManager : CRWWebUIManager
 // Use mock URLFetcherBlockAdapter.
 - (std::unique_ptr<web::URLFetcherBlockAdapter>)
     fetcherForURL:(const GURL&)URL
 completionHandler:(web::URLFetcherBlockAdapterCompletion)handler;
 @end

 @implementation CRWTestWebUIManager
 - (std::unique_ptr<web::URLFetcherBlockAdapter>)
     fetcherForURL:(const GURL&)URL
 completionHandler:(web::URLFetcherBlockAdapterCompletion)handler {
   return std::unique_ptr<web::URLFetcherBlockAdapter>(
       new web::MockURLFetcherBlockAdapter(URL, nil, handler));
 }
 @end

 namespace web {

 // Test fixture for testing CRWWebUIManager
 class CRWWebUIManagerTest : public web::WebTest {
  protected:
   void SetUp() override {
     PlatformTest::SetUp();
     test_browser_state_.reset(new TestBrowserState());
     web_state_impl_.reset(new MockWebStateImpl(test_browser_state_.get()));
     web_ui_manager_.reset(
         [[CRWTestWebUIManager alloc] initWithWebState:web_state_impl_.get()]);
   }

   // TestBrowserState for creation of WebStateImpl.
   std::unique_ptr<TestBrowserState> test_browser_state_;
   // MockWebStateImpl for detection of LoadHtml and EvaluateJavaScriptAync
   // calls.
   std::unique_ptr<MockWebStateImpl> web_state_impl_;
   // WebUIManager for testing.
   base::scoped_nsobject<CRWTestWebUIManager> web_ui_manager_;
 };

 // Tests that CRWWebUIManager observes provisional navigation and invokes an
 // HTML load in web state.
 TEST_F(CRWWebUIManagerTest, LoadWebUI) {
   base::string16 html(base::SysNSStringToUTF16(kHtml));
   GURL url(kTestWebUIUrl);
   EXPECT_CALL(*web_state_impl_, LoadWebUIHtml(html, url));
   web_state_impl_->OnProvisionalNavigationStarted(url);
 }

 // Tests that CRWWebUIManager responds to OnScriptCommandReceieved call and runs
 // JavaScript to set favicon background.
 TEST_F(CRWWebUIManagerTest, HandleFaviconRequest) {
   GURL test_url(kTestWebUIUrl);
   std::string favicon_url_string(kFaviconUrl);

   // Create mock JavaScript message to request favicon.
   base::ListValue* arguments(new base::ListValue());
   arguments->AppendString(favicon_url_string);
   std::unique_ptr<base::DictionaryValue> message(new base::DictionaryValue());
   message->SetString("message", "webui.requestFavicon");
   message->Set("arguments", arguments);

   // Create expected JavaScript to call.
   base::FilePath favicon_path;
   ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &favicon_path));
   favicon_path = favicon_path.AppendASCII(kFaviconPath);
   NSData* expected_data = [NSData
       dataWithContentsOfFile:base::SysUTF8ToNSString(favicon_path.value())];
   base::string16 expected_javascript = base::SysNSStringToUTF16([NSString
       stringWithFormat:
           @"chrome.setFaviconBackground('%s', 'data:image/png;base64,%@');",
           favicon_url_string.c_str(),
           [expected_data base64EncodedStringWithOptions:0]]);

   EXPECT_CALL(*web_state_impl_, ExecuteJavaScript(expected_javascript));
   web_state_impl_->OnScriptCommandReceived("webui.requestFavicon", *message,
                                            test_url, false);
 }

 // Tests that CRWWebUIManager responds to OnScriptCommandReceieved call and runs
 // JavaScript to fetch a mojo resource.
 TEST_F(CRWWebUIManagerTest, HandleLoadMojoRequest) {
   // Create mock JavaScript message to request a mojo resource.
   std::unique_ptr<base::ListValue> arguments(new base::ListValue());
   arguments->AppendString(kMojoModuleName);
   const char kTestLoadId[] = "test-load-id";
   arguments->AppendString(kTestLoadId);
   base::DictionaryValue message;
   message.SetString("message", "webui.loadMojo");
   message.Set("arguments", std::move(arguments));

   // Create expected JavaScript to call.
   std::string expected_javascript = base::StringPrintf(
       "%s__crWeb.webUIModuleLoadNotifier.moduleLoadCompleted(\"%s\", \"%s\");",
       base::SysNSStringToUTF8(kMojoModule).c_str(), kMojoModuleName,
       kTestLoadId);

   EXPECT_CALL(*web_state_impl_,
               ExecuteJavaScript(base::UTF8ToUTF16(expected_javascript)));
   web_state_impl_->OnScriptCommandReceived("webui.loadMojo", message,
                                            GURL(kTestWebUIUrl), false);
 }
 }  // namespace web
	// Copyright 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.

	#import "ios/web/webui/crw_web_ui_manager.h"

	#include <memory>

	#include "base/files/file_path.h"
	#include "base/files/file_util.h"
	#include "base/logging.h"
	#import "base/mac/scoped_nsobject.h"
	#include "base/memory/ptr_util.h"
	#include "base/message_loop/message_loop.h"
	#include "base/path_service.h"
	#include "base/strings/stringprintf.h"
	#import "base/strings/sys_string_conversions.h"
	#include "base/strings/utf_string_conversions.h"
	#include "base/values.h"
	#include "ios/web/public/test/scoped_testing_web_client.h"
	#include "ios/web/public/test/test_browser_state.h"
	#import "ios/web/public/test/test_web_client.h"
	#include "ios/web/public/test/web_test.h"
	#include "ios/web/web_state/web_state_impl.h"
	#import "ios/web/webui/crw_web_ui_page_builder.h"
	#include "ios/web/webui/url_fetcher_block_adapter.h"
	#include "testing/gmock/include/gmock/gmock.h"
	#include "testing/gtest/include/gtest/gtest.h"
	#include "testing/gtest_mac.h"

	namespace web {

	// Path for test favicon file.
	const char kFaviconPath[] = "ios/web/test/data/testfavicon.png";
	// URL for mock WebUI page.
	const char kTestWebUIUrl[] = "testwebui://test/";
	// URL for mock favicon.
	const char kFaviconUrl[] = "testwebui://favicon/";
	// Name of test Mojo module.
	const char kMojoModuleName[] = "test-mojo-module";
	// HTML for mock WebUI page.
	NSString* kHtml = @"<html>Hello World</html>";
	// Mojo module for WebUI page.
	NSString* kMojoModule = @"service_provider.connect('Test');";

	// Mock of WebStateImpl to check that LoadHtml and ExecuteJavaScript are called
	// as expected.
	class MockWebStateImpl : public WebStateImpl {
	public:
	MockWebStateImpl(BrowserState* browser_state)
	: WebStateImpl(browser_state), last_committed_url_(kTestWebUIUrl) {}
	MOCK_METHOD2(LoadWebUIHtml,
	void(const base::string16& html, const GURL& url));
	MOCK_METHOD1(ExecuteJavaScript, void(const base::string16& javascript));
	const GURL& GetLastCommittedURL() const override {
	return last_committed_url_;
	}

	private:
	GURL last_committed_url_;
	};

	// Mock of URLFetcherBlockAdapter to provide mock resources.
	class MockURLFetcherBlockAdapter : public URLFetcherBlockAdapter {
	public:
	MockURLFetcherBlockAdapter(
	const GURL& url,
	net::URLRequestContextGetter* request_context,
	URLFetcherBlockAdapterCompletion completion_handler)
	: URLFetcherBlockAdapter(url, request_context, completion_handler),
	url_(url),
	completion_handler_(completion_handler) {}

	void Start() override {
	if (url_.spec() == kFaviconUrl) {
	base::FilePath favicon_path;
	ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &favicon_path));
	favicon_path = favicon_path.AppendASCII(kFaviconPath);
	NSData* favicon = [NSData
	dataWithContentsOfFile:base::SysUTF8ToNSString(favicon_path.value())];
	completion_handler_.get()(favicon, this);
	} else if (url_.path().find(kMojoModuleName) != std::string::npos) {
	completion_handler_.get()(
	[kMojoModule dataUsingEncoding:NSUTF8StringEncoding], this);

	} else if (url_.scheme().find("test") != std::string::npos) {
	completion_handler_.get()([kHtml dataUsingEncoding:NSUTF8StringEncoding],
	this);
	} else {
	NOTREACHED();
	}
	}

	private:
	// The URL to fetch.
	const GURL url_;
	// Callback for resource load.
	base::mac::ScopedBlock<URLFetcherBlockAdapterCompletion> completion_handler_;
	};

	} // namespace web

	// Subclass of CRWWebUIManager for testing.
	@interface CRWTestWebUIManager : CRWWebUIManager
	// Use mock URLFetcherBlockAdapter.
	- (std::unique_ptr<web::URLFetcherBlockAdapter>)
	fetcherForURL:(const GURL&)URL
	completionHandler:(web::URLFetcherBlockAdapterCompletion)handler;
	@end

	@implementation CRWTestWebUIManager
	- (std::unique_ptr<web::URLFetcherBlockAdapter>)
	fetcherForURL:(const GURL&)URL
	completionHandler:(web::URLFetcherBlockAdapterCompletion)handler {
	return std::unique_ptr<web::URLFetcherBlockAdapter>(
	new web::MockURLFetcherBlockAdapter(URL, nil, handler));
	}
	@end

	namespace web {

	// Test fixture for testing CRWWebUIManager
	class CRWWebUIManagerTest : public web::WebTest {
	protected:
	void SetUp() override {
	PlatformTest::SetUp();
	test_browser_state_.reset(new TestBrowserState());
	web_state_impl_.reset(new MockWebStateImpl(test_browser_state_.get()));
	web_ui_manager_.reset(
	[[CRWTestWebUIManager alloc] initWithWebState:web_state_impl_.get()]);
	}

	// TestBrowserState for creation of WebStateImpl.
	std::unique_ptr<TestBrowserState> test_browser_state_;
	// MockWebStateImpl for detection of LoadHtml and EvaluateJavaScriptAync
	// calls.
	std::unique_ptr<MockWebStateImpl> web_state_impl_;
	// WebUIManager for testing.
	base::scoped_nsobject<CRWTestWebUIManager> web_ui_manager_;
	};

	// Tests that CRWWebUIManager observes provisional navigation and invokes an
	// HTML load in web state.
	TEST_F(CRWWebUIManagerTest, LoadWebUI) {
	base::string16 html(base::SysNSStringToUTF16(kHtml));
	GURL url(kTestWebUIUrl);
	EXPECT_CALL(*web_state_impl_, LoadWebUIHtml(html, url));
	web_state_impl_->OnProvisionalNavigationStarted(url);
	}

	// Tests that CRWWebUIManager responds to OnScriptCommandReceieved call and runs
	// JavaScript to set favicon background.
	TEST_F(CRWWebUIManagerTest, HandleFaviconRequest) {
	GURL test_url(kTestWebUIUrl);
	std::string favicon_url_string(kFaviconUrl);

	// Create mock JavaScript message to request favicon.
	base::ListValue* arguments(new base::ListValue());
	arguments->AppendString(favicon_url_string);
	std::unique_ptr<base::DictionaryValue> message(new base::DictionaryValue());
	message->SetString("message", "webui.requestFavicon");
	message->Set("arguments", arguments);

	// Create expected JavaScript to call.
	base::FilePath favicon_path;
	ASSERT_TRUE(PathService::Get(base::DIR_SOURCE_ROOT, &favicon_path));
	favicon_path = favicon_path.AppendASCII(kFaviconPath);
	NSData* expected_data = [NSData
	dataWithContentsOfFile:base::SysUTF8ToNSString(favicon_path.value())];
	base::string16 expected_javascript = base::SysNSStringToUTF16([NSString
	stringWithFormat:
	@"chrome.setFaviconBackground('%s', 'data:image/png;base64,%@');",
	favicon_url_string.c_str(),
	[expected_data base64EncodedStringWithOptions:0]]);

	EXPECT_CALL(*web_state_impl_, ExecuteJavaScript(expected_javascript));
	web_state_impl_->OnScriptCommandReceived("webui.requestFavicon", *message,
	test_url, false);
	}

	// Tests that CRWWebUIManager responds to OnScriptCommandReceieved call and runs
	// JavaScript to fetch a mojo resource.
	TEST_F(CRWWebUIManagerTest, HandleLoadMojoRequest) {
	// Create mock JavaScript message to request a mojo resource.
	std::unique_ptr<base::ListValue> arguments(new base::ListValue());
	arguments->AppendString(kMojoModuleName);
	const char kTestLoadId[] = "test-load-id";
	arguments->AppendString(kTestLoadId);
	base::DictionaryValue message;
	message.SetString("message", "webui.loadMojo");
	message.Set("arguments", std::move(arguments));

	// Create expected JavaScript to call.
	std::string expected_javascript = base::StringPrintf(
	"%s__crWeb.webUIModuleLoadNotifier.moduleLoadCompleted(\"%s\", \"%s\");",
	base::SysNSStringToUTF8(kMojoModule).c_str(), kMojoModuleName,
	kTestLoadId);

	EXPECT_CALL(*web_state_impl_,
	ExecuteJavaScript(base::UTF8ToUTF16(expected_javascript)));
	web_state_impl_->OnScriptCommandReceived("webui.loadMojo", message,
	GURL(kTestWebUIUrl), false);
	}
	} // namespace web