/sqlite3cc

To get this branch, use:
bzr branch http://bzr.ed.am/sqlite3cc

« back to all changes in this revision

Viewing changes to test/test-main.cc

  • Committer: edam
  • Date: 2010-07-29 06:28:53 UTC
  • Revision ID: edam@waxworlds.org-20100729062853-4i2fec52m86mh724
- made basic_statement::step() protected, for use by query and command only
- moved basic_statement::operator<<() to command and query instead; one needs to accept sqlite::exec, the other doesn't
- added tests for query::operator<<()
- added code to invlaidate in-progress queries during any transaction rollbacks (currently segfaults in basic_statement::finalize())
- added new sqlite_error constructor that obtains a full error message
- implemented database::database_mutex_guard class
- swapped command's step mutex in favour of the database mutex

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
/*
2
 
 * test-main.cc
 
2
 * test-database.cc
3
3
 *
4
 
 * Copyright (C) 2009 Tim Marston <tim@ed.am>
 
4
 * Copyright (C) 2009 Tim Marston <edam@waxworlds.org>
5
5
 *
6
6
 * This file is part of sqlite3cc (hereafter referred to as "this program").
7
 
 * See http://ed.am/dev/sqlite3cc for more information.
8
 
 *
9
 
 * This program is free software: you can redistribute it and/or modify it under
10
 
 * the terms of the GNU Lesser General Public License as published by the Free
11
 
 * Software Foundation, either version 3 of the License, or (at your option) any
12
 
 * later version.
13
 
 *
14
 
 * This program is distributed in the hope that it will be useful, but WITHOUT
15
 
 * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
16
 
 * FOR A PARTICULAR PURPOSE.  See the GNU Lesser General Public License for more
17
 
 * details.
 
7
 * See http://www.waxworlds.org/edam/software/sqlite3cc for more information.
 
8
 *
 
9
 * This program is free software: you can redistribute it and/or modify
 
10
 * it under the terms of the GNU Lesser General Public License as published
 
11
 * by the Free Software Foundation, either version 3 of the License, or
 
12
 * (at your option) any later version.
 
13
 *
 
14
 * This program is distributed in the hope that it will be useful,
 
15
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
16
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
17
 * GNU Lesser General Public License for more details.
18
18
 *
19
19
 * You should have received a copy of the GNU Lesser General Public License
20
20
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
34
34
                boost::filesystem::remove( DBFILE );
35
35
 
36
36
        // open database
37
 
        sqlite::connection conn( DBFILE );
38
 
 
39
 
        std::cout << "SQLite threading mode: " << sqlite::threadsafe() << "\n";
 
37
        sqlite::database db( DBFILE );
40
38
 
41
39
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
42
40
 
43
41
        // test basic statement, via command class
44
 
        sqlite::command c1( conn,
 
42
        sqlite::command c1( db,
45
43
                "CREATE TABLE pets ( "
46
44
                        "name TEXT PRIMARY KEY, "
47
45
                        "age INTEGER "
48
46
                ")" );
49
47
        c1.step();
50
48
 
51
 
        // test basic_statement::bind() calls, via command
52
 
        sqlite::command c2( conn, "INSERT INTO pets VALUES( ?, ? )" );
 
49
        // test basic_statement bind() calls, via command
 
50
        sqlite::command c2( db, "INSERT INTO pets VALUES( ?, ? )" );
53
51
        c2.bind( 1, "billy" ); c2.bind( 2, 12 ); c2.exec();
54
 
        sqlite::command c3( conn, "INSERT INTO pets VALUES( ?6, ?9 )" );
 
52
        sqlite::command c3( db, "INSERT INTO pets VALUES( ?6, ?9 )" );
55
53
        c3.bind( 9, 16 ); c3.bind( 6, "mopti" ); c3.exec();
56
 
        sqlite::command c6( conn, "INSERT INTO pets VALUES( ?, ? )" );
 
54
        sqlite::command c6( db, "INSERT INTO pets VALUES( ?, ? )" );
57
55
        c6.bind_null( 1 ); c6.bind( 2, 123 ); c6.exec();
58
 
        sqlite::command c7( conn, "INSERT INTO pets VALUES( :foo, :bar )" );
59
 
        c7.bind( ":foo", "foocat" ); c7.bind( ":bar", 22.5 ); c7.exec();
 
56
        sqlite::command c7( db, "INSERT INTO pets VALUES( :foo, :bar )" );
 
57
        c7.bind( ":foo", "foocat" ); c7.bind( ":bar", 22 ); c7.exec();
60
58
 
61
59
        // test command binding via stream operator
62
 
        sqlite::command( conn, "INSERT INTO pets VALUES( ?, ? )" ) <<
 
60
        sqlite::command( db, "INSERT INTO pets VALUES( ?, ? )" ) <<
63
61
                "tessa" << 16 << sqlite::exec;
64
 
        sqlite::command c4( conn, "INSERT INTO pets VALUES( ?, ? )" );
 
62
        sqlite::command c4( db, "INSERT INTO pets VALUES( ?, ? )" );
65
63
        c4 << sqlite::null << sqlite::null << sqlite::set_index( 1 ) <<
66
64
                "tamara" << sqlite::exec;
67
65
        assert( c4.changes() == 1 );
68
66
 
69
67
        // test query binding via stream operator
70
 
        sqlite::query( conn, "SELECT * FROM pets WHERE name = ? OR age = ?" )
 
68
        sqlite::query( db, "SELECT * FROM pets WHERE name = ? OR age = ?" )
71
69
                << "foo" << 16;
72
 
        sqlite::query( conn, "SELECT * FROM pets WHERE name = ? OR name = ? OR "
 
70
        sqlite::query( db, "SELECT * FROM pets WHERE name = ? OR name = ? OR "
73
71
                "name = ? OR name = ?" ) << "foo" << sqlite::null << sqlite::null
74
72
                << sqlite::set_index( 1 ) << "bar" << "baz";
75
73
 
76
 
        // test connection command and query factory methods
77
 
        conn.make_command( "PRAGMA user_version = 12")->step();
78
 
        assert( conn.make_query( "PRAGMA user_version" )->step()
79
 
                .column< int >( 0 ) == 12 );
80
 
 
81
74
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
82
75
 
83
76
        // test basic queries
84
 
        sqlite::query q1( conn, "SELECT age, age + 3, name "
 
77
        sqlite::query q1( db, "SELECT age, age + 3, name "
85
78
                "FROM pets WHERE NOT age IS NULL "
86
79
                "ORDER BY age ASC" );
87
 
 
88
 
        // test column count and names
89
80
        assert( q1.column_count() == 3 );
90
 
        assert( q1.column_name( 2 ) == "name" );
91
81
 
92
82
        // test rows
93
83
        unsigned int age;
94
 
        double dblage;
95
84
        std::string name;
96
85
        q1.step().column( 0, age ); assert( age == 12 );
97
86
        name = q1.step().column< std::string >( 2 ); assert( name == "mopti" );
101
90
        r1 >> age; assert( age == 16 );
102
91
        r1 >> age; assert( age == 19 );
103
92
        r1 >> name; assert( name == "tessa" );
104
 
        q1.step() >> dblage >> sqlite::null >> name;
105
 
        assert( dblage == 22.5 && name == "foocat" );
 
93
        q1.step() >> age >> sqlite::null >> name;
 
94
        assert( age == 22 && name == "foocat" );
106
95
        sqlite::row row = q1.step();
107
96
        row >> age >> sqlite::null >> sqlite::null; assert( age == 123 );
108
97
        row >> sqlite::set_index( 1 ) >> age; assert( age == 126 );
115
104
        // test end of results
116
105
        assert( !q1.step() );
117
106
 
118
 
        // test num results
119
 
        sqlite::query q3( conn, "SELECT name FROM pets WHERE age = 16" );
120
 
        assert( q3.num_results() == 2 );
121
 
 
122
107
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
123
108
 
124
 
        // test query iterator
125
 
        sqlite::query q2( conn, "SELECT * FROM pets ORDER BY age DESC" );
126
 
        sqlite::query::iterator i1 = q2.begin();
127
 
        assert( i1 != q2.end() );
128
 
 
129
 
        // test resetting iterators
130
 
        unsigned int count = 0;
131
 
        for( sqlite::query::iterator i = q2.begin(); i != q2.end(); i++ )
132
 
        {
133
 
                // test iterator dereferencing and row numbers
134
 
                assert( count++ == i->row_number() );
135
 
                if( i->row_number() == 3 )
 
109
        // test query itterator
 
110
        sqlite::query q2( db, "SELECT * FROM pets ORDER BY age DESC" );
 
111
        for( sqlite::query::iterator i = q2.begin(); i != q2.end(); i++ ) {
 
112
                if( i->index() == 3 )
136
113
                        assert( i->column< std::string >( 0 ) == "tessa" &&
137
114
                                i->column< unsigned int >( 1 ) == 16 );
138
 
        }
139
 
 
140
 
//      for( sqlite::query::iterator i = q2.begin(); i != q2.end(); i++ ) {
141
115
//              for( unsigned int a = 0; a < q2.column_count(); a++ )
142
116
//                      std::cout << q2.column_name( a ) << "[" <<
143
117
//                      i->column< std::string >( a ) << "] ";
144
118
//              std::cout << "\n";
145
 
//      }
 
119
        }
 
120
        assert( q2.column_count() == 2 );
 
121
        assert( q2.column_name( 0 ) == "name" );
146
122
 
147
123
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
148
124
 
149
125
        // test transactions
150
 
        {
151
 
                sqlite::transaction_guard< sqlite::deferred_transaction > t1( conn );
152
 
        }
153
 
        {
154
 
                sqlite::transaction_guard< sqlite::immediate_transaction > t1( conn );
155
 
        }
156
 
        {
157
 
                sqlite::transaction_guard< sqlite::exclusive_transaction > t1( conn );
158
 
        }
159
 
        {
160
 
                sqlite::transaction_guard< sqlite::recursive_transaction > t1( conn );
161
 
                sqlite::transaction_guard< sqlite::recursive_transaction > t2( conn );
162
 
        }
 
126
        {       sqlite::transaction_guard< > t1( db ); }
 
127
        {       sqlite::transaction_guard< sqlite::exclusive_transaction > t1( db ); }
 
128
        {       sqlite::transaction_guard< sqlite::recursive_transaction > t1( db );
 
129
                sqlite::transaction_guard< sqlite::recursive_transaction > t2( db ); }
163
130
 
164
131
        // test transaction guard rollback
165
132
        {
166
 
                sqlite::transaction_guard< > t1( conn );
167
 
                conn.exec( "UPDATE pets SET age = 99" );
 
133
                sqlite::transaction_guard< > t1( db );
 
134
                db.exec( "UPDATE pets SET age = 99" );
168
135
        }
169
 
        sqlite::query( conn, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
 
136
        sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
170
137
                .step() >> age; assert( age == 123 );
171
138
 
172
 
        // test recursive transactions
 
139
        // text recursive transactions
173
140
        {
174
 
                sqlite::transaction_guard< sqlite::recursive_transaction > t1( conn );
175
 
                conn.exec( "UPDATE pets SET age = 66" );
 
141
                sqlite::transaction_guard< sqlite::recursive_transaction > t1( db );
 
142
                db.exec( "UPDATE pets SET age = 66" );
176
143
 
177
144
                {
178
 
                        sqlite::transaction_guard< sqlite::recursive_transaction > t2( conn );
179
 
                        conn.exec( "UPDATE pets SET age = 1" );
 
145
                        sqlite::transaction_guard< sqlite::recursive_transaction > t2( db );
 
146
                        db.exec( "UPDATE pets SET age = 1" );
180
147
                }
181
148
 
182
 
                sqlite::query( conn, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
 
149
                sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
183
150
                        .step() >> age; assert( age == 66 );
184
151
        }
185
 
        sqlite::query( conn, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
 
152
        sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
186
153
                .step() >> age; assert( age == 123 );
187
154
 
188
 
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
189
 
 
190
155
        // ok
191
156
        return 0;
192
157
}