// 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/navigation/navigation_context_impl.h"

#import "ios/web/navigation/navigation_item_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"
#import "testing/gtest_mac.h"
#include "testing/platform_test.h"

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

namespace web {
namespace {
const char kRawResponseHeaders[] = "HTTP/1.1 200 OK\0"
                                   "Content-Length: 450\0"
                                   "Connection: keep-alive\0";
}  // namespace

// Test fixture for NavigationContextImplTest testing.
class NavigationContextImplTest : public PlatformTest {
 protected:
  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) {
  std::unique_ptr<NavigationContext> context =
      NavigationContextImpl::CreateNavigationContext(
          &web_state_, url_, /*has_user_gesture=*/true,
          ui::PageTransition::PAGE_TRANSITION_FORWARD_BACK,
          /*is_renderer_initiated=*/true);
  ASSERT_TRUE(context);

  EXPECT_EQ(&web_state_, context->GetWebState());
  EXPECT_TRUE(PageTransitionTypeIncludingQualifiersIs(
      context->GetPageTransition(),
      ui::PageTransition::PAGE_TRANSITION_FORWARD_BACK));
  EXPECT_EQ(url_, context->GetUrl());
  EXPECT_TRUE(context->HasUserGesture());
  EXPECT_FALSE(context->IsSameDocument());
  EXPECT_FALSE(context->HasCommitted());
  EXPECT_FALSE(context->IsDownload());
  EXPECT_FALSE(context->GetError());
  EXPECT_FALSE(context->GetResponseHeaders());
  EXPECT_TRUE(context->IsRendererInitiated());
}

TEST_F(NavigationContextImplTest, NavigationId) {
  std::unique_ptr<NavigationContextImpl> context1 =
      NavigationContextImpl::CreateNavigationContext(
          &web_state_, url_, /*has_user_gesture=*/true,
          ui::PageTransition::PAGE_TRANSITION_FORWARD_BACK,
          /*is_renderer_initiated=*/false);
  ASSERT_TRUE(context1);
  std::unique_ptr<NavigationContextImpl> context2 =
      NavigationContextImpl::CreateNavigationContext(
          &web_state_, url_, /*has_user_gesture=*/true,
          ui::PageTransition::PAGE_TRANSITION_FORWARD_BACK,
          /*is_renderer_initiated=*/false);
  ASSERT_TRUE(context2);
  ASSERT_NE(0, context1->GetNavigationId());
  ASSERT_NE(0, context2->GetNavigationId());
  ASSERT_NE(context1->GetNavigationId(), context2->GetNavigationId());
}

// Tests NavigationContextImpl Setters.
TEST_F(NavigationContextImplTest, Setters) {
  std::unique_ptr<NavigationContextImpl> context =
      NavigationContextImpl::CreateNavigationContext(
          &web_state_, url_, /*has_user_gesture=*/true,
          ui::PageTransition::PAGE_TRANSITION_FORWARD_BACK,
          /*is_renderer_initiated=*/false);
  ASSERT_TRUE(context);

  EXPECT_EQ(url_, context->GetUrl());
  ASSERT_FALSE(context->IsSameDocument());
  EXPECT_FALSE(context->HasCommitted());
  EXPECT_FALSE(context->IsDownload());
  ASSERT_FALSE(context->IsPost());
  ASSERT_FALSE(context->GetError());
  ASSERT_FALSE(context->IsRendererInitiated());
  ASSERT_NE(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetUrl
  GURL new_url("https://new.test");
  context->SetUrl(new_url);
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_FALSE(context->IsSameDocument());
  EXPECT_FALSE(context->HasCommitted());
  EXPECT_FALSE(context->IsDownload());
  ASSERT_FALSE(context->IsPost());
  EXPECT_FALSE(context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_NE(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetSameDocument
  context->SetIsSameDocument(true);
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_FALSE(context->HasCommitted());
  EXPECT_FALSE(context->IsDownload());
  ASSERT_FALSE(context->IsPost());
  EXPECT_FALSE(context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_NE(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetHasCommitted
  context->SetHasCommitted(true);
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_TRUE(context->HasCommitted());
  EXPECT_FALSE(context->IsDownload());
  ASSERT_FALSE(context->IsPost());
  EXPECT_FALSE(context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_NE(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetIsDownload
  context->SetIsDownload(true);
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_TRUE(context->HasCommitted());
  EXPECT_TRUE(context->IsDownload());
  ASSERT_FALSE(context->IsPost());
  EXPECT_FALSE(context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_NE(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetPost
  context->SetIsPost(true);
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_TRUE(context->HasCommitted());
  EXPECT_TRUE(context->IsDownload());
  ASSERT_TRUE(context->IsPost());
  EXPECT_FALSE(context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_NE(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetErrorPage
  NSError* error = [[NSError alloc] initWithDomain:@"" code:0 userInfo:nil];
  context->SetError(error);
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_TRUE(context->HasCommitted());
  EXPECT_TRUE(context->IsDownload());
  ASSERT_TRUE(context->IsPost());
  EXPECT_EQ(error, context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_NE(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetResponseHeaders
  context->SetResponseHeaders(response_headers_);
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_TRUE(context->HasCommitted());
  EXPECT_TRUE(context->IsDownload());
  ASSERT_TRUE(context->IsPost());
  EXPECT_EQ(error, context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_EQ(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeOther, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetWKNavigationType
  context->SetWKNavigationType(WKNavigationTypeBackForward);
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_TRUE(context->HasCommitted());
  EXPECT_TRUE(context->IsDownload());
  ASSERT_TRUE(context->IsPost());
  EXPECT_EQ(error, context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_EQ(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeBackForward, context->GetWKNavigationType());
  EXPECT_FALSE(context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetMimeType
  context->SetMimeType(@"test/mime");
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_TRUE(context->HasCommitted());
  EXPECT_TRUE(context->IsDownload());
  ASSERT_TRUE(context->IsPost());
  EXPECT_EQ(error, context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_EQ(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeBackForward, context->GetWKNavigationType());
  EXPECT_NSEQ(@"test/mime", context->GetMimeType());
  EXPECT_FALSE(context->GetItem());

  // SetItem and ReleaseItem
  auto item = std::make_unique<NavigationItemImpl>();
  NavigationItemImpl* item_ptr = item.get();
  context->SetNavigationItemUniqueID(item->GetUniqueID());
  context->SetItem(std::move(item));
  EXPECT_EQ(new_url, context->GetUrl());
  EXPECT_TRUE(context->IsSameDocument());
  EXPECT_TRUE(context->HasCommitted());
  EXPECT_TRUE(context->IsDownload());
  ASSERT_TRUE(context->IsPost());
  EXPECT_EQ(error, context->GetError());
  EXPECT_FALSE(context->IsRendererInitiated());
  EXPECT_EQ(response_headers_.get(), context->GetResponseHeaders());
  EXPECT_EQ(WKNavigationTypeBackForward, context->GetWKNavigationType());
  EXPECT_NSEQ(@"test/mime", context->GetMimeType());
  EXPECT_EQ(item_ptr, context->GetItem());
  item = context->ReleaseItem();
  EXPECT_EQ(item_ptr, item.get());
}

}  // namespace web