// Copyright (c) 2012 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 "base/memory/ref_counted.h"
#include "base/message_loop.h"
#include "base/memory/scoped_ptr.h"
#include "base/time.h"
#include "chrome/browser/common/url_database/url_database.h"
#include "chrome/browser/history/history.h"
#include "chrome/browser/history/history_service_factory.h"
#include "chrome/browser/history/history_types.h"
#include "chrome/browser/history/in_memory_database.h"
#include "chrome/browser/predictors/resource_prefetch_predictor.h"
#include "chrome/browser/predictors/resource_prefetch_predictor_tables.h"
#include "chrome/test/base/testing_profile.h"
#include "content/public/test/test_browser_thread.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "testing/gmock/include/gmock/gmock.h"

using testing::ContainerEq;
using testing::Pointee;
using testing::SetArgPointee;

namespace predictors {

typedef ResourcePrefetchPredictor::URLRequestSummary URLRequestSummary;
typedef ResourcePrefetchPredictorTables::UrlTableRow UrlTableRow;
typedef ResourcePrefetchPredictorTables::UrlTableRows UrlTableRows;

class MockResourcePrefetchPredictorTables
    : public ResourcePrefetchPredictorTables {
 public:
  MockResourcePrefetchPredictorTables() { }

  MOCK_METHOD1(GetAllRows, void(std::vector<UrlTableRow>* url_row_buffer));
  MOCK_METHOD2(UpdateRowsForUrl, void(const GURL& main_page_url,
                                      const UrlTableRows& row_buffer));
  MOCK_METHOD1(DeleteRowsForUrls, void(const std::vector<GURL>& urls));
  MOCK_METHOD0(DeleteAllRows, void());

 private:
  ~MockResourcePrefetchPredictorTables() { }
};

class ResourcePrefetchPredictorTest : public testing::Test {
 public:
  ResourcePrefetchPredictorTest();
  ~ResourcePrefetchPredictorTest();
  void SetUp() OVERRIDE;
  void TearDown() OVERRIDE;

 protected:
  void AddUrlToHistory(const std::string& url,
                       int visit_count,
                       time_t last_visit) {
    history::URLRow row = history::URLRow(GURL(url));
    row.set_visit_count(visit_count);
    row.set_last_visit(base::Time::FromTimeT(last_visit));
    url_db_->AddURL(row);
  }

  NavigationID CreateNavigationID(int process_id,
                                  int render_view_id,
                                  const std::string& main_frame_url) {
    NavigationID navigation_id;
    navigation_id.render_process_id = process_id;
    navigation_id.render_view_id = render_view_id;
    navigation_id.main_frame_url = GURL(main_frame_url);
    navigation_id.creation_time = base::TimeTicks::Now();
    return navigation_id;
  }

  ResourcePrefetchPredictor::URLRequestSummary CreateURLRequestSummary(
      int process_id,
      int render_view_id,
      const std::string& main_frame_url,
      const std::string& resource_url,
      ResourceType::Type resource_type,
      const std::string& mime_type,
      bool was_cached) {
    ResourcePrefetchPredictor::URLRequestSummary summary;
    summary.navigation_id = CreateNavigationID(process_id, render_view_id,
                                               main_frame_url);
    summary.resource_url = GURL(resource_url);
    summary.resource_type = resource_type;
    summary.mime_type = mime_type;
    summary.was_cached = was_cached;
    return summary;
  }

  bool URLRequestSummaryAreEqual(const URLRequestSummary& lhs,
                                 const URLRequestSummary& rhs) {
    return lhs.navigation_id == rhs.navigation_id &&
        lhs.resource_url == rhs.resource_url &&
        lhs.resource_type == rhs.resource_type &&
        lhs.mime_type == rhs.mime_type &&
        lhs.was_cached == rhs.was_cached;
  }

  void ResetPredictor() {
    ResourcePrefetchPredictor::Config config;
    config.max_urls_to_track = 3;
    config.min_url_visit_count = 2;
    config.max_resources_per_entry = 4;
    config.max_consecutive_misses = 2;
    predictor_.reset(new ResourcePrefetchPredictor(config, &profile_));
    predictor_->SetTablesForTesting(mock_tables_);
  }

  MessageLoop loop_;
  content::TestBrowserThread ui_thread_;
  content::TestBrowserThread db_thread_;
  TestingProfile profile_;

  scoped_ptr<ResourcePrefetchPredictor> predictor_;
  scoped_refptr<MockResourcePrefetchPredictorTables> mock_tables_;
  UrlTableRows test_url_rows_;
  history::URLDatabase* url_db_;
};

ResourcePrefetchPredictorTest::ResourcePrefetchPredictorTest()
    : loop_(MessageLoop::TYPE_DEFAULT),
      ui_thread_(content::BrowserThread::UI, &loop_),
      db_thread_(content::BrowserThread::DB, &loop_),
      predictor_(NULL),
      mock_tables_(new MockResourcePrefetchPredictorTables()) {
  test_url_rows_.push_back(UrlTableRow(
      "http://www.google.com",
      "http://google.com/style1.css",
      ResourceType::STYLESHEET,
      3, 2, 1, 1.0));
  test_url_rows_.push_back(UrlTableRow(
      "http://www.google.com",
      "http://google.com/script3.js",
      ResourceType::SCRIPT,
      4, 0, 1, 2.1));
  test_url_rows_.push_back(UrlTableRow(
      "http://www.google.com",
      "http://google.com/script4.js",
      ResourceType::SCRIPT,
      11, 0, 0, 2.1));
  test_url_rows_.push_back(UrlTableRow(
      "http://www.google.com",
      "http://google.com/image1.png",
      ResourceType::IMAGE,
      6, 3, 0, 2.2));
  test_url_rows_.push_back(UrlTableRow(
      "http://www.google.com",
      "http://google.com/a.font",
      ResourceType::LAST_TYPE,
      2, 0, 0, 5.1));

  test_url_rows_.push_back(UrlTableRow(
      "http://www.reddit.com",
      "http://reddit-resource.com/script1.js",
      ResourceType::SCRIPT,
      4, 0, 1, 1.0));
  test_url_rows_.push_back(UrlTableRow(
      "http://www.reddit.com",
      "http://reddit-resource.com/script2.js",
      ResourceType::SCRIPT,
      2, 0, 0, 2.1));

  test_url_rows_.push_back(UrlTableRow(
      "http://www.yahoo.com",
      "http://google.com/image.png",
      ResourceType::IMAGE,
      20, 1, 0, 10.0));
}

ResourcePrefetchPredictorTest::~ResourcePrefetchPredictorTest() {
}

void ResourcePrefetchPredictorTest::SetUp() {
  profile_.CreateHistoryService(true, false);
  profile_.BlockUntilHistoryProcessesPendingRequests();
  EXPECT_TRUE(HistoryServiceFactory::GetForProfile(&profile_,
                                                   Profile::EXPLICIT_ACCESS));
  url_db_ = HistoryServiceFactory::GetForProfile(&profile_,
      Profile::EXPLICIT_ACCESS)->InMemoryDatabase();

  // Initialize the predictor with empty data.
  ResetPredictor();
  EXPECT_EQ(predictor_->initialization_state_,
            ResourcePrefetchPredictor::NOT_INITIALIZED);
  EXPECT_CALL(*mock_tables_, GetAllRows(Pointee(ContainerEq(UrlTableRows()))));
  predictor_->LazilyInitialize();
  profile_.BlockUntilHistoryProcessesPendingRequests();
  EXPECT_TRUE(predictor_->inflight_navigations_.empty());
  EXPECT_EQ(predictor_->initialization_state_,
            ResourcePrefetchPredictor::INITIALIZED);
}

void ResourcePrefetchPredictorTest::TearDown() {
  predictor_.reset(NULL);
  profile_.DestroyHistoryService();
}

TEST_F(ResourcePrefetchPredictorTest, LazilyInitializeEmpty) {
  // Tests that the predictor initializes correctly without any data.
  EXPECT_TRUE(predictor_->url_table_cache_.empty());
}

TEST_F(ResourcePrefetchPredictorTest, LazilyInitializeWithData) {
  // Tests that the history and the db tables data are loaded correctly.
  AddUrlToHistory("http://www.google.com", 4, 12345);
  AddUrlToHistory("http://www.yahoo.com", 2, 12346);

  EXPECT_CALL(*mock_tables_, GetAllRows(Pointee(ContainerEq(UrlTableRows()))))
      .WillOnce(SetArgPointee<0>(test_url_rows_));

  std::vector<GURL> urls_to_delete;
  urls_to_delete.push_back(GURL("http://www.reddit.com"));
  EXPECT_CALL(*mock_tables_, DeleteRowsForUrls(ContainerEq(urls_to_delete)));

  ResetPredictor();
  predictor_->LazilyInitialize();
  loop_.RunAllPending();

  // Test that the internal variables correctly initialized.
  EXPECT_EQ(predictor_->initialization_state_,
            ResourcePrefetchPredictor::INITIALIZED);
  EXPECT_TRUE(predictor_->inflight_navigations_.empty());
  EXPECT_EQ(2, static_cast<int>(predictor_->url_table_cache_.size()));

  UrlTableRows google_rows(test_url_rows_.begin(),
                           test_url_rows_.begin() + 5);
  std::sort(google_rows.begin(), google_rows.end(),
            ResourcePrefetchPredictorTables::UrlTableRowSorter());
  EXPECT_EQ(predictor_->url_table_cache_[GURL("http://www.google.com")].rows,
            google_rows);
  EXPECT_EQ(
      predictor_->url_table_cache_[GURL("http://www.google.com")].last_visit,
      base::Time::FromTimeT(12345));

  UrlTableRows yahoo_rows(test_url_rows_.begin() + 7,
                          test_url_rows_.begin() + 8);
  EXPECT_EQ(predictor_->url_table_cache_[GURL("http://www.yahoo.com")].rows,
            yahoo_rows);
  EXPECT_EQ(
      predictor_->url_table_cache_[GURL("http://www.yahoo.com")].last_visit,
      base::Time::FromTimeT(12346));
}

TEST_F(ResourcePrefetchPredictorTest, NavigationNotRecorded) {
  // Single navigation but history count is low, so should not record.
  AddUrlToHistory("http://www.google.com", 1, 12345);

  URLRequestSummary main_frame = CreateURLRequestSummary(
      1, 1, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);
  predictor_->RecordURLRequest(main_frame);
  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));

  // Now add a few subresources.
  URLRequestSummary resource1 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/style1.css",
      ResourceType::STYLESHEET, "text/css", false);
  predictor_->RecordUrlResponse(resource1);
  URLRequestSummary resource2 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script1.js",
      ResourceType::SCRIPT, "text/javascript", false);
  predictor_->RecordUrlResponse(resource2);
  URLRequestSummary resource3 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script2.js",
      ResourceType::SCRIPT, "text/javascript", false);
  predictor_->RecordUrlResponse(resource3);
  predictor_->OnNavigationComplete(main_frame.navigation_id);
}

TEST_F(ResourcePrefetchPredictorTest, NavigationUrlNotInDB) {
  // Single navigation that will be recorded. Will check for duplicate
  // resources and also for number of resources saved.
  AddUrlToHistory("http://www.google.com", 4, 12345);

  URLRequestSummary main_frame = CreateURLRequestSummary(
      1, 1, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);
  predictor_->RecordURLRequest(main_frame);
  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));

  URLRequestSummary resource1 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/style1.css",
      ResourceType::STYLESHEET, "text/css", false);
  predictor_->RecordUrlResponse(resource1);
  URLRequestSummary resource2 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script1.js",
      ResourceType::SCRIPT, "text/javascript", false);
  predictor_->RecordUrlResponse(resource2);
  URLRequestSummary resource3 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script2.js",
      ResourceType::SCRIPT, "text/javascript", false);
  predictor_->RecordUrlResponse(resource3);
  URLRequestSummary resource4 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script1.js",
      ResourceType::SCRIPT, "text/javascript", true);
  predictor_->RecordUrlResponse(resource4);
  URLRequestSummary resource5 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/image1.png",
      ResourceType::IMAGE, "image/png", false);
  predictor_->RecordUrlResponse(resource5);
  URLRequestSummary resource6 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/image2.png",
      ResourceType::IMAGE, "image/png", false);
  predictor_->RecordUrlResponse(resource6);
  URLRequestSummary resource7 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/style2.css",
      ResourceType::STYLESHEET, "text/css", false);
  predictor_->OnSubresourceLoadedFromMemory(
      resource7.navigation_id,
      resource7.resource_url,
      resource7.mime_type,
      resource7.resource_type);

  UrlTableRows db_rows;
  db_rows.push_back(UrlTableRow(
      "http://www.google.com", "http://google.com/style1.css",
      ResourceType::STYLESHEET, 1, 0, 0, 1.0));
  db_rows.push_back(UrlTableRow(
      "http://www.google.com", "http://google.com/script1.js",
      ResourceType::SCRIPT, 1, 0, 0, 2.0));
  db_rows.push_back(UrlTableRow(
      "http://www.google.com", "http://google.com/script2.js",
      ResourceType::SCRIPT, 1, 0, 0, 3.0));
  db_rows.push_back(UrlTableRow(
      "http://www.google.com", "http://google.com/style2.css",
      ResourceType::STYLESHEET, 1, 0, 0, 7.0));
  EXPECT_CALL(*mock_tables_,
              UpdateRowsForUrl(GURL("http://www.google.com"),
                               ContainerEq(db_rows)));

  predictor_->OnNavigationComplete(main_frame.navigation_id);
}

TEST_F(ResourcePrefetchPredictorTest, NavigationUrlInDB) {
  // Tests that navigation is recoreded correctly for URL already present in
  // the database cache.
  AddUrlToHistory("http://www.google.com", 4, 12345);

  EXPECT_CALL(*mock_tables_, GetAllRows(Pointee(ContainerEq(UrlTableRows()))))
      .WillOnce(SetArgPointee<0>(test_url_rows_));

  std::vector<GURL> urls_to_delete;
  urls_to_delete.push_back(GURL("http://www.reddit.com"));
  urls_to_delete.push_back(GURL("http://www.yahoo.com"));
  EXPECT_CALL(*mock_tables_, DeleteRowsForUrls(ContainerEq(urls_to_delete)));

  ResetPredictor();
  predictor_->LazilyInitialize();
  loop_.RunAllPending();

  URLRequestSummary main_frame = CreateURLRequestSummary(
      1, 1, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);
  predictor_->RecordURLRequest(main_frame);
  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));

  URLRequestSummary resource1 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/style1.css",
      ResourceType::STYLESHEET, "text/css", false);
  predictor_->RecordUrlResponse(resource1);
  URLRequestSummary resource2 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script1.js",
      ResourceType::SCRIPT, "text/javascript", false);
  predictor_->RecordUrlResponse(resource2);
  URLRequestSummary resource3 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script2.js",
      ResourceType::SCRIPT, "text/javascript", false);
  predictor_->RecordUrlResponse(resource3);
  URLRequestSummary resource4 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script1.js",
      ResourceType::SCRIPT, "text/javascript", true);
  predictor_->RecordUrlResponse(resource4);
  URLRequestSummary resource5 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/image1.png",
      ResourceType::IMAGE, "image/png", false);
  predictor_->RecordUrlResponse(resource5);
  URLRequestSummary resource6 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/image2.png",
      ResourceType::IMAGE, "image/png", false);
  predictor_->RecordUrlResponse(resource6);
  URLRequestSummary resource7 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/style2.css",
      ResourceType::STYLESHEET, "text/css", false);
  predictor_->OnSubresourceLoadedFromMemory(
      resource7.navigation_id,
      resource7.resource_url,
      resource7.mime_type,
      resource7.resource_type);

  UrlTableRows db_rows;
  db_rows.push_back(UrlTableRow(
      "http://www.google.com", "http://google.com/style1.css",
      ResourceType::STYLESHEET, 4, 2, 0, 1.0));
  db_rows.push_back(UrlTableRow(
      "http://www.google.com", "http://google.com/script1.js",
      ResourceType::SCRIPT, 1, 0, 0, 2.0));
  db_rows.push_back(UrlTableRow(
      "http://www.google.com", "http://google.com/script4.js",
      ResourceType::SCRIPT, 11, 1, 1, 2.1));
  db_rows.push_back(UrlTableRow(
      "http://www.google.com", "http://google.com/script2.js",
      ResourceType::SCRIPT, 1, 0, 0, 3.0));
  EXPECT_CALL(*mock_tables_,
              UpdateRowsForUrl(GURL("http://www.google.com"),
                               ContainerEq(db_rows)));

  predictor_->OnNavigationComplete(main_frame.navigation_id);
}

TEST_F(ResourcePrefetchPredictorTest, NavigationUrlNotInDBAndDBFull) {
  // Tests that a URL is deleted before another is added if the cache is full.
  AddUrlToHistory("http://www.google.com", 4, 12345);
  AddUrlToHistory("http://www.reddit.com", 4, 12340);  // Should be deleted.
  AddUrlToHistory("http://www.yahoo.com", 4, 12350);
  AddUrlToHistory("http://www.nike.com", 4, 10000);

  EXPECT_CALL(*mock_tables_, GetAllRows(Pointee(ContainerEq(UrlTableRows()))))
      .WillOnce(SetArgPointee<0>(test_url_rows_));

  ResetPredictor();
  predictor_->LazilyInitialize();
  loop_.RunAllPending();
  EXPECT_EQ(3, static_cast<int>(predictor_->url_table_cache_.size()));

  URLRequestSummary main_frame = CreateURLRequestSummary(
      1, 1, "http://www.nike.com", "http://www.nike.com",
      ResourceType::MAIN_FRAME, "", false);
  predictor_->RecordURLRequest(main_frame);
  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));

  URLRequestSummary resource1 = CreateURLRequestSummary(
      1, 1, "http://www.nike.com",  "http://nike.com/style1.css",
      ResourceType::STYLESHEET, "text/css", false);
  predictor_->RecordUrlResponse(resource1);
  URLRequestSummary resource2 = CreateURLRequestSummary(
      1, 1, "http://www.nike.com",  "http://nike.com/image2.png",
      ResourceType::IMAGE, "image/png", false);
  predictor_->RecordUrlResponse(resource2);

  std::vector<GURL> urls_to_delete;
  urls_to_delete.push_back(GURL("http://www.reddit.com"));
  EXPECT_CALL(*mock_tables_, DeleteRowsForUrls(ContainerEq(urls_to_delete)));

  UrlTableRows db_rows;
  db_rows.push_back(UrlTableRow(
      "http://www.nike.com", "http://nike.com/style1.css",
      ResourceType::STYLESHEET, 1, 0, 0, 1.0));
  db_rows.push_back(UrlTableRow(
      "http://www.nike.com", "http://nike.com/image2.png",
      ResourceType::IMAGE, 1, 0, 0, 2.0));
  EXPECT_CALL(*mock_tables_,
              UpdateRowsForUrl(GURL("http://www.nike.com"),
                               ContainerEq(db_rows)));

  predictor_->OnNavigationComplete(main_frame.navigation_id);
}

TEST_F(ResourcePrefetchPredictorTest, DeleteUrls) {
  // Add some dummy entries to cache.
  predictor_->url_table_cache_[GURL("http://www.google.com")];
  predictor_->url_table_cache_[GURL("http://www.yahoo.com")];
  predictor_->url_table_cache_[GURL("http://www.apple.com")];
  predictor_->url_table_cache_[GURL("http://www.nike.com")];
  predictor_->url_table_cache_[GURL("http://www.reddit.com")];

  history::URLRows rows;
  rows.push_back(history::URLRow(GURL("http://www.apple.com")));
  rows.push_back(history::URLRow(GURL("http://www.yahoo.com")));

  std::vector<GURL> urls_to_delete;
  urls_to_delete.push_back(GURL("http://www.apple.com"));
  urls_to_delete.push_back(GURL("http://www.yahoo.com"));
  EXPECT_CALL(*mock_tables_, DeleteRowsForUrls(ContainerEq(urls_to_delete)));
  predictor_->DeleteUrls(rows);
  EXPECT_EQ(3, static_cast<int>(predictor_->url_table_cache_.size()));

  EXPECT_CALL(*mock_tables_, DeleteAllRows());

  predictor_->DeleteAllUrls();
  EXPECT_TRUE(predictor_->url_table_cache_.empty());
}

TEST_F(ResourcePrefetchPredictorTest, OnMainFrameRequest) {
  URLRequestSummary summary1 = CreateURLRequestSummary(
      1, 1, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);
  URLRequestSummary summary2 = CreateURLRequestSummary(
      1, 2, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);
  URLRequestSummary summary3 = CreateURLRequestSummary(
      2, 1, "http://www.yahoo.com", "http://www.yahoo.com",
      ResourceType::MAIN_FRAME, "", false);

  predictor_->OnMainFrameRequest(summary1);
  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
  predictor_->OnMainFrameRequest(summary2);
  EXPECT_EQ(2, static_cast<int>(predictor_->inflight_navigations_.size()));
  predictor_->OnMainFrameRequest(summary3);
  EXPECT_EQ(3, static_cast<int>(predictor_->inflight_navigations_.size()));

  // Insert anther with same navigation id. It should replace.
  URLRequestSummary summary4 = CreateURLRequestSummary(
      1, 1, "http://www.nike.com", "http://www.nike.com",
      ResourceType::MAIN_FRAME, "", false);
  URLRequestSummary summary5 = CreateURLRequestSummary(
      1, 2, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);

  predictor_->OnMainFrameRequest(summary4);
  EXPECT_EQ(3, static_cast<int>(predictor_->inflight_navigations_.size()));

  // Change this creation time so that it will go away on the next insert.
  summary5.navigation_id.creation_time = base::TimeTicks::Now() -
      base::TimeDelta::FromDays(1);
  predictor_->OnMainFrameRequest(summary5);
  EXPECT_EQ(3, static_cast<int>(predictor_->inflight_navigations_.size()));

  URLRequestSummary summary6 = CreateURLRequestSummary(
      3, 1, "http://www.shoes.com", "http://www.shoes.com",
      ResourceType::MAIN_FRAME, "", false);
  predictor_->OnMainFrameRequest(summary6);
  EXPECT_EQ(3, static_cast<int>(predictor_->inflight_navigations_.size()));

  EXPECT_TRUE(predictor_->inflight_navigations_.find(summary3.navigation_id) !=
              predictor_->inflight_navigations_.end());
  EXPECT_TRUE(predictor_->inflight_navigations_.find(summary4.navigation_id) !=
              predictor_->inflight_navigations_.end());
  EXPECT_TRUE(predictor_->inflight_navigations_.find(summary6.navigation_id) !=
              predictor_->inflight_navigations_.end());
}

TEST_F(ResourcePrefetchPredictorTest, OnMainFrameRedirect) {
  URLRequestSummary summary1 = CreateURLRequestSummary(
      1, 1, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);
  URLRequestSummary summary2 = CreateURLRequestSummary(
      1, 2, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);
  URLRequestSummary summary3 = CreateURLRequestSummary(
      2, 1, "http://www.yahoo.com", "http://www.yahoo.com",
      ResourceType::MAIN_FRAME, "", false);

  predictor_->OnMainFrameRedirect(summary1);
  EXPECT_TRUE(predictor_->inflight_navigations_.empty());

  predictor_->OnMainFrameRequest(summary1);
  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
  predictor_->OnMainFrameRequest(summary2);
  EXPECT_EQ(2, static_cast<int>(predictor_->inflight_navigations_.size()));

  predictor_->OnMainFrameRedirect(summary3);
  EXPECT_EQ(2, static_cast<int>(predictor_->inflight_navigations_.size()));
  predictor_->OnMainFrameRedirect(summary1);
  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
  predictor_->OnMainFrameRedirect(summary2);
  EXPECT_TRUE(predictor_->inflight_navigations_.empty());
}

TEST_F(ResourcePrefetchPredictorTest, OnSubresourceResponse) {
  // If there is no inflight navigation, nothing happens.
  URLRequestSummary resource1 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/style1.css",
      ResourceType::STYLESHEET, "text/css", false);
  predictor_->OnSubresourceResponse(resource1);
  EXPECT_TRUE(predictor_->inflight_navigations_.empty());

  // Add an inflight navigation.
  URLRequestSummary main_frame1 = CreateURLRequestSummary(
      1, 1, "http://www.google.com", "http://www.google.com",
      ResourceType::MAIN_FRAME, "", false);
  predictor_->OnMainFrameRequest(main_frame1);
  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));

  // Now add a few subresources.
  URLRequestSummary resource2 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script1.js",
      ResourceType::SCRIPT, "text/javascript", false);
  URLRequestSummary resource3 = CreateURLRequestSummary(
      1, 1, "http://www.google.com",  "http://google.com/script2.js",
      ResourceType::SCRIPT, "text/javascript", false);
  predictor_->OnSubresourceResponse(resource1);
  predictor_->OnSubresourceResponse(resource2);
  predictor_->OnSubresourceResponse(resource3);

  EXPECT_EQ(1, static_cast<int>(predictor_->inflight_navigations_.size()));
  EXPECT_EQ(3, static_cast<int>(
      predictor_->inflight_navigations_[main_frame1.navigation_id].size()));
  EXPECT_TRUE(URLRequestSummaryAreEqual(
      resource1,
      predictor_->inflight_navigations_[main_frame1.navigation_id][0]));
  EXPECT_TRUE(URLRequestSummaryAreEqual(
      resource2,
      predictor_->inflight_navigations_[main_frame1.navigation_id][1]));
  EXPECT_TRUE(URLRequestSummaryAreEqual(
      resource3,
      predictor_->inflight_navigations_[main_frame1.navigation_id][2]));
}

}  // namespace predictors