/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: 2012-01-23 20:00:05 UTC
  • Revision ID: edam@waxworlds.org-20120123200005-038div0hpe78if8e
update TODO

Show diffs side-by-side

added added

removed removed

Lines of Context:
1
1
/*
2
2
 * test-database.cc
3
3
 *
4
 
 * Copyright (C) 2009 Tim Marston <edam@waxworlds.org>
 
4
 * Copyright (C) 2009 Tim Marston <tim@ed.am>
5
5
 *
6
6
 * This file is part of sqlite3cc (hereafter referred to as "this program").
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.
 
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.
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::database db( DBFILE );
 
37
        sqlite::connection conn( DBFILE );
 
38
 
 
39
        std::cout << "SQLite threading mode: " << sqlite::threadsafe() << "\n";
38
40
 
39
41
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
40
42
 
41
43
        // test basic statement, via command class
42
 
        sqlite::command c1( db,
 
44
        sqlite::command c1( conn,
43
45
                "CREATE TABLE pets ( "
44
46
                        "name TEXT PRIMARY KEY, "
45
47
                        "age INTEGER "
46
48
                ")" );
47
49
        c1.step();
48
50
 
49
 
        // test basic_statement bind() calls, via command
50
 
        sqlite::command c2( db, "INSERT INTO pets VALUES( ?, ? )" );
 
51
        // test basic_statement::bind() calls, via command
 
52
        sqlite::command c2( conn, "INSERT INTO pets VALUES( ?, ? )" );
51
53
        c2.bind( 1, "billy" ); c2.bind( 2, 12 ); c2.exec();
52
 
        sqlite::command c3( db, "INSERT INTO pets VALUES( ?6, ?9 )" );
 
54
        sqlite::command c3( conn, "INSERT INTO pets VALUES( ?6, ?9 )" );
53
55
        c3.bind( 9, 16 ); c3.bind( 6, "mopti" ); c3.exec();
54
 
        sqlite::command c6( db, "INSERT INTO pets VALUES( ?, ? )" );
 
56
        sqlite::command c6( conn, "INSERT INTO pets VALUES( ?, ? )" );
55
57
        c6.bind_null( 1 ); c6.bind( 2, 123 ); c6.exec();
56
 
        sqlite::command c7( db, "INSERT INTO pets VALUES( :foo, :bar )" );
 
58
        sqlite::command c7( conn, "INSERT INTO pets VALUES( :foo, :bar )" );
57
59
        c7.bind( ":foo", "foocat" ); c7.bind( ":bar", 22 ); c7.exec();
58
60
 
59
 
        // test basic_statement binding via stream operator
60
 
        sqlite::command c4( db, "INSERT INTO pets VALUES( ?, ? )" );
61
 
        c4 << "tessa" << 16 << sqlite::exec;
62
 
        sqlite::command( db, "INSERT INTO pets VALUES( ?, ? )" ) <<
63
 
                sqlite::null << sqlite::null <<
64
 
                sqlite::set_index( 1 ) << "tamara" << sqlite::exec;
 
61
        // test command binding via stream operator
 
62
        sqlite::command( conn, "INSERT INTO pets VALUES( ?, ? )" ) <<
 
63
                "tessa" << 16 << sqlite::exec;
 
64
        sqlite::command c4( conn, "INSERT INTO pets VALUES( ?, ? )" );
 
65
        c4 << sqlite::null << sqlite::null << sqlite::set_index( 1 ) <<
 
66
                "tamara" << sqlite::exec;
 
67
        assert( c4.changes() == 1 );
 
68
 
 
69
        // test query binding via stream operator
 
70
        sqlite::query( conn, "SELECT * FROM pets WHERE name = ? OR age = ?" )
 
71
                << "foo" << 16;
 
72
        sqlite::query( conn, "SELECT * FROM pets WHERE name = ? OR name = ? OR "
 
73
                "name = ? OR name = ?" ) << "foo" << sqlite::null << sqlite::null
 
74
                << sqlite::set_index( 1 ) << "bar" << "baz";
 
75
 
 
76
        // test connection command and query factory methods
 
77
        *conn.make_command( "PRAGMA user_version = 1") << sqlite::exec;
 
78
        conn.make_query( "PRAGMA user_version")->step();
65
79
 
66
80
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
67
81
 
68
82
        // test basic queries
69
 
        sqlite::query q1( db, "SELECT age, age + 3, name "
 
83
        sqlite::query q1( conn, "SELECT age, age + 3, name "
70
84
                "FROM pets WHERE NOT age IS NULL "
71
85
                "ORDER BY age ASC" );
 
86
 
 
87
        // test column count and names
72
88
        assert( q1.column_count() == 3 );
 
89
        assert( q1.column_name( 2 ) == "name" );
73
90
 
74
91
        // test rows
75
92
        unsigned int age;
76
93
        std::string name;
77
94
        q1.step().column( 0, age ); assert( age == 12 );
78
 
        q1.step().column( 2, name ); assert( name == "mopti" );
 
95
        name = q1.step().column< std::string >( 2 ); assert( name == "mopti" );
79
96
 
80
97
        // test row stream operator
81
98
        sqlite::row r1 = q1.step();
90
107
 
91
108
        // test NULL value handling
92
109
        assert( row.column_type( 2 ) == SQLITE_NULL );
93
 
        row.column( 2, age ); assert( age == 0 );
 
110
        age = row.column< unsigned int >( 2 ); assert( age == 0 );
94
111
        row.column( 2, name ); assert( name == "" );
95
112
 
96
113
        // test end of results
97
114
        assert( !q1.step() );
98
115
 
 
116
        // test num results
 
117
        sqlite::query q3( conn, "SELECT name FROM pets WHERE age = 16" );
 
118
        assert( q3.num_results() == 2 );
 
119
 
 
120
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
 
121
 
 
122
        // test query iterator
 
123
        sqlite::query q2( conn, "SELECT * FROM pets ORDER BY age DESC" );
 
124
        sqlite::query::iterator i1 = q2.begin();
 
125
        assert( i1 != q2.end() );
 
126
 
 
127
        // test resetting iterators
 
128
        unsigned int count = 0;
 
129
        for( sqlite::query::iterator i = q2.begin(); i != q2.end(); i++ )
 
130
        {
 
131
                // test iterator dereferencing and row numbers
 
132
                assert( count++ == i->row_number() );
 
133
                if( i->row_number() == 3 )
 
134
                        assert( i->column< std::string >( 0 ) == "tessa" &&
 
135
                                i->column< unsigned int >( 1 ) == 16 );
 
136
        }
 
137
 
 
138
//      for( sqlite::query::iterator i = q2.begin(); i != q2.end(); i++ ) {
 
139
//              for( unsigned int a = 0; a < q2.column_count(); a++ )
 
140
//                      std::cout << q2.column_name( a ) << "[" <<
 
141
//                      i->column< std::string >( a ) << "] ";
 
142
//              std::cout << "\n";
 
143
//      }
 
144
 
99
145
        // - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
100
146
 
101
147
        // test transactions
102
 
        {       sqlite::transaction_guard< > t1( db ); }
103
 
        {       sqlite::transaction_guard< sqlite::exclusive_transaction > t1( db ); }
104
 
        {       sqlite::transaction_guard< sqlite::recursive_transaction > t1( db );
105
 
                sqlite::transaction_guard< sqlite::recursive_transaction > t2( db ); }
 
148
        {
 
149
                sqlite::transaction_guard< sqlite::deferred_transaction > t1( conn );
 
150
        }
 
151
        {
 
152
                sqlite::transaction_guard< sqlite::immediate_transaction > t1( conn );
 
153
        }
 
154
        {
 
155
                sqlite::transaction_guard< sqlite::exclusive_transaction > t1( conn );
 
156
        }
 
157
        {
 
158
                sqlite::transaction_guard< sqlite::recursive_transaction > t1( conn );
 
159
                sqlite::transaction_guard< sqlite::recursive_transaction > t2( conn );
 
160
        }
106
161
 
107
162
        // test transaction guard rollback
108
163
        {
109
 
                sqlite::transaction_guard< > t1( db );
110
 
                db.exec( "UPDATE pets SET age = 99" );
 
164
                sqlite::transaction_guard< > t1( conn );
 
165
                conn.exec( "UPDATE pets SET age = 99" );
111
166
        }
112
 
        sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
 
167
        sqlite::query( conn, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
113
168
                .step() >> age; assert( age == 123 );
114
169
 
115
170
        // text recursive transactions
116
171
        {
117
 
                sqlite::transaction_guard< sqlite::recursive_transaction > t1( db );
118
 
                db.exec( "UPDATE pets SET age = 99" );
 
172
                sqlite::transaction_guard< sqlite::recursive_transaction > t1( conn );
 
173
                conn.exec( "UPDATE pets SET age = 66" );
119
174
 
120
175
                {
121
 
                        sqlite::transaction_guard< sqlite::recursive_transaction > t2( db );
122
 
                        db.exec( "UPDATE pets SET age = 1" );
 
176
                        sqlite::transaction_guard< sqlite::recursive_transaction > t2( conn );
 
177
                        conn.exec( "UPDATE pets SET age = 1" );
123
178
                }
124
179
 
125
 
                sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
126
 
                        .step() >> age; assert( age == 99 );
127
 
 
128
 
                t1.commit();
 
180
                sqlite::query( conn, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
 
181
                        .step() >> age; assert( age == 66 );
129
182
        }
130
 
        sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
131
 
                .step() >> age; assert( age == 99 );
 
183
        sqlite::query( conn, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
 
184
                .step() >> age; assert( age == 123 );
132
185
 
 
186
        // ok
133
187
        return 0;
134
188
}