15821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Copyright (c) 2011 The Chromium Authors. All rights reserved.
25821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// Use of this source code is governed by a BSD-style license that can be
35821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)// found in the LICENSE file.
45821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
55821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include <vector>
65821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
71320f92c476a1ad9d19dba2a48c72b75566198e9Primiano Tucci#include "base/files/file_util.h"
85821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "base/format_macros.h"
9868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/string_util.h"
10868fa2fe829687343ffae624259930155e16dbd8Torne (Richard Coles)#include "base/strings/stringprintf.h"
115821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "chrome/browser/history/history_unittest_base.h"
125821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)#include "sql/connection.h"
135821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
145821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)namespace history {
155821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
165821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HistoryUnitTestBase::~HistoryUnitTestBase() {
175821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
185821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
192a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)void HistoryUnitTestBase::ExecuteSQLScript(const base::FilePath& sql_path,
202a99a7e74a7f215066514fe81d2bfa6639d9edddTorne (Richard Coles)                                           const base::FilePath& db_path) {
215821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::string sql;
2258537e28ecd584eab876aee8be7156509866d23aTorne (Richard Coles)  ASSERT_TRUE(base::ReadFileToString(sql_path, &sql));
235821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
245821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // Replace the 'last_visit_time', 'visit_time', 'time_slot' values in this
255821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  // SQL with the current time.
265821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  int64 now = base::Time::Now().ToInternalValue();
275821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  std::vector<std::string> sql_time;
285821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  sql_time.push_back(base::StringPrintf("%" PRId64, now));  // last_visit_time
295821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  sql_time.push_back(base::StringPrintf("%" PRId64, now));  // visit_time
305821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  sql_time.push_back(base::StringPrintf("%" PRId64, now));  // time_slot
315821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  sql = ReplaceStringPlaceholders(sql, sql_time, NULL);
325821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
335821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  sql::Connection connection;
345821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ASSERT_TRUE(connection.Open(db_path));
355821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)  ASSERT_TRUE(connection.Execute(sql.c_str()));
365821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
375821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
385821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)HistoryUnitTestBase::HistoryUnitTestBase() {
395821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}
405821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)
415821806d5e7f356e8fa4b058a389a808ea183019Torne (Richard Coles)}  // namespace history
42