| Index: chrome/browser/visitedlink/visitedlink_unittest.cc
|
| diff --git a/chrome/browser/visitedlink/visitedlink_unittest.cc b/chrome/browser/visitedlink/visitedlink_unittest.cc
|
| deleted file mode 100644
|
| index 3cd85e698e8810b8b60ee5c9c345690b103c40f2..0000000000000000000000000000000000000000
|
| --- a/chrome/browser/visitedlink/visitedlink_unittest.cc
|
| +++ /dev/null
|
| @@ -1,759 +0,0 @@
|
| -// 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 <cstdio>
|
| -#include <string>
|
| -#include <vector>
|
| -
|
| -#include "base/file_util.h"
|
| -#include "base/message_loop.h"
|
| -#include "base/path_service.h"
|
| -#include "base/process_util.h"
|
| -#include "base/shared_memory.h"
|
| -#include "base/string_util.h"
|
| -#include "base/time.h"
|
| -#include "chrome/browser/visitedlink/visitedlink_delegate.h"
|
| -#include "chrome/browser/visitedlink/visitedlink_event_listener.h"
|
| -#include "chrome/browser/visitedlink/visitedlink_master.h"
|
| -#include "chrome/common/render_messages.h"
|
| -#include "chrome/renderer/visitedlink_slave.h"
|
| -#include "chrome/test/base/chrome_render_view_host_test_harness.h"
|
| -#include "chrome/test/base/testing_profile.h"
|
| -#include "content/public/browser/notification_service.h"
|
| -#include "content/public/browser/notification_types.h"
|
| -#include "content/public/test/mock_render_process_host.h"
|
| -#include "content/public/test/test_browser_context.h"
|
| -#include "content/public/test/test_browser_thread.h"
|
| -#include "content/public/test/test_renderer_host.h"
|
| -#include "googleurl/src/gurl.h"
|
| -#include "testing/gtest/include/gtest/gtest.h"
|
| -
|
| -using content::BrowserThread;
|
| -using content::MockRenderProcessHost;
|
| -using content::RenderViewHostTester;
|
| -
|
| -namespace {
|
| -
|
| -typedef std::vector<GURL> URLs;
|
| -
|
| -// a nice long URL that we can append numbers to to get new URLs
|
| -const char g_test_prefix[] =
|
| - "http://www.google.com/products/foo/index.html?id=45028640526508376&seq=";
|
| -const int g_test_count = 1000;
|
| -
|
| -// Returns a test URL for index |i|
|
| -GURL TestURL(int i) {
|
| - return GURL(StringPrintf("%s%d", g_test_prefix, i));
|
| -}
|
| -
|
| -std::vector<VisitedLinkSlave*> g_slaves;
|
| -
|
| -class TestVisitedLinkDelegate : public VisitedLinkDelegate {
|
| - public:
|
| - virtual bool AreEquivalentContexts(
|
| - content::BrowserContext* context1,
|
| - content::BrowserContext* context2) OVERRIDE;
|
| - virtual void RebuildTable(
|
| - const scoped_refptr<URLEnumerator>& enumerator) OVERRIDE;
|
| -
|
| - void AddURLForRebuild(const GURL& url);
|
| -
|
| - private:
|
| -
|
| - URLs rebuild_urls_;
|
| -};
|
| -
|
| -bool TestVisitedLinkDelegate::AreEquivalentContexts(
|
| - content::BrowserContext* context1, content::BrowserContext* context2) {
|
| - DCHECK_EQ(context1, context2);
|
| - return true; // Test only has one profile.
|
| -}
|
| -
|
| -void TestVisitedLinkDelegate::RebuildTable(
|
| - const scoped_refptr<URLEnumerator>& enumerator) {
|
| - for (URLs::const_iterator itr = rebuild_urls_.begin();
|
| - itr != rebuild_urls_.end();
|
| - ++itr)
|
| - enumerator->OnURL(*itr);
|
| - enumerator->OnComplete(true);
|
| -}
|
| -
|
| -void TestVisitedLinkDelegate::AddURLForRebuild(const GURL& url) {
|
| - rebuild_urls_.push_back(url);
|
| -}
|
| -
|
| -class TestURLIterator : public VisitedLinkMaster::URLIterator {
|
| - public:
|
| - explicit TestURLIterator(const URLs& urls);
|
| -
|
| - virtual const GURL& NextURL() OVERRIDE;
|
| - virtual bool HasNextURL() const OVERRIDE;
|
| -
|
| - private:
|
| - URLs::const_iterator iterator_;
|
| - URLs::const_iterator end_;
|
| -};
|
| -
|
| -TestURLIterator::TestURLIterator(const URLs& urls)
|
| - : iterator_(urls.begin()),
|
| - end_(urls.end()) {
|
| -}
|
| -
|
| -const GURL& TestURLIterator::NextURL() {
|
| - return *(iterator_++);
|
| -}
|
| -
|
| -bool TestURLIterator::HasNextURL() const {
|
| - return iterator_ != end_;
|
| -}
|
| -
|
| -} // namespace
|
| -
|
| -class TrackingVisitedLinkEventListener : public VisitedLinkMaster::Listener {
|
| - public:
|
| - TrackingVisitedLinkEventListener()
|
| - : reset_count_(0),
|
| - add_count_(0) {}
|
| -
|
| - virtual void NewTable(base::SharedMemory* table) {
|
| - if (table) {
|
| - for (std::vector<VisitedLinkSlave>::size_type i = 0;
|
| - i < g_slaves.size(); i++) {
|
| - base::SharedMemoryHandle new_handle = base::SharedMemory::NULLHandle();
|
| - table->ShareToProcess(base::GetCurrentProcessHandle(), &new_handle);
|
| - g_slaves[i]->OnUpdateVisitedLinks(new_handle);
|
| - }
|
| - }
|
| - }
|
| - virtual void Add(VisitedLinkCommon::Fingerprint) { add_count_++; }
|
| - virtual void Reset() { reset_count_++; }
|
| -
|
| - void SetUp() {
|
| - reset_count_ = 0;
|
| - add_count_ = 0;
|
| - }
|
| -
|
| - int reset_count() const { return reset_count_; }
|
| - int add_count() const { return add_count_; }
|
| -
|
| - private:
|
| - int reset_count_;
|
| - int add_count_;
|
| -};
|
| -
|
| -class VisitedLinkTest : public testing::Test {
|
| - protected:
|
| - VisitedLinkTest()
|
| - : ui_thread_(BrowserThread::UI, &message_loop_),
|
| - file_thread_(BrowserThread::FILE, &message_loop_) {}
|
| - // Initializes the visited link objects. Pass in the size that you want a
|
| - // freshly created table to be. 0 means use the default.
|
| - //
|
| - // |suppress_rebuild| is set when we're not testing rebuilding, see
|
| - // the VisitedLinkMaster constructor.
|
| - bool InitVisited(int initial_size, bool suppress_rebuild) {
|
| - // Initialize the visited link system.
|
| - master_.reset(new VisitedLinkMaster(new TrackingVisitedLinkEventListener(),
|
| - &delegate_,
|
| - suppress_rebuild, visited_file_,
|
| - initial_size));
|
| - return master_->Init();
|
| - }
|
| -
|
| - // May be called multiple times (some tests will do this to clear things,
|
| - // and TearDown will do this to make sure eveything is shiny before quitting.
|
| - void ClearDB() {
|
| - if (master_.get())
|
| - master_.reset(NULL);
|
| -
|
| - // Wait for all pending file I/O to be completed.
|
| - BrowserThread::GetBlockingPool()->FlushForTesting();
|
| - }
|
| -
|
| - // Loads the database from disk and makes sure that the same URLs are present
|
| - // as were generated by TestIO_Create(). This also checks the URLs with a
|
| - // slave to make sure it reads the data properly.
|
| - void Reload() {
|
| - // Clean up after our caller, who may have left the database open.
|
| - ClearDB();
|
| -
|
| - ASSERT_TRUE(InitVisited(0, true));
|
| - master_->DebugValidate();
|
| -
|
| - // check that the table has the proper number of entries
|
| - int used_count = master_->GetUsedCount();
|
| - ASSERT_EQ(used_count, g_test_count);
|
| -
|
| - // Create a slave database.
|
| - VisitedLinkSlave slave;
|
| - base::SharedMemoryHandle new_handle = base::SharedMemory::NULLHandle();
|
| - master_->shared_memory()->ShareToProcess(
|
| - base::GetCurrentProcessHandle(), &new_handle);
|
| - slave.OnUpdateVisitedLinks(new_handle);
|
| - g_slaves.push_back(&slave);
|
| -
|
| - bool found;
|
| - for (int i = 0; i < g_test_count; i++) {
|
| - GURL cur = TestURL(i);
|
| - found = master_->IsVisited(cur);
|
| - EXPECT_TRUE(found) << "URL " << i << "not found in master.";
|
| -
|
| - found = slave.IsVisited(cur);
|
| - EXPECT_TRUE(found) << "URL " << i << "not found in slave.";
|
| - }
|
| -
|
| - // test some random URL so we know that it returns false sometimes too
|
| - found = master_->IsVisited(GURL("http://unfound.site/"));
|
| - ASSERT_FALSE(found);
|
| - found = slave.IsVisited(GURL("http://unfound.site/"));
|
| - ASSERT_FALSE(found);
|
| -
|
| - master_->DebugValidate();
|
| -
|
| - g_slaves.clear();
|
| - }
|
| -
|
| - // testing::Test
|
| - virtual void SetUp() {
|
| - ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
|
| -
|
| - history_dir_ = temp_dir_.path().AppendASCII("VisitedLinkTest");
|
| - ASSERT_TRUE(file_util::CreateDirectory(history_dir_));
|
| -
|
| - visited_file_ = history_dir_.Append(FILE_PATH_LITERAL("VisitedLinks"));
|
| - }
|
| -
|
| - virtual void TearDown() {
|
| - ClearDB();
|
| - }
|
| -
|
| - base::ScopedTempDir temp_dir_;
|
| -
|
| - MessageLoop message_loop_;
|
| - content::TestBrowserThread ui_thread_;
|
| - content::TestBrowserThread file_thread_;
|
| -
|
| - // Filenames for the services;
|
| - FilePath history_dir_;
|
| - FilePath visited_file_;
|
| -
|
| - scoped_ptr<VisitedLinkMaster> master_;
|
| - TestVisitedLinkDelegate delegate_;
|
| -};
|
| -
|
| -// This test creates and reads some databases to make sure the data is
|
| -// preserved throughout those operations.
|
| -TEST_F(VisitedLinkTest, DatabaseIO) {
|
| - ASSERT_TRUE(InitVisited(0, true));
|
| -
|
| - for (int i = 0; i < g_test_count; i++)
|
| - master_->AddURL(TestURL(i));
|
| -
|
| - // Test that the database was written properly
|
| - Reload();
|
| -}
|
| -
|
| -// Checks that we can delete things properly when there are collisions.
|
| -TEST_F(VisitedLinkTest, Delete) {
|
| - static const int32 kInitialSize = 17;
|
| - ASSERT_TRUE(InitVisited(kInitialSize, true));
|
| -
|
| - // Add a cluster from 14-17 wrapping around to 0. These will all hash to the
|
| - // same value.
|
| - const VisitedLinkCommon::Fingerprint kFingerprint0 = kInitialSize * 0 + 14;
|
| - const VisitedLinkCommon::Fingerprint kFingerprint1 = kInitialSize * 1 + 14;
|
| - const VisitedLinkCommon::Fingerprint kFingerprint2 = kInitialSize * 2 + 14;
|
| - const VisitedLinkCommon::Fingerprint kFingerprint3 = kInitialSize * 3 + 14;
|
| - const VisitedLinkCommon::Fingerprint kFingerprint4 = kInitialSize * 4 + 14;
|
| - master_->AddFingerprint(kFingerprint0, false); // @14
|
| - master_->AddFingerprint(kFingerprint1, false); // @15
|
| - master_->AddFingerprint(kFingerprint2, false); // @16
|
| - master_->AddFingerprint(kFingerprint3, false); // @0
|
| - master_->AddFingerprint(kFingerprint4, false); // @1
|
| -
|
| - // Deleting 14 should move the next value up one slot (we do not specify an
|
| - // order).
|
| - EXPECT_EQ(kFingerprint3, master_->hash_table_[0]);
|
| - master_->DeleteFingerprint(kFingerprint3, false);
|
| - VisitedLinkCommon::Fingerprint zero_fingerprint = 0;
|
| - EXPECT_EQ(zero_fingerprint, master_->hash_table_[1]);
|
| - EXPECT_NE(zero_fingerprint, master_->hash_table_[0]);
|
| -
|
| - // Deleting the other four should leave the table empty.
|
| - master_->DeleteFingerprint(kFingerprint0, false);
|
| - master_->DeleteFingerprint(kFingerprint1, false);
|
| - master_->DeleteFingerprint(kFingerprint2, false);
|
| - master_->DeleteFingerprint(kFingerprint4, false);
|
| -
|
| - EXPECT_EQ(0, master_->used_items_);
|
| - for (int i = 0; i < kInitialSize; i++)
|
| - EXPECT_EQ(zero_fingerprint, master_->hash_table_[i]) <<
|
| - "Hash table has values in it.";
|
| -}
|
| -
|
| -// When we delete more than kBigDeleteThreshold we trigger different behavior
|
| -// where the entire file is rewritten.
|
| -TEST_F(VisitedLinkTest, BigDelete) {
|
| - ASSERT_TRUE(InitVisited(16381, true));
|
| -
|
| - // Add the base set of URLs that won't be deleted.
|
| - // Reload() will test for these.
|
| - for (int32 i = 0; i < g_test_count; i++)
|
| - master_->AddURL(TestURL(i));
|
| -
|
| - // Add more URLs than necessary to trigger this case.
|
| - const int kTestDeleteCount = VisitedLinkMaster::kBigDeleteThreshold + 2;
|
| - URLs urls_to_delete;
|
| - for (int32 i = g_test_count; i < g_test_count + kTestDeleteCount; i++) {
|
| - GURL url(TestURL(i));
|
| - master_->AddURL(url);
|
| - urls_to_delete.push_back(url);
|
| - }
|
| -
|
| - TestURLIterator iterator(urls_to_delete);
|
| - master_->DeleteURLs(&iterator);
|
| - master_->DebugValidate();
|
| -
|
| - Reload();
|
| -}
|
| -
|
| -TEST_F(VisitedLinkTest, DeleteAll) {
|
| - ASSERT_TRUE(InitVisited(0, true));
|
| -
|
| - {
|
| - VisitedLinkSlave slave;
|
| - base::SharedMemoryHandle new_handle = base::SharedMemory::NULLHandle();
|
| - master_->shared_memory()->ShareToProcess(
|
| - base::GetCurrentProcessHandle(), &new_handle);
|
| - slave.OnUpdateVisitedLinks(new_handle);
|
| - g_slaves.push_back(&slave);
|
| -
|
| - // Add the test URLs.
|
| - for (int i = 0; i < g_test_count; i++) {
|
| - master_->AddURL(TestURL(i));
|
| - ASSERT_EQ(i + 1, master_->GetUsedCount());
|
| - }
|
| - master_->DebugValidate();
|
| -
|
| - // Make sure the slave picked up the adds.
|
| - for (int i = 0; i < g_test_count; i++)
|
| - EXPECT_TRUE(slave.IsVisited(TestURL(i)));
|
| -
|
| - // Clear the table and make sure the slave picked it up.
|
| - master_->DeleteAllURLs();
|
| - EXPECT_EQ(0, master_->GetUsedCount());
|
| - for (int i = 0; i < g_test_count; i++) {
|
| - EXPECT_FALSE(master_->IsVisited(TestURL(i)));
|
| - EXPECT_FALSE(slave.IsVisited(TestURL(i)));
|
| - }
|
| -
|
| - // Close the database.
|
| - g_slaves.clear();
|
| - ClearDB();
|
| - }
|
| -
|
| - // Reopen and validate.
|
| - ASSERT_TRUE(InitVisited(0, true));
|
| - master_->DebugValidate();
|
| - EXPECT_EQ(0, master_->GetUsedCount());
|
| - for (int i = 0; i < g_test_count; i++)
|
| - EXPECT_FALSE(master_->IsVisited(TestURL(i)));
|
| -}
|
| -
|
| -// This tests that the master correctly resizes its tables when it gets too
|
| -// full, notifies its slaves of the change, and updates the disk.
|
| -TEST_F(VisitedLinkTest, Resizing) {
|
| - // Create a very small database.
|
| - const int32 initial_size = 17;
|
| - ASSERT_TRUE(InitVisited(initial_size, true));
|
| -
|
| - // ...and a slave
|
| - VisitedLinkSlave slave;
|
| - base::SharedMemoryHandle new_handle = base::SharedMemory::NULLHandle();
|
| - master_->shared_memory()->ShareToProcess(
|
| - base::GetCurrentProcessHandle(), &new_handle);
|
| - slave.OnUpdateVisitedLinks(new_handle);
|
| - g_slaves.push_back(&slave);
|
| -
|
| - int32 used_count = master_->GetUsedCount();
|
| - ASSERT_EQ(used_count, 0);
|
| -
|
| - for (int i = 0; i < g_test_count; i++) {
|
| - master_->AddURL(TestURL(i));
|
| - used_count = master_->GetUsedCount();
|
| - ASSERT_EQ(i + 1, used_count);
|
| - }
|
| -
|
| - // Verify that the table got resized sufficiently.
|
| - int32 table_size;
|
| - VisitedLinkCommon::Fingerprint* table;
|
| - master_->GetUsageStatistics(&table_size, &table);
|
| - used_count = master_->GetUsedCount();
|
| - ASSERT_GT(table_size, used_count);
|
| - ASSERT_EQ(used_count, g_test_count) <<
|
| - "table count doesn't match the # of things we added";
|
| -
|
| - // Verify that the slave got the resize message and has the same
|
| - // table information.
|
| - int32 child_table_size;
|
| - VisitedLinkCommon::Fingerprint* child_table;
|
| - slave.GetUsageStatistics(&child_table_size, &child_table);
|
| - ASSERT_EQ(table_size, child_table_size);
|
| - for (int32 i = 0; i < table_size; i++) {
|
| - ASSERT_EQ(table[i], child_table[i]);
|
| - }
|
| -
|
| - master_->DebugValidate();
|
| - g_slaves.clear();
|
| -
|
| - // This tests that the file is written correctly by reading it in using
|
| - // a new database.
|
| - Reload();
|
| -}
|
| -
|
| -// Tests that if the database doesn't exist, it will be rebuilt from history.
|
| -TEST_F(VisitedLinkTest, Rebuild) {
|
| - // Add half of our URLs to history. This needs to be done before we
|
| - // initialize the visited link DB.
|
| - int history_count = g_test_count / 2;
|
| - for (int i = 0; i < history_count; i++)
|
| - delegate_.AddURLForRebuild(TestURL(i));
|
| -
|
| - // Initialize the visited link DB. Since the visited links file doesn't exist
|
| - // and we don't suppress history rebuilding, this will load from history.
|
| - ASSERT_TRUE(InitVisited(0, false));
|
| -
|
| - // While the table is rebuilding, add the rest of the URLs to the visited
|
| - // link system. This isn't guaranteed to happen during the rebuild, so we
|
| - // can't be 100% sure we're testing the right thing, but in practice is.
|
| - // All the adds above will generally take some time queuing up on the
|
| - // history thread, and it will take a while to catch up to actually
|
| - // processing the rebuild that has queued behind it. We will generally
|
| - // finish adding all of the URLs before it has even found the first URL.
|
| - for (int i = history_count; i < g_test_count; i++)
|
| - master_->AddURL(TestURL(i));
|
| -
|
| - // Add one more and then delete it.
|
| - master_->AddURL(TestURL(g_test_count));
|
| - URLs urls_to_delete;
|
| - urls_to_delete.push_back(TestURL(g_test_count));
|
| - TestURLIterator iterator(urls_to_delete);
|
| - master_->DeleteURLs(&iterator);
|
| -
|
| - // Wait for the rebuild to complete. The task will terminate the message
|
| - // loop when the rebuild is done. There's no chance that the rebuild will
|
| - // complete before we set the task because the rebuild completion message
|
| - // is posted to the message loop; until we Run() it, rebuild can not
|
| - // complete.
|
| - master_->set_rebuild_complete_task(MessageLoop::QuitClosure());
|
| - MessageLoop::current()->Run();
|
| -
|
| - // Test that all URLs were written to the database properly.
|
| - Reload();
|
| -
|
| - // Make sure the extra one was *not* written (Reload won't test this).
|
| - EXPECT_FALSE(master_->IsVisited(TestURL(g_test_count)));
|
| -}
|
| -
|
| -// Test that importing a large number of URLs will work
|
| -TEST_F(VisitedLinkTest, BigImport) {
|
| - ASSERT_TRUE(InitVisited(0, false));
|
| -
|
| - // Before the table rebuilds, add a large number of URLs
|
| - int total_count = VisitedLinkMaster::kDefaultTableSize + 10;
|
| - for (int i = 0; i < total_count; i++)
|
| - master_->AddURL(TestURL(i));
|
| -
|
| - // Wait for the rebuild to complete.
|
| - master_->set_rebuild_complete_task(MessageLoop::QuitClosure());
|
| - MessageLoop::current()->Run();
|
| -
|
| - // Ensure that the right number of URLs are present
|
| - int used_count = master_->GetUsedCount();
|
| - ASSERT_EQ(used_count, total_count);
|
| -}
|
| -
|
| -TEST_F(VisitedLinkTest, Listener) {
|
| - ASSERT_TRUE(InitVisited(0, true));
|
| -
|
| - // Add test URLs.
|
| - for (int i = 0; i < g_test_count; i++) {
|
| - master_->AddURL(TestURL(i));
|
| - ASSERT_EQ(i + 1, master_->GetUsedCount());
|
| - }
|
| -
|
| - // Delete an URL.
|
| - URLs urls_to_delete;
|
| - urls_to_delete.push_back(TestURL(0));
|
| - TestURLIterator iterator(urls_to_delete);
|
| - master_->DeleteURLs(&iterator);
|
| -
|
| - // ... and all of the remaining ones.
|
| - master_->DeleteAllURLs();
|
| -
|
| - TrackingVisitedLinkEventListener* listener =
|
| - static_cast<TrackingVisitedLinkEventListener*>(master_->GetListener());
|
| -
|
| - // Verify that VisitedLinkMaster::Listener::Add was called for each added URL.
|
| - EXPECT_EQ(g_test_count, listener->add_count());
|
| - // Verify that VisitedLinkMaster::Listener::Reset was called both when one and
|
| - // all URLs are deleted.
|
| - EXPECT_EQ(2, listener->reset_count());
|
| -}
|
| -
|
| -// TODO(boliu): Inherit content::TestBrowserContext when componentized.
|
| -class VisitCountingProfile : public TestingProfile {
|
| - public:
|
| - VisitCountingProfile()
|
| - : add_count_(0),
|
| - add_event_count_(0),
|
| - reset_event_count_(0) {}
|
| -
|
| - void CountAddEvent(int by) {
|
| - add_count_ += by;
|
| - add_event_count_++;
|
| - }
|
| -
|
| - void CountResetEvent() {
|
| - reset_event_count_++;
|
| - }
|
| -
|
| - int add_count() const { return add_count_; }
|
| - int add_event_count() const { return add_event_count_; }
|
| - int reset_event_count() const { return reset_event_count_; }
|
| -
|
| - private:
|
| - int add_count_;
|
| - int add_event_count_;
|
| - int reset_event_count_;
|
| -};
|
| -
|
| -// Stub out as little as possible, borrowing from RenderProcessHost.
|
| -class VisitRelayingRenderProcessHost : public MockRenderProcessHost {
|
| - public:
|
| - explicit VisitRelayingRenderProcessHost(
|
| - content::BrowserContext* browser_context)
|
| - : MockRenderProcessHost(browser_context), widgets_(0) {
|
| - content::NotificationService::current()->Notify(
|
| - content::NOTIFICATION_RENDERER_PROCESS_CREATED,
|
| - content::Source<RenderProcessHost>(this),
|
| - content::NotificationService::NoDetails());
|
| - }
|
| - virtual ~VisitRelayingRenderProcessHost() {
|
| - content::NotificationService::current()->Notify(
|
| - content::NOTIFICATION_RENDERER_PROCESS_TERMINATED,
|
| - content::Source<content::RenderProcessHost>(this),
|
| - content::NotificationService::NoDetails());
|
| - }
|
| -
|
| - virtual void WidgetRestored() OVERRIDE { widgets_++; }
|
| - virtual void WidgetHidden() OVERRIDE { widgets_--; }
|
| - virtual int VisibleWidgetCount() const OVERRIDE { return widgets_; }
|
| -
|
| - virtual bool Send(IPC::Message* msg) OVERRIDE {
|
| - VisitCountingProfile* counting_profile =
|
| - static_cast<VisitCountingProfile*>(
|
| - GetBrowserContext());
|
| -
|
| - if (msg->type() == ChromeViewMsg_VisitedLink_Add::ID) {
|
| - PickleIterator iter(*msg);
|
| - std::vector<uint64> fingerprints;
|
| - CHECK(IPC::ReadParam(msg, &iter, &fingerprints));
|
| - counting_profile->CountAddEvent(fingerprints.size());
|
| - } else if (msg->type() == ChromeViewMsg_VisitedLink_Reset::ID) {
|
| - counting_profile->CountResetEvent();
|
| - }
|
| -
|
| - delete msg;
|
| - return true;
|
| - }
|
| -
|
| - private:
|
| - int widgets_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(VisitRelayingRenderProcessHost);
|
| -};
|
| -
|
| -class VisitedLinkRenderProcessHostFactory
|
| - : public content::RenderProcessHostFactory {
|
| - public:
|
| - VisitedLinkRenderProcessHostFactory()
|
| - : content::RenderProcessHostFactory() {}
|
| - virtual content::RenderProcessHost* CreateRenderProcessHost(
|
| - content::BrowserContext* browser_context) const OVERRIDE {
|
| - return new VisitRelayingRenderProcessHost(browser_context);
|
| - }
|
| -
|
| - private:
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(VisitedLinkRenderProcessHostFactory);
|
| -};
|
| -
|
| -// TODO(boliu): Inherit content::RenderViewHostTestHarness when componentized.
|
| -class VisitedLinkEventsTest : public ChromeRenderViewHostTestHarness {
|
| - public:
|
| - VisitedLinkEventsTest()
|
| - : ui_thread_(BrowserThread::UI, &message_loop_),
|
| - file_thread_(BrowserThread::FILE, &message_loop_) {}
|
| - virtual ~VisitedLinkEventsTest() {}
|
| - virtual void SetUp() {
|
| - browser_context_.reset(new VisitCountingProfile());
|
| - master_.reset(new VisitedLinkMaster(profile(), &delegate_));
|
| - master_->Init();
|
| - SetRenderProcessHostFactory(&vc_rph_factory_);
|
| - content::RenderViewHostTestHarness::SetUp();
|
| - }
|
| -
|
| - VisitCountingProfile* profile() const {
|
| - return static_cast<VisitCountingProfile*>(browser_context_.get());
|
| - }
|
| -
|
| - VisitedLinkMaster* master() const {
|
| - return master_.get();
|
| - }
|
| -
|
| - void WaitForCoalescense() {
|
| - // Let the timer fire.
|
| - MessageLoop::current()->PostDelayedTask(
|
| - FROM_HERE,
|
| - MessageLoop::QuitClosure(),
|
| - base::TimeDelta::FromMilliseconds(110));
|
| - MessageLoop::current()->Run();
|
| - }
|
| -
|
| - protected:
|
| - VisitedLinkRenderProcessHostFactory vc_rph_factory_;
|
| -
|
| - private:
|
| - TestVisitedLinkDelegate delegate_;
|
| - scoped_ptr<VisitedLinkMaster> master_;
|
| - content::TestBrowserThread ui_thread_;
|
| - content::TestBrowserThread file_thread_;
|
| -
|
| - DISALLOW_COPY_AND_ASSIGN(VisitedLinkEventsTest);
|
| -};
|
| -
|
| -TEST_F(VisitedLinkEventsTest, Coalescense) {
|
| - // add some URLs to master.
|
| - // Add a few URLs.
|
| - master()->AddURL(GURL("http://acidtests.org/"));
|
| - master()->AddURL(GURL("http://google.com/"));
|
| - master()->AddURL(GURL("http://chromium.org/"));
|
| - // Just for kicks, add a duplicate URL. This shouldn't increase the resulting
|
| - master()->AddURL(GURL("http://acidtests.org/"));
|
| -
|
| - // Make sure that coalescing actually occurs. There should be no links or
|
| - // events received by the renderer.
|
| - EXPECT_EQ(0, profile()->add_count());
|
| - EXPECT_EQ(0, profile()->add_event_count());
|
| -
|
| - WaitForCoalescense();
|
| -
|
| - // We now should have 3 entries added in 1 event.
|
| - EXPECT_EQ(3, profile()->add_count());
|
| - EXPECT_EQ(1, profile()->add_event_count());
|
| -
|
| - // Test whether the coalescing continues by adding a few more URLs.
|
| - master()->AddURL(GURL("http://google.com/chrome/"));
|
| - master()->AddURL(GURL("http://webkit.org/"));
|
| - master()->AddURL(GURL("http://acid3.acidtests.org/"));
|
| -
|
| - WaitForCoalescense();
|
| -
|
| - // We should have 6 entries added in 2 events.
|
| - EXPECT_EQ(6, profile()->add_count());
|
| - EXPECT_EQ(2, profile()->add_event_count());
|
| -
|
| - // Test whether duplicate entries produce add events.
|
| - master()->AddURL(GURL("http://acidtests.org/"));
|
| -
|
| - WaitForCoalescense();
|
| -
|
| - // We should have no change in results.
|
| - EXPECT_EQ(6, profile()->add_count());
|
| - EXPECT_EQ(2, profile()->add_event_count());
|
| -
|
| - // Ensure that the coalescing does not resume after resetting.
|
| - master()->AddURL(GURL("http://build.chromium.org/"));
|
| - master()->DeleteAllURLs();
|
| -
|
| - WaitForCoalescense();
|
| -
|
| - // We should have no change in results except for one new reset event.
|
| - EXPECT_EQ(6, profile()->add_count());
|
| - EXPECT_EQ(2, profile()->add_event_count());
|
| - EXPECT_EQ(1, profile()->reset_event_count());
|
| -}
|
| -
|
| -TEST_F(VisitedLinkEventsTest, Basics) {
|
| - RenderViewHostTester::For(rvh())->CreateRenderView(string16(),
|
| - MSG_ROUTING_NONE,
|
| - -1);
|
| -
|
| - // Add a few URLs.
|
| - master()->AddURL(GURL("http://acidtests.org/"));
|
| - master()->AddURL(GURL("http://google.com/"));
|
| - master()->AddURL(GURL("http://chromium.org/"));
|
| -
|
| - WaitForCoalescense();
|
| -
|
| - // We now should have 1 add event.
|
| - EXPECT_EQ(1, profile()->add_event_count());
|
| - EXPECT_EQ(0, profile()->reset_event_count());
|
| -
|
| - master()->DeleteAllURLs();
|
| -
|
| - WaitForCoalescense();
|
| -
|
| - // We should have no change in add results, plus one new reset event.
|
| - EXPECT_EQ(1, profile()->add_event_count());
|
| - EXPECT_EQ(1, profile()->reset_event_count());
|
| -}
|
| -
|
| -TEST_F(VisitedLinkEventsTest, TabVisibility) {
|
| - RenderViewHostTester::For(rvh())->CreateRenderView(string16(),
|
| - MSG_ROUTING_NONE,
|
| - -1);
|
| -
|
| - // Simulate tab becoming inactive.
|
| - RenderViewHostTester::For(rvh())->SimulateWasHidden();
|
| -
|
| - // Add a few URLs.
|
| - master()->AddURL(GURL("http://acidtests.org/"));
|
| - master()->AddURL(GURL("http://google.com/"));
|
| - master()->AddURL(GURL("http://chromium.org/"));
|
| -
|
| - WaitForCoalescense();
|
| -
|
| - // We shouldn't have any events.
|
| - EXPECT_EQ(0, profile()->add_event_count());
|
| - EXPECT_EQ(0, profile()->reset_event_count());
|
| -
|
| - // Simulate the tab becoming active.
|
| - RenderViewHostTester::For(rvh())->SimulateWasShown();
|
| -
|
| - // We should now have 3 add events, still no reset events.
|
| - EXPECT_EQ(1, profile()->add_event_count());
|
| - EXPECT_EQ(0, profile()->reset_event_count());
|
| -
|
| - // Deactivate the tab again.
|
| - RenderViewHostTester::For(rvh())->SimulateWasHidden();
|
| -
|
| - // Add a bunch of URLs (over 50) to exhaust the link event buffer.
|
| - for (int i = 0; i < 100; i++)
|
| - master()->AddURL(TestURL(i));
|
| -
|
| - WaitForCoalescense();
|
| -
|
| - // Again, no change in events until tab is active.
|
| - EXPECT_EQ(1, profile()->add_event_count());
|
| - EXPECT_EQ(0, profile()->reset_event_count());
|
| -
|
| - // Activate the tab.
|
| - RenderViewHostTester::For(rvh())->SimulateWasShown();
|
| -
|
| - // We should have only one more reset event.
|
| - EXPECT_EQ(1, profile()->add_event_count());
|
| - EXPECT_EQ(1, profile()->reset_event_count());
|
| -}
|
|
|