external/bsd/kyua-cli/dist/utils/sqlite/transaction_test.cpp

   1 // Copyright 2011 Google Inc.
   2 // All rights reserved.
   3 //
   4 // Redistribution and use in source and binary forms, with or without
   5 // modification, are permitted provided that the following conditions are
   6 // met:
   7 //
   8 // * Redistributions of source code must retain the above copyright
   9 //   notice, this list of conditions and the following disclaimer.
  10 // * Redistributions in binary form must reproduce the above copyright
  11 //   notice, this list of conditions and the following disclaimer in the
  12 //   documentation and/or other materials provided with the distribution.
  13 // * Neither the name of Google Inc. nor the names of its contributors
  14 //   may be used to endorse or promote products derived from this software
  15 //   without specific prior written permission.
  16 //
  17 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  18 // "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  19 // LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  20 // A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  21 // OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  22 // SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  23 // LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  24 // DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  25 // THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  26 // (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  27 // OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  28
  29 #include "utils/sqlite/transaction.hpp"
  30
  31 #include <atf-c++.hpp>
  32
  33 #include "utils/format/macros.hpp"
  34 #include "utils/sqlite/database.hpp"
  35 #include "utils/sqlite/exceptions.hpp"
  36 #include "utils/sqlite/statement.ipp"
  37
  38 namespace sqlite = utils::sqlite;
  39
  40
  41 namespace {
  42
  43
  44 /// Ensures that a table has a single specific value in a column.
  45 ///
  46 /// \param db The SQLite database.
  47 /// \param table_name The table to be checked.
  48 /// \param column_name The column to be checked.
  49 /// \param exp_value The value expected to be found in the column.
  50 ///
  51 /// \return True if the column contains a single value and it matches exp_value;
  52 /// false if not.  If the query fails, the calling test is marked as bad.
  53 static bool
  54 check_in_table(sqlite::database& db, const char* table_name,
  55                const char* column_name, int exp_value)
  56 {
  57     sqlite::statement stmt = db.create_statement(
  58         F("SELECT * FROM %s WHERE %s == %s") % table_name % column_name %
  59         exp_value);
  60     if (!stmt.step())
  61         return false;
  62     if (stmt.step())
  63         ATF_FAIL("More than one value found in table");
  64     return true;
  65 }
  66
  67
  68 }  // anonymous namespace
  69
  70
  71 ATF_TEST_CASE_WITHOUT_HEAD(automatic_rollback);
  72 ATF_TEST_CASE_BODY(automatic_rollback)
  73 {
  74     sqlite::database db = sqlite::database::in_memory();
  75     db.exec("CREATE TABLE t (col INTEGER PRIMARY KEY)");
  76     db.exec("INSERT INTO t VALUES (3)");
  77     {
  78         sqlite::transaction tx = db.begin_transaction();
  79         db.exec("INSERT INTO t VALUES (5)");
  80     }
  81     ATF_REQUIRE( check_in_table(db, "t", "col", 3));
  82     ATF_REQUIRE(!check_in_table(db, "t", "col", 5));
  83 }
  84
  85
  86 ATF_TEST_CASE_WITHOUT_HEAD(explicit_commit);
  87 ATF_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
 102 ATF_TEST_CASE_WITHOUT_HEAD(explicit_rollback);
 103 ATF_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
 118 ATF_TEST_CASE_WITHOUT_HEAD(nested_fail);
 119 ATF_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
 129 ATF_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 }