blob: 1aff80e9bbac20d11dfe40c9e90e0021166de30c [file] [log] [blame]
// 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 <string>
#include "base/bind.h"
#include "base/files/file_util.h"
#include "base/files/scoped_temp_dir.h"
#include "sql/connection.h"
#include "sql/statement.h"
#include "testing/gtest/include/gtest/gtest.h"
#include "third_party/sqlite/sqlite3.h"
// Test that certain features are/are-not enabled in our SQLite.
namespace {
void CaptureErrorCallback(int* error_pointer, std::string* sql_text,
int error, sql::Statement* stmt) {
*error_pointer = error;
const char* text = stmt ? stmt->GetSQLStatement() : NULL;
*sql_text = text ? text : "no statement available";
}
class SQLiteFeaturesTest : public testing::Test {
public:
SQLiteFeaturesTest() : error_(SQLITE_OK) {}
void SetUp() override {
ASSERT_TRUE(temp_dir_.CreateUniqueTempDir());
ASSERT_TRUE(db_.Open(temp_dir_.path().AppendASCII("SQLStatementTest.db")));
// The error delegate will set |error_| and |sql_text_| when any sqlite
// statement operation returns an error code.
db_.set_error_callback(base::Bind(&CaptureErrorCallback,
&error_, &sql_text_));
}
void TearDown() override {
// If any error happened the original sql statement can be found in
// |sql_text_|.
EXPECT_EQ(SQLITE_OK, error_);
db_.Close();
}
sql::Connection& db() { return db_; }
private:
base::ScopedTempDir temp_dir_;
sql::Connection db_;
// The error code of the most recent error.
int error_;
// Original statement which has caused the error.
std::string sql_text_;
};
// Do not include fts1 support, it is not useful, and nobody is
// looking at it.
TEST_F(SQLiteFeaturesTest, NoFTS1) {
ASSERT_EQ(SQLITE_ERROR, db().ExecuteAndReturnErrorCode(
"CREATE VIRTUAL TABLE foo USING fts1(x)"));
}
#if defined(SQLITE_ENABLE_FTS2)
// fts2 is used for older history files, so we're signed on for keeping our
// version up-to-date.
// TODO(shess): Think up a crazy way to get out from having to support
// this forever.
TEST_F(SQLiteFeaturesTest, FTS2) {
ASSERT_TRUE(db().Execute("CREATE VIRTUAL TABLE foo USING fts2(x)"));
}
// A standard SQLite will not include our patch. This includes iOS.
#if !defined(USE_SYSTEM_SQLITE)
// Chromium fts2 was patched to treat "foo*" as a prefix search, though the icu
// tokenizer will return it as two tokens {"foo", "*"}.
TEST_F(SQLiteFeaturesTest, FTS2_Prefix) {
const char kCreateSql[] =
"CREATE VIRTUAL TABLE foo USING fts2(x, tokenize icu)";
ASSERT_TRUE(db().Execute(kCreateSql));
ASSERT_TRUE(db().Execute("INSERT INTO foo (x) VALUES ('test')"));
sql::Statement s(db().GetUniqueStatement(
"SELECT x FROM foo WHERE x MATCH 'te*'"));
ASSERT_TRUE(s.Step());
EXPECT_EQ("test", s.ColumnString(0));
}
#endif
#endif
// fts3 is used for current history files, and also for WebDatabase.
TEST_F(SQLiteFeaturesTest, FTS3) {
ASSERT_TRUE(db().Execute("CREATE VIRTUAL TABLE foo USING fts3(x)"));
}
#if !defined(USE_SYSTEM_SQLITE)
// Test that fts3 doesn't need fts2's patch (see above).
TEST_F(SQLiteFeaturesTest, FTS3_Prefix) {
const char kCreateSql[] =
"CREATE VIRTUAL TABLE foo USING fts3(x, tokenize icu)";
ASSERT_TRUE(db().Execute(kCreateSql));
ASSERT_TRUE(db().Execute("INSERT INTO foo (x) VALUES ('test')"));
sql::Statement s(db().GetUniqueStatement(
"SELECT x FROM foo WHERE x MATCH 'te*'"));
ASSERT_TRUE(s.Step());
EXPECT_EQ("test", s.ColumnString(0));
}
#endif
} // namespace