1 // Copyright 2011 The Kyua Authors.
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 }
136