1// Copyright (c) 2012 The Chromium Authors. All rights reserved.
2// Use of this source code is governed by a BSD-style license that can be
3// found in the LICENSE file.
4
5#include "chrome/browser/history/history_database.h"
6
7#include "base/file_util.h"
8#include "base/files/file_path.h"
9#include "base/files/scoped_temp_dir.h"
10#include "base/path_service.h"
11#include "chrome/common/chrome_paths.h"
12#include "sql/init_status.h"
13#include "testing/gtest/include/gtest/gtest.h"
14
15namespace history {
16
17TEST(HistoryDatabaseTest, DropBookmarks) {
18  base::ScopedTempDir temp_dir;
19  base::FilePath db_file;
20
21  ASSERT_TRUE(temp_dir.CreateUniqueTempDir());
22  db_file = temp_dir.path().AppendASCII("DropBookmarks.db");
23  sql::Connection::Delete(db_file);
24
25  // Copy db file over that contains starred URLs.
26  base::FilePath old_history_path;
27  PathService::Get(chrome::DIR_TEST_DATA, &old_history_path);
28  old_history_path = old_history_path.AppendASCII("bookmarks");
29  old_history_path = old_history_path.Append(
30      FILE_PATH_LITERAL("History_with_starred"));
31  base::CopyFile(old_history_path, db_file);
32
33  // Load the DB twice. The first time it should migrate. Make sure that the
34  // migration leaves it in a state fit to load again later.
35  for (int i = 0; i < 2; ++i) {
36    HistoryDatabase history_db;
37    ASSERT_EQ(sql::INIT_OK, history_db.Init(db_file));
38    HistoryDatabase::URLEnumerator url_enumerator;
39    ASSERT_TRUE(history_db.InitURLEnumeratorForEverything(&url_enumerator));
40    int num_urls = 0;
41    URLRow url_row;
42    while (url_enumerator.GetNextURL(&url_row)) {
43      ++num_urls;
44    }
45    ASSERT_EQ(5, num_urls);
46  }
47}
48
49}  // namespace history
50