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