Added web::NavigationContext::GetPageTransition.

ios/web/web_state/navigation_callbacks_inttest.mm

 // 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.
 
 #include <memory>
 #include <string>
 
 #include "base/memory/ptr_util.h"
 #import "ios/web/public/navigation_item.h"
 #import "ios/web/public/navigation_manager.h"
 #import "ios/web/public/test/http_server.h"
 #include "ios/web/public/test/http_server_util.h"
 #import "ios/web/public/web_state/navigation_context.h"
 #include "ios/web/public/web_state/web_state_observer.h"
 #include "ios/web/test/test_url_constants.h"
 #import "ios/web/test/web_int_test.h"
 #include "testing/gmock/include/gmock/gmock.h"
 #include "testing/gtest/include/gtest/gtest.h"
+#include "ui/base/page_transition_types.h"
 #include "url/gurl.h"
 #include "url/scheme_host_port.h"
 
 namespace web {
 
 namespace {
 
 const char kExpectedMimeType[] = "text/html";
 
 // Verifies correctness of |NavigationContext| (|arg0|) for new page navigation
 // passed to |DidStartNavigation|. Stores |NavigationContext| in |context|
 // pointer.
 ACTION_P3(VerifyNewPageStartedContext, web_state, url, context) {
   *context = arg0;
   ASSERT_TRUE(*context);
   EXPECT_EQ(web_state, (*context)->GetWebState());
   EXPECT_EQ(url, (*context)->GetUrl());
+  EXPECT_TRUE(
+      PageTransitionCoreTypeIs(ui::PageTransition::PAGE_TRANSITION_TYPED,
+                               (*context)->GetPageTransition()));
   EXPECT_FALSE((*context)->IsSameDocument());
   EXPECT_FALSE((*context)->GetError());
   ASSERT_FALSE((*context)->GetResponseHeaders());
   NavigationManager* navigation_manager = web_state->GetNavigationManager();
   NavigationItem* item = navigation_manager->GetPendingItem();
   EXPECT_EQ(url, item->GetURL());
 }
 
 // Verifies correctness of |NavigationContext| (|arg0|) for new page navigation
 // passed to |DidFinishNavigation|. Asserts that |NavigationContext| the same as
 // |context|.
 ACTION_P3(VerifyNewPageFinishedContext, web_state, url, context) {
   ASSERT_EQ(*context, arg0);
   EXPECT_EQ(web_state, (*context)->GetWebState());
   ASSERT_TRUE((*context));
   EXPECT_EQ(web_state, (*context)->GetWebState());
   EXPECT_EQ(url, (*context)->GetUrl());
+  EXPECT_TRUE(
+      PageTransitionCoreTypeIs(ui::PageTransition::PAGE_TRANSITION_TYPED,
+                               (*context)->GetPageTransition()));
   EXPECT_FALSE((*context)->IsSameDocument());
   EXPECT_FALSE((*context)->GetError());
   ASSERT_TRUE((*context)->GetResponseHeaders());
   std::string mime_type;
   (*context)->GetResponseHeaders()->GetMimeType(&mime_type);
   EXPECT_EQ(kExpectedMimeType, mime_type);
   NavigationManager* navigation_manager = web_state->GetNavigationManager();
   NavigationItem* item = navigation_manager->GetLastCommittedItem();
   EXPECT_GT(item->GetTimestamp().ToInternalValue(), 0);
   EXPECT_EQ(url, item->GetURL());
 }
 
 // Verifies correctness of |NavigationContext| (|arg0|) for same page navigation
 // passed to |DidFinishNavigation|. Stores |NavigationContext| in |context|
 // pointer.
-ACTION_P3(VerifySameDocumentStartedContext, web_state, url, context) {
+ACTION_P4(VerifySameDocumentStartedContext,
+          web_state,
+          url,
+          context,
+          page_transition) {
   *context = arg0;
   ASSERT_TRUE(*context);
   EXPECT_EQ(web_state, (*context)->GetWebState());
   EXPECT_EQ(url, (*context)->GetUrl());
+  EXPECT_TRUE(PageTransitionTypeIncludingQualifiersIs(
+      page_transition, (*context)->GetPageTransition()));
   EXPECT_FALSE((*context)->IsSameDocument());
   EXPECT_FALSE((*context)->GetError());
   EXPECT_FALSE((*context)->GetResponseHeaders());
 }
 
 // Verifies correctness of |NavigationContext| (|arg0|) for same page navigation
 // passed to |DidFinishNavigation|. Asserts that |NavigationContext| the same as
 // |context|.
-ACTION_P3(VerifySameDocumentFinishedContext, web_state, url, context) {
+ACTION_P4(VerifySameDocumentFinishedContext,
+          web_state,
+          url,
+          context,
+          page_transition) {
   ASSERT_EQ(*context, arg0);
   ASSERT_TRUE(*context);
   EXPECT_EQ(web_state, (*context)->GetWebState());
   EXPECT_EQ(url, (*context)->GetUrl());
+  EXPECT_TRUE(PageTransitionTypeIncludingQualifiersIs(
+      page_transition, (*context)->GetPageTransition()));
   EXPECT_TRUE((*context)->IsSameDocument());
   EXPECT_FALSE((*context)->GetError());
   EXPECT_FALSE((*context)->GetResponseHeaders());
   NavigationManager* navigation_manager = web_state->GetNavigationManager();
   NavigationItem* item = navigation_manager->GetLastCommittedItem();
   EXPECT_GT(item->GetTimestamp().ToInternalValue(), 0);
   EXPECT_EQ(url, item->GetURL());
 }
 
 // Verifies correctness of |NavigationContext| (|arg0|) for new page navigation
 // to native URLs passed to |DidStartNavigation|. Stores |NavigationContext| in
 // |context| pointer.
 ACTION_P3(VerifyNewNativePageStartedContext, web_state, url, context) {
   *context = arg0;
   ASSERT_TRUE(*context);
   EXPECT_EQ(web_state, (*context)->GetWebState());
   EXPECT_EQ(url, (*context)->GetUrl());
+  EXPECT_TRUE(
+      PageTransitionCoreTypeIs(ui::PageTransition::PAGE_TRANSITION_TYPED,
+                               (*context)->GetPageTransition()));
   EXPECT_FALSE((*context)->IsSameDocument());
   EXPECT_FALSE((*context)->GetError());
   EXPECT_FALSE((*context)->GetResponseHeaders());
   NavigationManager* navigation_manager = web_state->GetNavigationManager();
   NavigationItem* item = navigation_manager->GetPendingItem();
   EXPECT_EQ(url, item->GetURL());
 }
 
 // Verifies correctness of |NavigationContext| (|arg0|) for new page navigation
 // to native URLs passed to |DidFinishNavigation|. Asserts that
 // |NavigationContext| the same as |context|.
 ACTION_P3(VerifyNewNativePageFinishedContext, web_state, url, context) {
   ASSERT_EQ(*context, arg0);
   ASSERT_TRUE(*context);
   EXPECT_EQ(web_state, (*context)->GetWebState());
   EXPECT_EQ(url, (*context)->GetUrl());
+  EXPECT_TRUE(
+      PageTransitionCoreTypeIs(ui::PageTransition::PAGE_TRANSITION_TYPED,
+                               (*context)->GetPageTransition()));
   EXPECT_FALSE((*context)->IsSameDocument());
   EXPECT_FALSE((*context)->GetError());
   EXPECT_FALSE((*context)->GetResponseHeaders());
   NavigationManager* navigation_manager = web_state->GetNavigationManager();
   NavigationItem* item = navigation_manager->GetLastCommittedItem();
   EXPECT_GT(item->GetTimestamp().ToInternalValue(), 0);
   EXPECT_EQ(url, item->GetURL());
 }
 
 // Mocks DidFinishNavigation navigation callback.
@@ skipping 48 matching lines @@
   NavigationContext* context = nullptr;
   EXPECT_CALL(*observer_, DidStartNavigation(_))
       .WillOnce(VerifyNewPageStartedContext(web_state(), url, &context));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
       .WillOnce(VerifyNewPageFinishedContext(web_state(), url, &context));
   LoadUrl(url);
 
   // Perform same-page navigation.
   const GURL hash_url = HttpServer::MakeUrl("http://chromium.test#1");
   EXPECT_CALL(*observer_, DidStartNavigation(_))
-      .WillOnce(
-          VerifySameDocumentStartedContext(web_state(), hash_url, &context));
+      .WillOnce(VerifySameDocumentStartedContext(
+          web_state(), hash_url, &context,
+          ui::PageTransition::PAGE_TRANSITION_TYPED));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
-      .WillOnce(
-          VerifySameDocumentFinishedContext(web_state(), hash_url, &context));
+      .WillOnce(VerifySameDocumentFinishedContext(
+          web_state(), hash_url, &context,
+          ui::PageTransition::PAGE_TRANSITION_TYPED));
   LoadUrl(hash_url);
 
   // Perform same-page navigation by going back.
   EXPECT_CALL(*observer_, DidStartNavigation(_))
-      .WillOnce(VerifySameDocumentStartedContext(web_state(), url, &context));
+      .WillOnce(VerifySameDocumentStartedContext(
+          web_state(), url, &context,
+          ui::PageTransition::PAGE_TRANSITION_CLIENT_REDIRECT));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
-      .WillOnce(VerifySameDocumentFinishedContext(web_state(), url, &context));
+      .WillOnce(VerifySameDocumentFinishedContext(
+          web_state(), url, &context,
+          ui::PageTransition::PAGE_TRANSITION_CLIENT_REDIRECT));
   ExecuteBlockAndWaitForLoad(url, ^{
     navigation_manager()->GoBack();
   });
 }
 
 // Tests renderer-initiated hash change.
 TEST_F(StartAndFinishNavigationTest, RendererInitiatedHashChangeNavigation) {
   const GURL url = HttpServer::MakeUrl("http://chromium.test");
   std::map<GURL, std::string> responses;
   responses[url] = "Chromium Test";
   web::test::SetUpSimpleHttpServer(responses);
 
   // Perform new page navigation.
   NavigationContext* context = nullptr;
   EXPECT_CALL(*observer_, DidStartNavigation(_))
       .WillOnce(VerifyNewPageStartedContext(web_state(), url, &context));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
       .WillOnce(VerifyNewPageFinishedContext(web_state(), url, &context));
   LoadUrl(url);
 
   // Perform same-page navigation using JavaScript.
   const GURL hash_url = HttpServer::MakeUrl("http://chromium.test#1");
   EXPECT_CALL(*observer_, DidStartNavigation(_))
-      .WillOnce(
-          VerifySameDocumentStartedContext(web_state(), hash_url, &context));
+      .WillOnce(VerifySameDocumentStartedContext(
+          web_state(), hash_url, &context,
+          ui::PageTransition::PAGE_TRANSITION_CLIENT_REDIRECT));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
-      .WillOnce(
-          VerifySameDocumentFinishedContext(web_state(), hash_url, &context));
+      .WillOnce(VerifySameDocumentFinishedContext(
+          web_state(), hash_url, &context,
+          ui::PageTransition::PAGE_TRANSITION_CLIENT_REDIRECT));
   ExecuteJavaScript(@"window.location.hash = '#1'");
 }
 
 // Tests state change.
 TEST_F(StartAndFinishNavigationTest, StateNavigation) {
   const GURL url = HttpServer::MakeUrl("http://chromium.test");
   std::map<GURL, std::string> responses;
   responses[url] = "Chromium Test";
   web::test::SetUpSimpleHttpServer(responses);
 
   // Perform new page navigation.
   NavigationContext* context = nullptr;
   EXPECT_CALL(*observer_, DidStartNavigation(_))
       .WillOnce(VerifyNewPageStartedContext(web_state(), url, &context));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
       .WillOnce(VerifyNewPageFinishedContext(web_state(), url, &context));
   LoadUrl(url);
 
   // Perform push state using JavaScript.
   const GURL push_url = HttpServer::MakeUrl("http://chromium.test/test.html");
   EXPECT_CALL(*observer_, DidStartNavigation(_))
-      .WillOnce(
-          VerifySameDocumentStartedContext(web_state(), push_url, &context));
+      .WillOnce(VerifySameDocumentStartedContext(
+          web_state(), push_url, &context,
+          ui::PageTransition::PAGE_TRANSITION_CLIENT_REDIRECT));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
-      .WillOnce(
-          VerifySameDocumentFinishedContext(web_state(), push_url, &context));
+      .WillOnce(VerifySameDocumentFinishedContext(
+          web_state(), push_url, &context,
+          ui::PageTransition::PAGE_TRANSITION_CLIENT_REDIRECT));
   ExecuteJavaScript(@"window.history.pushState('', 'Test', 'test.html')");
 
   // Perform replace state using JavaScript.
   const GURL replace_url = HttpServer::MakeUrl("http://chromium.test/1.html");
   EXPECT_CALL(*observer_, DidStartNavigation(_))
-      .WillOnce(
-          VerifySameDocumentStartedContext(web_state(), replace_url, &context));
+      .WillOnce(VerifySameDocumentStartedContext(
+          web_state(), replace_url, &context,
+          ui::PageTransition::PAGE_TRANSITION_CLIENT_REDIRECT));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
-      .WillOnce(VerifySameDocumentFinishedContext(web_state(), replace_url,
-                                                  &context));
+      .WillOnce(VerifySameDocumentFinishedContext(
+          web_state(), replace_url, &context,
+          ui::PageTransition::PAGE_TRANSITION_CLIENT_REDIRECT));
   ExecuteJavaScript(@"window.history.replaceState('', 'Test', '1.html')");
 }
 
 // Tests native content navigation.
 TEST_F(StartAndFinishNavigationTest, NativeContentNavigation) {
   GURL url(url::SchemeHostPort(kTestNativeContentScheme, "ui", 0).Serialize());
   NavigationContext* context = nullptr;
   EXPECT_CALL(*observer_, DidStartNavigation(_))
       .WillOnce(VerifyNewNativePageStartedContext(web_state(), url, &context));
   EXPECT_CALL(*observer_, DidFinishNavigation(_))
       .WillOnce(VerifyNewNativePageFinishedContext(web_state(), url, &context));
   LoadUrl(url);
 }
 
 }  // namespace web