ios/web/web_state/web_state_unittest.mm - chromium/src - Git at Google

 // Copyright 2017 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/web_state/web_state.h"

 #import <UIKit/UIKit.h>

 #include "base/mac/bind_objc_block.h"
 #include "base/strings/utf_string_conversions.h"
 #include "base/test/ios/wait_util.h"
 #include "base/values.h"
 #import "ios/testing/wait_util.h"
 #import "ios/web/public/navigation_manager.h"
 #import "ios/web/public/test/fakes/test_web_state_delegate.h"
 #import "ios/web/public/test/web_test_with_web_state.h"
 #include "ui/gfx/image/image.h"
 #include "ui/gfx/image/image_unittest_util.h"

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

 using testing::WaitUntilConditionOrTimeout;
 using testing::kWaitForJSCompletionTimeout;

 namespace web {

 // Test fixture for web::WebTest class.
 typedef web::WebTestWithWebState WebStateTest;

 // Tests script execution with and without callback.
 TEST_F(WebStateTest, ScriptExecution) {
   ASSERT_TRUE(LoadHtml("<html></html>"));

   // Execute script without callback.
   web_state()->ExecuteJavaScript(base::UTF8ToUTF16("window.foo = 'bar'"));

   // Execute script with callback.
   __block std::unique_ptr<base::Value> execution_result;
   __block bool execution_complete = false;
   web_state()->ExecuteJavaScript(
       base::UTF8ToUTF16("window.foo"),
       base::BindBlockArc(^(const base::Value* value) {
         execution_result = value->CreateDeepCopy();
         execution_complete = true;
       }));
   WaitForCondition(^{
     return execution_complete;
   });

   ASSERT_TRUE(execution_result);
   std::string string_result;
   execution_result->GetAsString(&string_result);
   EXPECT_EQ("bar", string_result);
 }

 // Tests that executing user JavaScript registers user interaction.
 TEST_F(WebStateTest, UserScriptExecution) {
   web::TestWebStateDelegate delegate;
   web_state()->SetDelegate(&delegate);
   ASSERT_TRUE(delegate.child_windows().empty());

   ASSERT_TRUE(LoadHtml("<html></html>"));
   web_state()->ExecuteUserJavaScript(@"window.open('', target='_blank');");

   web::TestWebStateDelegate* delegate_ptr = &delegate;
   bool suceess = WaitUntilConditionOrTimeout(kWaitForJSCompletionTimeout, ^{
     // Child window can only be open if the user interaction was registered.
     return delegate_ptr->child_windows().size() == 1;
   });

   ASSERT_TRUE(suceess);
   EXPECT_TRUE(delegate.child_windows()[0]);
 }

 // Tests loading progress.
 TEST_F(WebStateTest, LoadingProgress) {
   EXPECT_FLOAT_EQ(0.0, web_state()->GetLoadingProgress());
   ASSERT_TRUE(LoadHtml("<html></html>"));
   WaitForCondition(^bool() {
     return web_state()->GetLoadingProgress() == 1.0;
   });
 }

 // Tests that page which overrides window.webkit object does not break the
 // messaging system.
 TEST_F(WebStateTest, OverridingWebKitObject) {
   // Add a script command handler.
   __block bool message_received = false;
   const web::WebState::ScriptCommandCallback callback = base::BindBlockArc(
       ^bool(const base::DictionaryValue&, const GURL&, bool) {
         message_received = true;
         return true;
       });
   web_state()->AddScriptCommandCallback(callback, "test");

   // Load the page which overrides window.webkit object and wait until the
   // test message is received.
   ASSERT_TRUE(LoadHtml(
       "<script>"
       "  webkit = undefined;"
       "  __gCrWeb.message.invokeOnHost({'command': 'test.webkit-overriding'});"
       "</script>"));

   WaitForCondition(^{
     return message_received;
   });
   web_state()->RemoveScriptCommandCallback("test");
 }

 // Tests that reload with web::ReloadType::NORMAL is no-op when navigation
 // manager is empty.
 TEST_F(WebStateTest, ReloadWithNormalTypeWithEmptyNavigationManager) {
   NavigationManager* navigation_manager = web_state()->GetNavigationManager();
   ASSERT_FALSE(navigation_manager->GetTransientItem());
   ASSERT_FALSE(navigation_manager->GetPendingItem());
   ASSERT_FALSE(navigation_manager->GetLastCommittedItem());

   navigation_manager->Reload(web::ReloadType::NORMAL,
                              false /* check_for_repost */);

   ASSERT_FALSE(navigation_manager->GetTransientItem());
   ASSERT_FALSE(navigation_manager->GetPendingItem());
   ASSERT_FALSE(navigation_manager->GetLastCommittedItem());
 }

 // Tests that reload with web::ReloadType::ORIGINAL_REQUEST_URL is no-op when
 // navigation manager is empty.
 TEST_F(WebStateTest, ReloadWithOriginalTypeWithEmptyNavigationManager) {
   NavigationManager* navigation_manager = web_state()->GetNavigationManager();
   ASSERT_FALSE(navigation_manager->GetTransientItem());
   ASSERT_FALSE(navigation_manager->GetPendingItem());
   ASSERT_FALSE(navigation_manager->GetLastCommittedItem());

   navigation_manager->Reload(web::ReloadType::ORIGINAL_REQUEST_URL,
                              false /* check_for_repost */);

   ASSERT_FALSE(navigation_manager->GetTransientItem());
   ASSERT_FALSE(navigation_manager->GetPendingItem());
   ASSERT_FALSE(navigation_manager->GetLastCommittedItem());
 }

 // Tests that the snapshot method returns an image of a rendered html page.
 TEST_F(WebStateTest, Snapshot) {
   ASSERT_TRUE(
       LoadHtml("<html><div style='background-color:#FF0000; width:50%; "
                "height:100%;'></div></html>"));
   __block bool snapshot_complete = false;
   [[[UIApplication sharedApplication] keyWindow]
       addSubview:web_state()->GetView()];
   // The subview is added but not immediately painted, so a small delay is
   // necessary.
   base::test::ios::SpinRunLoopWithMinDelay(base::TimeDelta::FromSecondsD(0.2));
   CGSize target_size = CGSizeMake(100.0f, 100.0f);
   web_state()->TakeSnapshot(
       base::BindBlockArc(^(const gfx::Image& snapshot) {
         ASSERT_FALSE(snapshot.IsEmpty());
         EXPECT_EQ(snapshot.Width(), target_size.width);
         EXPECT_EQ(snapshot.Height(), target_size.height);
         // Test a pixel on the left (red) side.
         gfx::test::CheckColors(gfx::test::GetPlatformImageColor(
                                    gfx::test::ToPlatformType(snapshot), 45, 50),
                                SK_ColorRED);
         // Test a pixel on the right (white) side.
         gfx::test::CheckColors(gfx::test::GetPlatformImageColor(
                                    gfx::test::ToPlatformType(snapshot), 55, 50),
                                SK_ColorWHITE);
         snapshot_complete = true;
       }),
       target_size);
   WaitForCondition(^{
     return snapshot_complete;
   });
 }

 // Tests that the web state has an opener after calling SetHasOpener().
 TEST_F(WebStateTest, SetHasOpener) {
   ASSERT_FALSE(web_state()->HasOpener());
   web_state()->SetHasOpener(true);
   EXPECT_TRUE(web_state()->HasOpener());
 }

 }  // namespace web
	// Copyright 2017 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/web_state/web_state.h"

	#import <UIKit/UIKit.h>

	#include "base/mac/bind_objc_block.h"
	#include "base/strings/utf_string_conversions.h"
	#include "base/test/ios/wait_util.h"
	#include "base/values.h"
	#import "ios/testing/wait_util.h"
	#import "ios/web/public/navigation_manager.h"
	#import "ios/web/public/test/fakes/test_web_state_delegate.h"
	#import "ios/web/public/test/web_test_with_web_state.h"
	#include "ui/gfx/image/image.h"
	#include "ui/gfx/image/image_unittest_util.h"

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

	using testing::WaitUntilConditionOrTimeout;
	using testing::kWaitForJSCompletionTimeout;

	namespace web {

	// Test fixture for web::WebTest class.
	typedef web::WebTestWithWebState WebStateTest;

	// Tests script execution with and without callback.
	TEST_F(WebStateTest, ScriptExecution) {
	ASSERT_TRUE(LoadHtml("<html></html>"));

	// Execute script without callback.
	web_state()->ExecuteJavaScript(base::UTF8ToUTF16("window.foo = 'bar'"));

	// Execute script with callback.
	__block std::unique_ptr<base::Value> execution_result;
	__block bool execution_complete = false;
	web_state()->ExecuteJavaScript(
	base::UTF8ToUTF16("window.foo"),
	base::BindBlockArc(^(const base::Value* value) {
	execution_result = value->CreateDeepCopy();
	execution_complete = true;
	}));
	WaitForCondition(^{
	return execution_complete;
	});

	ASSERT_TRUE(execution_result);
	std::string string_result;
	execution_result->GetAsString(&string_result);
	EXPECT_EQ("bar", string_result);
	}

	// Tests that executing user JavaScript registers user interaction.
	TEST_F(WebStateTest, UserScriptExecution) {
	web::TestWebStateDelegate delegate;
	web_state()->SetDelegate(&delegate);
	ASSERT_TRUE(delegate.child_windows().empty());

	ASSERT_TRUE(LoadHtml("<html></html>"));
	web_state()->ExecuteUserJavaScript(@"window.open('', target='_blank');");

	web::TestWebStateDelegate* delegate_ptr = &delegate;
	bool suceess = WaitUntilConditionOrTimeout(kWaitForJSCompletionTimeout, ^{
	// Child window can only be open if the user interaction was registered.
	return delegate_ptr->child_windows().size() == 1;
	});

	ASSERT_TRUE(suceess);
	EXPECT_TRUE(delegate.child_windows()[0]);
	}

	// Tests loading progress.
	TEST_F(WebStateTest, LoadingProgress) {
	EXPECT_FLOAT_EQ(0.0, web_state()->GetLoadingProgress());
	ASSERT_TRUE(LoadHtml("<html></html>"));
	WaitForCondition(^bool() {
	return web_state()->GetLoadingProgress() == 1.0;
	});
	}

	// Tests that page which overrides window.webkit object does not break the
	// messaging system.
	TEST_F(WebStateTest, OverridingWebKitObject) {
	// Add a script command handler.
	__block bool message_received = false;
	const web::WebState::ScriptCommandCallback callback = base::BindBlockArc(
	^bool(const base::DictionaryValue&, const GURL&, bool) {
	message_received = true;
	return true;
	});
	web_state()->AddScriptCommandCallback(callback, "test");

	// Load the page which overrides window.webkit object and wait until the
	// test message is received.
	ASSERT_TRUE(LoadHtml(
	"<script>"
	" webkit = undefined;"
	" __gCrWeb.message.invokeOnHost({'command': 'test.webkit-overriding'});"
	"</script>"));

	WaitForCondition(^{
	return message_received;
	});
	web_state()->RemoveScriptCommandCallback("test");
	}

	// Tests that reload with web::ReloadType::NORMAL is no-op when navigation
	// manager is empty.
	TEST_F(WebStateTest, ReloadWithNormalTypeWithEmptyNavigationManager) {
	NavigationManager* navigation_manager = web_state()->GetNavigationManager();
	ASSERT_FALSE(navigation_manager->GetTransientItem());
	ASSERT_FALSE(navigation_manager->GetPendingItem());
	ASSERT_FALSE(navigation_manager->GetLastCommittedItem());

	navigation_manager->Reload(web::ReloadType::NORMAL,
	false /* check_for_repost */);

	ASSERT_FALSE(navigation_manager->GetTransientItem());
	ASSERT_FALSE(navigation_manager->GetPendingItem());
	ASSERT_FALSE(navigation_manager->GetLastCommittedItem());
	}

	// Tests that reload with web::ReloadType::ORIGINAL_REQUEST_URL is no-op when
	// navigation manager is empty.
	TEST_F(WebStateTest, ReloadWithOriginalTypeWithEmptyNavigationManager) {
	NavigationManager* navigation_manager = web_state()->GetNavigationManager();
	ASSERT_FALSE(navigation_manager->GetTransientItem());
	ASSERT_FALSE(navigation_manager->GetPendingItem());
	ASSERT_FALSE(navigation_manager->GetLastCommittedItem());

	navigation_manager->Reload(web::ReloadType::ORIGINAL_REQUEST_URL,
	false /* check_for_repost */);

	ASSERT_FALSE(navigation_manager->GetTransientItem());
	ASSERT_FALSE(navigation_manager->GetPendingItem());
	ASSERT_FALSE(navigation_manager->GetLastCommittedItem());
	}

	// Tests that the snapshot method returns an image of a rendered html page.
	TEST_F(WebStateTest, Snapshot) {
	ASSERT_TRUE(
	LoadHtml("<html><div style='background-color:#FF0000; width:50%; "
	"height:100%;'></div></html>"));
	__block bool snapshot_complete = false;
	[[[UIApplication sharedApplication] keyWindow]
	addSubview:web_state()->GetView()];
	// The subview is added but not immediately painted, so a small delay is
	// necessary.
	base::test::ios::SpinRunLoopWithMinDelay(base::TimeDelta::FromSecondsD(0.2));
	CGSize target_size = CGSizeMake(100.0f, 100.0f);
	web_state()->TakeSnapshot(
	base::BindBlockArc(^(const gfx::Image& snapshot) {
	ASSERT_FALSE(snapshot.IsEmpty());
	EXPECT_EQ(snapshot.Width(), target_size.width);
	EXPECT_EQ(snapshot.Height(), target_size.height);
	// Test a pixel on the left (red) side.
	gfx::test::CheckColors(gfx::test::GetPlatformImageColor(
	gfx::test::ToPlatformType(snapshot), 45, 50),
	SK_ColorRED);
	// Test a pixel on the right (white) side.
	gfx::test::CheckColors(gfx::test::GetPlatformImageColor(
	gfx::test::ToPlatformType(snapshot), 55, 50),
	SK_ColorWHITE);
	snapshot_complete = true;
	}),
	target_size);
	WaitForCondition(^{
	return snapshot_complete;
	});
	}

	// Tests that the web state has an opener after calling SetHasOpener().
	TEST_F(WebStateTest, SetHasOpener) {
	ASSERT_FALSE(web_state()->HasOpener());
	web_state()->SetHasOpener(true);
	EXPECT_TRUE(web_state()->HasOpener());
	}

	} // namespace web