Do not rewrite about urls to chrome:// for cetain renderer-initated loads

ios/web/navigation/navigation_manager_impl_unittest.mm

 // 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/navigation/navigation_manager_impl.h"
 
 #include "base/logging.h"
+#include "base/mac/bind_objc_block.h"
 #import "base/mac/scoped_nsobject.h"
 #import "ios/web/navigation/crw_session_controller+private_constructors.h"
 #import "ios/web/navigation/navigation_manager_delegate.h"
 #include "ios/web/public/navigation_item.h"
 #include "ios/web/public/test/fakes/test_browser_state.h"
 #include "ios/web/test/test_url_constants.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
+#include "url/scheme_host_port.h"
+#include "url/url_util.h"
 
 namespace web {
 namespace {
+
+// URL scheme that will be rewritten by UrlRewriter installed in
+// NavigationManagerTest fixture. Scheme will be changed to kTestWebUIScheme.
+const char kSchemeToRewrite[] = "navigationmanagerschemetorewrite";
+
+// Replaces |kSchemeToRewrite| scheme with |kTestWebUIScheme|.
+bool UrlRewriter(GURL* url, BrowserState* browser_state) {
+  if (url->scheme() == kSchemeToRewrite) {
+    GURL::Replacements scheme_replacements;
+    scheme_replacements.SetSchemeStr(kTestWebUIScheme);
+    *url = url->ReplaceComponents(scheme_replacements);
+  }
+  return false;
+}
+
 // Stub class for NavigationManagerDelegate.
 class TestNavigationManagerDelegate : public NavigationManagerDelegate {
  public:
   bool reload_called() { return reload_called_; }
 
  private:
   // NavigationManagerDelegate overrides.
   void GoToIndex(int index) override {}
   void LoadURLWithParams(const NavigationManager::WebLoadParams&) override {}
   void Reload() override { reload_called_ = true; }
   void OnNavigationItemsPruned(size_t pruned_item_count) override {}
   void OnNavigationItemChanged() override{};
   void OnNavigationItemCommitted(const LoadCommittedDetails&) override {}
   WebState* GetWebState() override { return nullptr; }
 
   bool reload_called_ = false;
 };
 }  // namespace
 
 // Test fixture for NavigationManagerImpl testing.
 class NavigationManagerTest : public PlatformTest {
  protected:
   NavigationManagerTest() : manager_(new NavigationManagerImpl()) {
+    // Setup rewriter.
+    BrowserURLRewriter::GetInstance()->AddURLRewriter(UrlRewriter);
+    url::AddStandardScheme(kSchemeToRewrite, url::SCHEME_WITHOUT_PORT);
+
     manager_->SetDelegate(&delegate_);
     manager_->SetBrowserState(&browser_state_);
     controller_.reset(
         [[CRWSessionController alloc] initWithBrowserState:&browser_state_]);
     manager_->SetSessionController(controller_.get());
   }
   CRWSessionController* session_controller() { return controller_.get(); }
   NavigationManagerImpl* navigation_manager() { return manager_.get(); }
   TestNavigationManagerDelegate navigation_manager_delegate() {
     return delegate_;
@@ skipping 1240 matching lines @@
 
   navigation_manager()->Reload(web::ReloadType::ORIGINAL_REQUEST_URL,
                                false /* check_for_repost */);
   EXPECT_TRUE(navigation_manager_delegate().reload_called());
 
   ASSERT_TRUE(navigation_manager()->GetLastCommittedItem());
   EXPECT_EQ(expected_original_url,
             navigation_manager()->GetLastCommittedItem()->GetURL());
 }
 
+// Tests that app-specific URLs are not rewritten for renderer-initiated loads
+// unless requested by a page with app-specific url.
+TEST_F(NavigationManagerTest, RewritingAppSpecificUrls) {
+  // URL should not be rewritten as there is no committed URL.
+  GURL url1(url::SchemeHostPort(kSchemeToRewrite, "test", 0).Serialize());
+  navigation_manager()->AddPendingItem(
+      url1, Referrer(), ui::PAGE_TRANSITION_LINK,
+      web::NavigationInitiationType::RENDERER_INITIATED,
+      web::NavigationManager::UserAgentOverrideOption::INHERIT);
+  EXPECT_EQ(url1, navigation_manager()->GetPendingItem()->GetURL());
+
+  // URL should not be rewritten because last committed URL is not app-specific.
+  [session_controller() commitPendingItem];
+  GURL url2(url::SchemeHostPort(kSchemeToRewrite, "test2", 0).Serialize());
+  navigation_manager()->AddPendingItem(
+      url2, Referrer(), ui::PAGE_TRANSITION_LINK,
+      web::NavigationInitiationType::RENDERER_INITIATED,
+      web::NavigationManager::UserAgentOverrideOption::INHERIT);
+  EXPECT_EQ(url2, navigation_manager()->GetPendingItem()->GetURL());
+
+  // URL should be rewritten for user initiated navigations.
+  GURL url3(url::SchemeHostPort(kSchemeToRewrite, "test3", 0).Serialize());
+  navigation_manager()->AddPendingItem(
+      url3, Referrer(), ui::PAGE_TRANSITION_LINK,
+      web::NavigationInitiationType::USER_INITIATED,
+      web::NavigationManager::UserAgentOverrideOption::INHERIT);
+  GURL rewritten_url3(
+      url::SchemeHostPort(kTestWebUIScheme, "test3", 0).Serialize());
+  EXPECT_EQ(rewritten_url3, navigation_manager()->GetPendingItem()->GetURL());
+
+  // URL should be rewritten because last committed URL is app-specific.
+  [session_controller() commitPendingItem];
+  GURL url4(url::SchemeHostPort(kSchemeToRewrite, "test4", 0).Serialize());
+  navigation_manager()->AddPendingItem(
+      url4, Referrer(), ui::PAGE_TRANSITION_LINK,
+      web::NavigationInitiationType::RENDERER_INITIATED,
+      web::NavigationManager::UserAgentOverrideOption::INHERIT);
+  GURL rewritten_url4(
+      url::SchemeHostPort(kTestWebUIScheme, "test4", 0).Serialize());
+  EXPECT_EQ(rewritten_url4, navigation_manager()->GetPendingItem()->GetURL());
+}
+
 }  // namespace web