| OLD | NEW |
| (Empty) |
| 1 // Copyright (c) 2011 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 <string> | |
| 6 | |
| 7 #include "app/sql/connection.h" | |
| 8 #include "app/sql/statement.h" | |
| 9 #include "base/file_util.h" | |
| 10 #include "base/scoped_temp_dir.h" | |
| 11 #include "testing/gtest/include/gtest/gtest.h" | |
| 12 #include "third_party/sqlite/sqlite3.h" | |
| 13 | |
| 14 class StatementErrorHandler : public sql::ErrorDelegate { | |
| 15 public: | |
| 16 StatementErrorHandler() : error_(SQLITE_OK) {} | |
| 17 | |
| 18 virtual int OnError(int error, sql::Connection* connection, | |
| 19 sql::Statement* stmt) { | |
| 20 error_ = error; | |
| 21 const char* sql_txt = stmt ? stmt->GetSQLStatement() : NULL; | |
| 22 sql_text_ = sql_txt ? sql_txt : "no statement available"; | |
| 23 return error; | |
| 24 } | |
| 25 | |
| 26 int error() const { return error_; } | |
| 27 | |
| 28 void reset_error() { | |
| 29 sql_text_.clear(); | |
| 30 error_ = SQLITE_OK; | |
| 31 } | |
| 32 | |
| 33 const char* sql_statement() const { return sql_text_.c_str(); } | |
| 34 | |
| 35 private: | |
| 36 int error_; | |
| 37 std::string sql_text_; | |
| 38 }; | |
| 39 | |
| 40 class SQLStatementTest : public testing::Test { | |
| 41 public: | |
| 42 SQLStatementTest() : error_handler_(new StatementErrorHandler) {} | |
| 43 | |
| 44 void SetUp() { | |
| 45 ASSERT_TRUE(temp_dir_.CreateUniqueTempDir()); | |
| 46 ASSERT_TRUE(db_.Open(temp_dir_.path().AppendASCII("SQLStatementTest.db"))); | |
| 47 | |
| 48 // The |error_handler_| will be called if any sqlite statement operation | |
| 49 // returns an error code. | |
| 50 db_.set_error_delegate(error_handler_); | |
| 51 } | |
| 52 | |
| 53 void TearDown() { | |
| 54 // If any error happened the original sql statement can be found in | |
| 55 // error_handler_->sql_statement(). | |
| 56 EXPECT_EQ(SQLITE_OK, error_handler_->error()); | |
| 57 db_.Close(); | |
| 58 } | |
| 59 | |
| 60 sql::Connection& db() { return db_; } | |
| 61 | |
| 62 int sqlite_error() const { return error_handler_->error(); } | |
| 63 void reset_error() const { error_handler_->reset_error(); } | |
| 64 | |
| 65 private: | |
| 66 ScopedTempDir temp_dir_; | |
| 67 sql::Connection db_; | |
| 68 scoped_refptr<StatementErrorHandler> error_handler_; | |
| 69 }; | |
| 70 | |
| 71 TEST_F(SQLStatementTest, Assign) { | |
| 72 sql::Statement s; | |
| 73 EXPECT_FALSE(s); // bool conversion operator. | |
| 74 EXPECT_TRUE(!s); // ! operator. | |
| 75 EXPECT_FALSE(s.is_valid()); | |
| 76 | |
| 77 s.Assign(db().GetUniqueStatement("CREATE TABLE foo (a, b)")); | |
| 78 EXPECT_TRUE(s); | |
| 79 EXPECT_FALSE(!s); | |
| 80 EXPECT_TRUE(s.is_valid()); | |
| 81 } | |
| 82 | |
| 83 TEST_F(SQLStatementTest, Run) { | |
| 84 ASSERT_TRUE(db().Execute("CREATE TABLE foo (a, b)")); | |
| 85 ASSERT_TRUE(db().Execute("INSERT INTO foo (a, b) VALUES (3, 12)")); | |
| 86 | |
| 87 sql::Statement s(db().GetUniqueStatement("SELECT b FROM foo WHERE a=?")); | |
| 88 EXPECT_FALSE(s.Succeeded()); | |
| 89 | |
| 90 // Stepping it won't work since we haven't bound the value. | |
| 91 EXPECT_FALSE(s.Step()); | |
| 92 | |
| 93 // Run should fail since this produces output, and we should use Step(). This | |
| 94 // gets a bit wonky since sqlite says this is OK so succeeded is set. | |
| 95 s.Reset(); | |
| 96 s.BindInt(0, 3); | |
| 97 EXPECT_FALSE(s.Run()); | |
| 98 EXPECT_EQ(SQLITE_ROW, db().GetErrorCode()); | |
| 99 EXPECT_TRUE(s.Succeeded()); | |
| 100 | |
| 101 // Resetting it should put it back to the previous state (not runnable). | |
| 102 s.Reset(); | |
| 103 EXPECT_FALSE(s.Succeeded()); | |
| 104 | |
| 105 // Binding and stepping should produce one row. | |
| 106 s.BindInt(0, 3); | |
| 107 EXPECT_TRUE(s.Step()); | |
| 108 EXPECT_TRUE(s.Succeeded()); | |
| 109 EXPECT_EQ(12, s.ColumnInt(0)); | |
| 110 EXPECT_FALSE(s.Step()); | |
| 111 EXPECT_TRUE(s.Succeeded()); | |
| 112 } | |
| 113 | |
| 114 TEST_F(SQLStatementTest, BasicErrorCallback) { | |
| 115 ASSERT_TRUE(db().Execute("CREATE TABLE foo (a INTEGER PRIMARY KEY, b)")); | |
| 116 EXPECT_EQ(SQLITE_OK, sqlite_error()); | |
| 117 // Insert in the foo table the primary key. It is an error to insert | |
| 118 // something other than an number. This error causes the error callback | |
| 119 // handler to be called with SQLITE_MISMATCH as error code. | |
| 120 sql::Statement s(db().GetUniqueStatement("INSERT INTO foo (a) VALUES (?)")); | |
| 121 EXPECT_TRUE(s.is_valid()); | |
| 122 s.BindCString(0, "bad bad"); | |
| 123 EXPECT_FALSE(s.Run()); | |
| 124 EXPECT_EQ(SQLITE_MISMATCH, sqlite_error()); | |
| 125 reset_error(); | |
| 126 } | |
| OLD | NEW |