Added Setters to NavigationContextImpl.

ios/web/web_state/navigation_context_impl_unittest.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 "ios/web/web_state/navigation_context_impl.h"
 
 #import "ios/web/public/test/fakes/test_web_state.h"
 #include "net/http/http_response_headers.h"
 #include "testing/gtest/include/gtest/gtest.h"
 #include "testing/platform_test.h"
@@ skipping 12 matching lines @@
   NavigationContextImplTest()
       : url_("https://chromium.test"),
         response_headers_(new net::HttpResponseHeaders(
             std::string(kRawResponseHeaders, sizeof(kRawResponseHeaders)))) {}
 
   TestWebState web_state_;
   GURL url_;
   scoped_refptr<net::HttpResponseHeaders> response_headers_;
 };
 
-// Tests CreateNavigationContext factory method.
-TEST_F(NavigationContextImplTest, NavigationContext) {
+// Tests legacy CreateNavigationContext factory method.
+TEST_F(NavigationContextImplTest, LegacyNavigationContext) {
   std::unique_ptr<NavigationContext> context =
       NavigationContextImpl::CreateNavigationContext(&web_state_, url_,
                                                      response_headers_);
   ASSERT_TRUE(context);
 
   EXPECT_EQ(&web_state_, context->GetWebState());
   EXPECT_EQ(url_, context->GetUrl());
   EXPECT_FALSE(context->IsSameDocument());
   EXPECT_FALSE(context->IsErrorPage());
   EXPECT_EQ(response_headers_.get(), context->GetResponseHeaders());
 }
 
+// Tests CreateNavigationContext factory method.
+TEST_F(NavigationContextImplTest, NavigationContext) {
+  std::unique_ptr<NavigationContext> context =
+      NavigationContextImpl::CreateNavigationContext(&web_state_, url_);
+  ASSERT_TRUE(context);
+
+  EXPECT_EQ(&web_state_, context->GetWebState());
+  EXPECT_EQ(url_, context->GetUrl());
+  EXPECT_FALSE(context->IsSameDocument());
+  EXPECT_FALSE(context->IsErrorPage());
+  EXPECT_FALSE(context->GetResponseHeaders());
+}
+
 // Tests CreateSameDocumentNavigationContext factory method.
 TEST_F(NavigationContextImplTest, SameDocumentNavigationContext) {
   std::unique_ptr<NavigationContext> context =
       NavigationContextImpl::CreateSameDocumentNavigationContext(&web_state_,
                                                                  url_);
   ASSERT_TRUE(context);
 
   EXPECT_EQ(&web_state_, context->GetWebState());
   EXPECT_EQ(url_, context->GetUrl());
   EXPECT_TRUE(context->IsSameDocument());
   EXPECT_FALSE(context->IsErrorPage());
   EXPECT_FALSE(context->GetResponseHeaders());
 }
 
 // Tests CreateErrorPageNavigationContext factory method.
 TEST_F(NavigationContextImplTest, ErrorPageNavigationContext) {
   std::unique_ptr<NavigationContext> context =
       NavigationContextImpl::CreateErrorPageNavigationContext(
           &web_state_, url_, response_headers_);
   ASSERT_TRUE(context);
 
   EXPECT_EQ(&web_state_, context->GetWebState());
   EXPECT_EQ(url_, context->GetUrl());
   EXPECT_FALSE(context->IsSameDocument());
   EXPECT_TRUE(context->IsErrorPage());
   EXPECT_EQ(response_headers_.get(), context->GetResponseHeaders());
 }
 
+// Tests NavigationContextImpl Setters.
+TEST_F(NavigationContextImplTest, Setters) {
+  std::unique_ptr<NavigationContextImpl> context =
+      NavigationContextImpl::CreateNavigationContext(&web_state_, url_);
+  ASSERT_TRUE(context);
+
+  ASSERT_FALSE(context->IsSameDocument());
+  ASSERT_FALSE(context->IsErrorPage());
+  ASSERT_NE(response_headers_.get(), context->GetResponseHeaders());
+
+  // SetSameDocument
+  context->SetIsSameDocument(true);
+  EXPECT_TRUE(context->IsSameDocument());
+  EXPECT_FALSE(context->IsErrorPage());
+  EXPECT_NE(response_headers_.get(), context->GetResponseHeaders());
+
+  // SetErrorPage
+  context->SetIsErrorPage(true);
+  EXPECT_TRUE(context->IsSameDocument());
+  EXPECT_TRUE(context->IsErrorPage());
+  EXPECT_NE(response_headers_.get(), context->GetResponseHeaders());
+
+  // SetResponseHeaders
+  context->SetResponseHeaders(response_headers_);
+  EXPECT_TRUE(context->IsSameDocument());
+  EXPECT_TRUE(context->IsErrorPage());
+  EXPECT_EQ(response_headers_.get(), context->GetResponseHeaders());
+}
+
 }  // namespace web