Upstream oodles of Chrome for Android code into Chromium.

chrome/browser/android/history_report/delta_file_backend_leveldb_unittest.cc

+// Copyright 2015 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 "chrome/browser/android/history_report/delta_file_backend_leveldb.h"
+
+#include <vector>
+
+#include "base/files/file_path.h"
+#include "base/files/scoped_temp_dir.h"
+#include "base/memory/scoped_ptr.h"
+#include "chrome/browser/android/history_report/delta_file_commons.h"
+#include "testing/gtest/include/gtest/gtest.h"
+#include "url/gurl.h"
+
+namespace history_report {
+
+class DeltaFileBackendTest : public testing::Test {
+ public:
+  DeltaFileBackendTest() {}
+  ~DeltaFileBackendTest() override {}
+
+ protected:
+  void SetUp() override {
+    ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
+    backend_.reset(new DeltaFileBackend(temp_dir_.path()));
+  }
+
+  scoped_ptr<DeltaFileBackend> backend_;
+
+ private:
+  base::ScopedTempDir temp_dir_;
+
+  DISALLOW_COPY_AND_ASSIGN(DeltaFileBackendTest);
+};
+
+TEST_F(DeltaFileBackendTest, AddPage) {
+  GURL test_url("test.org");
+  backend_->PageAdded(test_url);
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(1u, result->size());
+  DeltaFileEntryWithData data = (*result)[0];
+  EXPECT_EQ(1, data.SeqNo());
+  EXPECT_EQ(data.Type(), "add");
+  EXPECT_EQ(data.Url(), test_url.spec());
+}
+
+TEST_F(DeltaFileBackendTest, DelPage) {
+  GURL test_url("test.org");
+  backend_->PageDeleted(test_url);
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(1u, result->size());
+  DeltaFileEntryWithData data = (*result)[0];
+  EXPECT_EQ(1, data.SeqNo());
+  EXPECT_EQ(data.Type(), "del");
+  EXPECT_EQ(data.Url(), test_url.spec());
+}
+
+TEST_F(DeltaFileBackendTest, Recreate) {
+  GURL test_url("test.org");
+  // Adding 5 deletion entries for test.org
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+
+  std::vector<std::string> urls;
+  urls.push_back("test.org");
+  urls.push_back("test2.org");
+  EXPECT_TRUE(backend_->Recreate(urls));
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(2u, result->size());
+  DeltaFileEntryWithData data = (*result)[0];
+  EXPECT_EQ(1, data.SeqNo());
+  EXPECT_EQ(data.Type(), "add");
+  EXPECT_EQ(data.Url(), "test.org");
+  DeltaFileEntryWithData data2 = (*result)[1];
+  EXPECT_EQ(2, data2.SeqNo());
+  EXPECT_EQ(data2.Type(), "add");
+  EXPECT_EQ(data2.Url(), "test2.org");
+}
+
+TEST_F(DeltaFileBackendTest, Clear) {
+  GURL test_url("test.org");
+  // Adding 5 deletion entries for test.org
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->Clear();
+
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(0u, result->size());
+}
+
+TEST_F(DeltaFileBackendTest, QueryStart) {
+  GURL test_url("test.org");
+  // Adding 5 deletion entries for test.org
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+
+  // Skip first entry (start with sequence number == 2).
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(1, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(4u, result->size());
+  DeltaFileEntryWithData data = (*result)[0];
+  // Check that first result is the second entry we added.
+  EXPECT_EQ(2, data.SeqNo());
+  EXPECT_EQ(data.Type(), "del");
+  EXPECT_EQ(data.Url(), test_url.spec());
+}
+
+TEST_F(DeltaFileBackendTest, QueryLimit) {
+  GURL test_url("test.org");
+  // Adding 5 deletion entries for test.org
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+
+  // Query for up to 3 results.
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 3);
+  EXPECT_TRUE(result.get() != NULL);
+  // Check that we got exactly 3 results
+  EXPECT_EQ(3u, result->size());
+  // Verify that first result is an entry we added first.
+  DeltaFileEntryWithData data = (*result)[0];
+  EXPECT_EQ(1, data.SeqNo());
+  EXPECT_EQ(data.Type(), "del");
+  EXPECT_EQ(data.Url(), test_url.spec());
+}
+
+TEST_F(DeltaFileBackendTest, Trim) {
+  GURL test_url("test.org");
+  // Adding 5 deletion entries for test.org
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  // Trim all entries with sequence number <= 3.
+  int64 max_seq_no = backend_->Trim(3);
+  EXPECT_EQ(5, max_seq_no);
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(2u, result->size());
+  DeltaFileEntryWithData data = (*result)[0];
+  // First entry in delta file should now have sequence number == 4 because
+  // entries with smaller sequence numbers were deleted.
+  EXPECT_EQ(4, data.SeqNo());
+  EXPECT_EQ(data.Type(), "del");
+  EXPECT_EQ(data.Url(), test_url.spec());
+}
+
+TEST_F(DeltaFileBackendTest, TrimLowerBoundEqualToMaxSeqNo) {
+  GURL test_url("test.org");
+  // Adding 5 deletion entries for test.org
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  // Trim all entries with sequence number <= 5 but leave at least one entry
+  // in delta file.
+  int64 max_seq_no = backend_->Trim(5);
+  EXPECT_EQ(5, max_seq_no);
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(1u, result->size());
+  DeltaFileEntryWithData data = (*result)[0];
+  // All entries but last were removed.
+  EXPECT_EQ(5, data.SeqNo());
+  EXPECT_EQ(data.Type(), "del");
+  EXPECT_EQ(data.Url(), test_url.spec());
+}
+
+TEST_F(DeltaFileBackendTest, TrimLowerBoundGreaterThanMaxSeqNo) {
+  GURL test_url("test.org");
+  // Adding 5 deletion entries for test.org
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  backend_->PageDeleted(test_url);
+  // Trim all entries with sequence number <= 6 but leave at least one entry
+  // in delta file.
+  int64 max_seq_no = backend_->Trim(6);
+  EXPECT_EQ(5, max_seq_no);
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(1u, result->size());
+  DeltaFileEntryWithData data = (*result)[0];
+  // All entries but last were removed.
+  EXPECT_EQ(5, data.SeqNo());
+  EXPECT_EQ(data.Type(), "del");
+  EXPECT_EQ(data.Url(), test_url.spec());
+}
+
+TEST_F(DeltaFileBackendTest, TrimDeltaFileWithSingleEntry) {
+  GURL test_url("test.org");
+  backend_->PageDeleted(test_url);
+  // Trim all entries with sequence number <= 1 but leave at least one entry
+  // in delta file. Should not remove any entries since there's only one
+  // in delta file.
+  int64 max_seq_no = backend_->Trim(1);
+  EXPECT_EQ(1, max_seq_no);
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(0, 10);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(1u, result->size());
+  DeltaFileEntryWithData data = (*result)[0];
+  // No entries removed because there was only one.
+  EXPECT_EQ(1, data.SeqNo());
+  EXPECT_EQ(data.Type(), "del");
+  EXPECT_EQ(data.Url(), test_url.spec());
+}
+
+TEST_F(DeltaFileBackendTest, LevelDbComparator) {
+  GURL test_url("test.org");
+  // Adding 50 deletion entries for test.org
+  for (int i = 0; i < 50; i++) {
+    backend_->PageDeleted(test_url);
+  }
+
+  // Skip first entry (start with sequence number == 2).
+  scoped_ptr<std::vector<DeltaFileEntryWithData> > result =
+      backend_->Query(1, 100);
+  EXPECT_TRUE(result.get() != NULL);
+  EXPECT_EQ(49u, result->size());
+  for (int i = 0; i < 49; i++) {
+    DeltaFileEntryWithData data = (*result)[i];
+    // +2 because we skipped first entry and
+    // sequence number starts with 1 not with 0.
+    EXPECT_EQ(i + 2, data.SeqNo());
+    EXPECT_EQ(data.Type(), "del");
+    EXPECT_EQ(data.Url(), test_url.spec());
+  }
+}
+
+} // namespace history_report