Index: app/sql/statement_unittest.cc |
=================================================================== |
--- app/sql/statement_unittest.cc (revision 27122) |
+++ app/sql/statement_unittest.cc (working copy) |
@@ -2,6 +2,8 @@ |
// Use of this source code is governed by a BSD-style license that can be |
// found in the LICENSE file. |
+#include <string> |
+ |
#include "app/sql/connection.h" |
#include "app/sql/statement.h" |
#include "base/file_path.h" |
@@ -10,20 +12,51 @@ |
#include "testing/gtest/include/gtest/gtest.h" |
#include "third_party/sqlite/preprocessed/sqlite3.h" |
+class StatementErrorHandler : public sql::ErrorDelegate { |
+ public: |
+ StatementErrorHandler() : error_(SQLITE_OK) {} |
+ |
+ virtual int OnError(int error, sql::Connection* connection, |
+ sql::Statement* stmt) { |
+ error_ = error; |
+ const char* sql_txt = stmt->GetSQLStatement(); |
+ sql_text_ = sql_txt ? sql_txt : "no statement available"; |
+ return error; |
+ } |
+ |
+ int error() const { return error_; } |
+ |
+ void reset_error() { |
+ sql_text_.clear(); |
+ error_ = SQLITE_OK; |
+ } |
+ |
+ const char* sql_statement() const { return sql_text_.c_str(); } |
+ |
+ private: |
+ int error_; |
+ std::string sql_text_; |
+}; |
+ |
class SQLStatementTest : public testing::Test { |
public: |
- SQLStatementTest() {} |
+ SQLStatementTest() : error_handler_(new StatementErrorHandler) {} |
void SetUp() { |
ASSERT_TRUE(PathService::Get(base::DIR_TEMP, &path_)); |
path_ = path_.AppendASCII("SQLStatementTest.db"); |
file_util::Delete(path_, false); |
ASSERT_TRUE(db_.Init(path_)); |
+ // The |error_handler_| will be called if any sqlite statement operation |
+ // returns an error code. |
+ db_.set_error_delegate(error_handler_); |
} |
void TearDown() { |
+ // If any error happened the original sql statement can be found in |
+ // error_handler_->sql_statement(). |
+ EXPECT_EQ(SQLITE_OK, error_handler_->error()); |
db_.Close(); |
- |
// If this fails something is going on with cleanup and later tests may |
// fail, so we want to identify problems right away. |
ASSERT_TRUE(file_util::Delete(path_, false)); |
@@ -31,9 +64,13 @@ |
sql::Connection& db() { return db_; } |
+ int sqlite_error() const { return error_handler_->error(); } |
+ void reset_error() const { error_handler_->reset_error(); } |
+ |
private: |
FilePath path_; |
sql::Connection db_; |
+ scoped_refptr<StatementErrorHandler> error_handler_; |
}; |
TEST_F(SQLStatementTest, Assign) { |
@@ -78,3 +115,17 @@ |
EXPECT_FALSE(s.Step()); |
EXPECT_TRUE(s.Succeeded()); |
} |
+ |
+TEST_F(SQLStatementTest, BasicErrorCallback) { |
+ ASSERT_TRUE(db().Execute("CREATE TABLE foo (a INTEGER PRIMARY KEY, b)")); |
+ EXPECT_EQ(SQLITE_OK, sqlite_error()); |
+ // Insert in the foo table the primary key. It is an error to insert |
+ // something other than an number. This error causes the error callback |
+ // handler to be called with SQLITE_MISMATCH as error code. |
+ sql::Statement s(db().GetUniqueStatement("INSERT INTO foo (a) VALUES (?)")); |
+ EXPECT_TRUE(s.is_valid()); |
+ s.BindCString(0, "bad bad"); |
+ EXPECT_FALSE(s.Run()); |
+ EXPECT_EQ(SQLITE_MISMATCH, sqlite_error()); |
+ reset_error(); |
+} |