content/browser/loader/reload_cache_control_browsertest.cc

// Copyright 2016 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 <vector>

#include "base/bind.h"
#include "base/bind_helpers.h"
#include "base/macros.h"
#include "base/synchronization/lock.h"
#include "content/public/common/browser_side_navigation_policy.h"
#include "content/public/test/content_browser_test.h"
#include "content/public/test/content_browser_test_utils.h"
#include "content/shell/browser/shell.h"
#include "net/test/embedded_test_server/embedded_test_server.h"
#include "net/test/embedded_test_server/http_request.h"
#include "net/test/embedded_test_server/http_response.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "url/gurl.h"

namespace content {

namespace {

using net::test_server::HttpRequest;
using net::test_server::HttpResponse;

const char kReloadTestPath[] = "/loader/reload.html";
const char kReloadImagePath[] = "/loader/empty16x16.png";

const char kNoCacheControl[] = "";
const char kMaxAgeCacheControl[] = "max-age=0";
const char kNoCacheCacheControl[] = "no-cache";

struct RequestLog {
  std::string relative_url;
  std::string cache_control;
};

// Tests end to end behaviors between Blink and content around reload variants.
class ReloadCacheControlBrowserTest : public ContentBrowserTest {
 protected:
  ReloadCacheControlBrowserTest() {}
  ~ReloadCacheControlBrowserTest() override = default;

  void SetUpOnMainThread() override {
    SetUpTestServerOnMainThread();
  }

  void SetUpTestServerOnMainThread() {
    // ContentBrowserTest creates embedded_test_server instance with
    // a registered HandleFileRequest for "content/test/data".
    // Because the handler is registered as the first handler, MonitorHandler
    // is needed to capture all requests.
    embedded_test_server()->RegisterRequestMonitor(base::Bind(
        &ReloadCacheControlBrowserTest::MonitorRequestHandler,
        base::Unretained(this)));

    ASSERT_TRUE(embedded_test_server()->Start());
  }

 protected:
  std::vector<RequestLog> request_log_;
  base::Lock request_log_lock_;

 private:
  void MonitorRequestHandler(const HttpRequest& request) {
    RequestLog log;
    log.relative_url = request.relative_url;
    auto cache_control = request.headers.find("Cache-Control");
    log.cache_control = cache_control == request.headers.end()
                            ? kNoCacheControl
                            : cache_control->second;
    base::AutoLock lock(request_log_lock_);
    request_log_.push_back(log);
  }

  DISALLOW_COPY_AND_ASSIGN(ReloadCacheControlBrowserTest);
};

IN_PROC_BROWSER_TEST_F(ReloadCacheControlBrowserTest, NormalReload) {
  GURL url(embedded_test_server()->GetURL(kReloadTestPath));

  EXPECT_TRUE(NavigateToURL(shell(), url));
  ReloadBlockUntilNavigationsComplete(shell(), 1);

  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(4UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[0].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[0].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[1].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[1].cache_control);

    EXPECT_EQ(kReloadTestPath, request_log_[2].relative_url);
    EXPECT_EQ(kMaxAgeCacheControl, request_log_[2].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[3].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[3].cache_control);
  }

  shell()->ShowDevTools();
  ReloadBlockUntilNavigationsComplete(shell(), 1);

  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(6UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[4].relative_url);
    EXPECT_EQ(kMaxAgeCacheControl, request_log_[4].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[5].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[5].cache_control);
  }

  shell()->CloseDevTools();
  ReloadBlockUntilNavigationsComplete(shell(), 1);

  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(8UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[6].relative_url);
    EXPECT_EQ(kMaxAgeCacheControl, request_log_[6].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[7].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[7].cache_control);
  }
}

IN_PROC_BROWSER_TEST_F(ReloadCacheControlBrowserTest, BypassingReload) {
  GURL url(embedded_test_server()->GetURL(kReloadTestPath));

  EXPECT_TRUE(NavigateToURL(shell(), url));
  ReloadBypassingCacheBlockUntilNavigationsComplete(shell(), 1);

  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(4UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[0].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[0].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[1].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[1].cache_control);

    EXPECT_EQ(kReloadTestPath, request_log_[2].relative_url);
    EXPECT_EQ(kNoCacheCacheControl, request_log_[2].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[3].relative_url);
    EXPECT_EQ(kNoCacheCacheControl, request_log_[3].cache_control);
  }

  shell()->ShowDevTools();
  ReloadBypassingCacheBlockUntilNavigationsComplete(shell(), 1);

  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(6UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[4].relative_url);
    EXPECT_EQ(kNoCacheCacheControl, request_log_[4].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[5].relative_url);
    EXPECT_EQ(kNoCacheCacheControl, request_log_[5].cache_control);
  }

  shell()->CloseDevTools();
  ReloadBypassingCacheBlockUntilNavigationsComplete(shell(), 1);

  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(8UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[6].relative_url);
    EXPECT_EQ(kNoCacheCacheControl, request_log_[6].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[7].relative_url);
    EXPECT_EQ(kNoCacheCacheControl, request_log_[7].cache_control);
  }
}

IN_PROC_BROWSER_TEST_F(ReloadCacheControlBrowserTest, NavigateToSame) {
  GURL url(embedded_test_server()->GetURL(kReloadTestPath));

  EXPECT_TRUE(NavigateToURL(shell(), url));
  EXPECT_TRUE(NavigateToURL(shell(), url));

  // The first navigation is just a normal load.
  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(4UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[0].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[0].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[1].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[1].cache_control);
  }

  // TODO(crbug.com/671545): This test does not work correctly if browser-side
  // navigation is enabled.
  if (IsBrowserSideNavigationEnabled())
    return;

  // The second navigation is the same page navigation. This should be handled
  // as a reload, revalidating the main resource, but following cache protocols
  // for others.
  {
    base::AutoLock lock(request_log_lock_);
    EXPECT_EQ(kReloadTestPath, request_log_[2].relative_url);
    EXPECT_EQ(kMaxAgeCacheControl, request_log_[2].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[3].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[3].cache_control);
  }

  shell()->ShowDevTools();
  EXPECT_TRUE(NavigateToURL(shell(), url));

  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(6UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[4].relative_url);
    EXPECT_EQ(kMaxAgeCacheControl, request_log_[4].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[5].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[5].cache_control);
  }

  shell()->CloseDevTools();
  EXPECT_TRUE(NavigateToURL(shell(), url));

  {
    base::AutoLock lock(request_log_lock_);
    ASSERT_EQ(8UL, request_log_.size());
    EXPECT_EQ(kReloadTestPath, request_log_[6].relative_url);
    EXPECT_EQ(kMaxAgeCacheControl, request_log_[6].cache_control);
    EXPECT_EQ(kReloadImagePath, request_log_[7].relative_url);
    EXPECT_EQ(kNoCacheControl, request_log_[7].cache_control);
  }
}

}  // namespace

}  // namespace content