1135446Strhodes// Copyright 2011 The Kyua Authors. 2193149Sdougb// All rights reserved. 3135446Strhodes// 4135446Strhodes// Redistribution and use in source and binary forms, with or without 5193149Sdougb// modification, are permitted provided that the following conditions are 6135446Strhodes// met: 7135446Strhodes// 8135446Strhodes// * Redistributions of source code must retain the above copyright 9135446Strhodes// notice, this list of conditions and the following disclaimer. 10135446Strhodes// * Redistributions in binary form must reproduce the above copyright 11135446Strhodes// notice, this list of conditions and the following disclaimer in the 12135446Strhodes// documentation and/or other materials provided with the distribution. 13135446Strhodes// * Neither the name of Google Inc. nor the names of its contributors 14135446Strhodes// may be used to endorse or promote products derived from this software 15135446Strhodes// without specific prior written permission. 16135446Strhodes// 17135446Strhodes// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 18193149Sdougb// "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 19135446Strhodes// LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 20135446Strhodes// A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 21135446Strhodes// OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 22135446Strhodes// SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 23193149Sdougb// LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 24170222Sdougb// DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 25135446Strhodes// THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 26135446Strhodes// (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 27135446Strhodes// OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 28135446Strhodes 29135446Strhodes#include "utils/sqlite/transaction.hpp" 30135446Strhodes 31135446Strhodes#include <atf-c++.hpp> 32135446Strhodes 33170222Sdougb#include "utils/format/macros.hpp" 34135446Strhodes#include "utils/sqlite/database.hpp" 35135446Strhodes#include "utils/sqlite/exceptions.hpp" 36135446Strhodes#include "utils/sqlite/statement.ipp" 37170222Sdougb 38135446Strhodesnamespace sqlite = utils::sqlite; 39170222Sdougb 40135446Strhodes 41135446Strhodesnamespace { 42170222Sdougb 43170222Sdougb 44135446Strhodes/// Ensures that a table has a single specific value in a column. 45135446Strhodes/// 46135446Strhodes/// \param db The SQLite database. 47135446Strhodes/// \param table_name The table to be checked. 48170222Sdougb/// \param column_name The column to be checked. 49135446Strhodes/// \param exp_value The value expected to be found in the column. 50135446Strhodes/// 51135446Strhodes/// \return True if the column contains a single value and it matches exp_value; 52170222Sdougb/// false if not. If the query fails, the calling test is marked as bad. 53135446Strhodesstatic bool 54170222Sdougbcheck_in_table(sqlite::database& db, const char* table_name, 55135446Strhodes const char* column_name, int exp_value) 56170222Sdougb{ 57170222Sdougb sqlite::statement stmt = db.create_statement( 58170222Sdougb F("SELECT * FROM %s WHERE %s == %s") % table_name % column_name % 59135446Strhodes exp_value); 60135446Strhodes if (!stmt.step()) 61170222Sdougb return false; 62170222Sdougb if (stmt.step()) 63135446Strhodes ATF_FAIL("More than one value found in table"); 64135446Strhodes return true; 65135446Strhodes} 66135446Strhodes 67135446Strhodes 68170222Sdougb} // anonymous namespace 69135446Strhodes 70135446Strhodes 71135446StrhodesATF_TEST_CASE_WITHOUT_HEAD(automatic_rollback); 72135446StrhodesATF_TEST_CASE_BODY(automatic_rollback) 73135446Strhodes{ 74135446Strhodes sqlite::database db = sqlite::database::in_memory(); 75170222Sdougb db.exec("CREATE TABLE t (col INTEGER PRIMARY KEY)"); 76135446Strhodes db.exec("INSERT INTO t VALUES (3)"); 77135446Strhodes { 78135446Strhodes sqlite::transaction tx = db.begin_transaction(); 79135446Strhodes db.exec("INSERT INTO t VALUES (5)"); 80135446Strhodes } 81135446Strhodes ATF_REQUIRE( check_in_table(db, "t", "col", 3)); 82 ATF_REQUIRE(!check_in_table(db, "t", "col", 5)); 83} 84 85 86ATF_TEST_CASE_WITHOUT_HEAD(explicit_commit); 87ATF_TEST_CASE_BODY(explicit_commit) 88{ 89 sqlite::database db = sqlite::database::in_memory(); 90 db.exec("CREATE TABLE t (col INTEGER PRIMARY KEY)"); 91 db.exec("INSERT INTO t VALUES (3)"); 92 { 93 sqlite::transaction tx = db.begin_transaction(); 94 db.exec("INSERT INTO t VALUES (5)"); 95 tx.commit(); 96 } 97 ATF_REQUIRE(check_in_table(db, "t", "col", 3)); 98 ATF_REQUIRE(check_in_table(db, "t", "col", 5)); 99} 100 101 102ATF_TEST_CASE_WITHOUT_HEAD(explicit_rollback); 103ATF_TEST_CASE_BODY(explicit_rollback) 104{ 105 sqlite::database db = sqlite::database::in_memory(); 106 db.exec("CREATE TABLE t (col INTEGER PRIMARY KEY)"); 107 db.exec("INSERT INTO t VALUES (3)"); 108 { 109 sqlite::transaction tx = db.begin_transaction(); 110 db.exec("INSERT INTO t VALUES (5)"); 111 tx.rollback(); 112 } 113 ATF_REQUIRE( check_in_table(db, "t", "col", 3)); 114 ATF_REQUIRE(!check_in_table(db, "t", "col", 5)); 115} 116 117 118ATF_TEST_CASE_WITHOUT_HEAD(nested_fail); 119ATF_TEST_CASE_BODY(nested_fail) 120{ 121 sqlite::database db = sqlite::database::in_memory(); 122 { 123 sqlite::transaction tx = db.begin_transaction(); 124 ATF_REQUIRE_THROW(sqlite::error, db.begin_transaction()); 125 } 126} 127 128 129ATF_INIT_TEST_CASES(tcs) 130{ 131 ATF_ADD_TEST_CASE(tcs, automatic_rollback); 132 ATF_ADD_TEST_CASE(tcs, explicit_commit); 133 ATF_ADD_TEST_CASE(tcs, explicit_rollback); 134 ATF_ADD_TEST_CASE(tcs, nested_fail); 135} 136