/sqlite3cc

To get this branch, use:
bzr branch http://bzr.ed.am/sqlite3cc
1 by edam
- initial commit
1
/*
2 by edam
- further initial development
2
 * test-database.cc
1 by edam
- initial commit
3
 *
22 by edam
updated email and web addresses
4
 * Copyright (C) 2009 Tim Marston <tim@ed.am>
1 by edam
- initial commit
5
 *
2 by edam
- further initial development
6
 * This file is part of sqlite3cc (hereafter referred to as "this program").
22 by edam
updated email and web addresses
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.
1 by edam
- initial commit
18
 *
19
 * You should have received a copy of the GNU Lesser General Public License
20
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
21
 */
22
2 by edam
- further initial development
23
#include <sqlite3cc.h>
1 by edam
- initial commit
24
#include <boost/filesystem.hpp>
25
26
27
#define DBFILE "test.db"
28
29
30
int main()
31
{
32
	// delete any existing test database
33
	if( boost::filesystem::exists( DBFILE ) )
34
		boost::filesystem::remove( DBFILE );
35
36
	// open database
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
37
	sqlite::connection db( DBFILE );
38
39
	std::cout << "SQLite threading mode: " << sqlite::threadsafe() << "\n";
1 by edam
- initial commit
40
2 by edam
- further initial development
41
	// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
42
1 by edam
- initial commit
43
	// test basic statement, via command class
44
	sqlite::command c1( db,
45
		"CREATE TABLE pets ( "
46
			"name TEXT PRIMARY KEY, "
47
			"age INTEGER "
48
		")" );
49
	c1.step();
50
2 by edam
- further initial development
51
	// test basic_statement bind() calls, via command
1 by edam
- initial commit
52
	sqlite::command c2( db, "INSERT INTO pets VALUES( ?, ? )" );
53
	c2.bind( 1, "billy" ); c2.bind( 2, 12 ); c2.exec();
2 by edam
- further initial development
54
	sqlite::command c3( db, "INSERT INTO pets VALUES( ?6, ?9 )" );
55
	c3.bind( 9, 16 ); c3.bind( 6, "mopti" ); c3.exec();
1 by edam
- initial commit
56
	sqlite::command c6( db, "INSERT INTO pets VALUES( ?, ? )" );
57
	c6.bind_null( 1 ); c6.bind( 2, 123 ); c6.exec();
2 by edam
- further initial development
58
	sqlite::command c7( db, "INSERT INTO pets VALUES( :foo, :bar )" );
59
	c7.bind( ":foo", "foocat" ); c7.bind( ":bar", 22 ); c7.exec();
1 by edam
- initial commit
60
13 by edam
- made basic_statement::step() protected, for use by query and command only
61
	// test command binding via stream operator
62
	sqlite::command( db, "INSERT INTO pets VALUES( ?, ? )" ) <<
63
		"tessa" << 16 << sqlite::exec;
1 by edam
- initial commit
64
	sqlite::command c4( db, "INSERT INTO pets VALUES( ?, ? )" );
13 by edam
- made basic_statement::step() protected, for use by query and command only
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( db, "SELECT * FROM pets WHERE name = ? OR age = ?" )
71
		<< "foo" << 16;
72
	sqlite::query( db, "SELECT * FROM pets WHERE name = ? OR name = ? OR "
73
		"name = ? OR name = ?" ) << "foo" << sqlite::null << sqlite::null
74
		<< sqlite::set_index( 1 ) << "bar" << "baz";
1 by edam
- initial commit
75
2 by edam
- further initial development
76
	// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
77
78
	// test basic queries
79
	sqlite::query q1( db, "SELECT age, age + 3, name "
80
		"FROM pets WHERE NOT age IS NULL "
81
		"ORDER BY age ASC" );
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
82
83
	// test column count and names
2 by edam
- further initial development
84
	assert( q1.column_count() == 3 );
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
85
	assert( q1.column_name( 2 ) == "name" );
2 by edam
- further initial development
86
87
	// test rows
88
	unsigned int age;
89
	std::string name;
90
	q1.step().column( 0, age ); assert( age == 12 );
9 by edam
- added NEWS
91
	name = q1.step().column< std::string >( 2 ); assert( name == "mopti" );
2 by edam
- further initial development
92
93
	// test row stream operator
94
	sqlite::row r1 = q1.step();
95
	r1 >> age; assert( age == 16 );
96
	r1 >> age; assert( age == 19 );
97
	r1 >> name; assert( name == "tessa" );
98
	q1.step() >> age >> sqlite::null >> name;
99
	assert( age == 22 && name == "foocat" );
100
	sqlite::row row = q1.step();
101
	row >> age >> sqlite::null >> sqlite::null; assert( age == 123 );
102
	row >> sqlite::set_index( 1 ) >> age; assert( age == 126 );
103
104
	// test NULL value handling
105
	assert( row.column_type( 2 ) == SQLITE_NULL );
9 by edam
- added NEWS
106
	age = row.column< unsigned int >( 2 ); assert( age == 0 );
2 by edam
- further initial development
107
	row.column( 2, name ); assert( name == "" );
108
109
	// test end of results
110
	assert( !q1.step() );
111
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
112
	// test num results
113
	sqlite::query q3( db, "SELECT name FROM pets WHERE age = 16" );
114
	assert( q3.num_results() == 2 );
115
3 by edam
- rewrote transaction classes and added transaction_guard
116
	// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
117
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
118
	// test query iterator
9 by edam
- added NEWS
119
	sqlite::query q2( db, "SELECT * FROM pets ORDER BY age DESC" );
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
120
	sqlite::query::iterator i1 = q2.begin();
121
	assert( i1 != q2.end() );
122
123
	// test resetting iterators
124
	unsigned int count = 0;
125
	for( sqlite::query::iterator i = q2.begin(); i != q2.end(); i++ )
126
	{
127
		// test iterator dereferencing and row numbers
128
		assert( count++ == i->row_number() );
129
		if( i->row_number() == 3 )
9 by edam
- added NEWS
130
			assert( i->column< std::string >( 0 ) == "tessa" &&
131
				i->column< unsigned int >( 1 ) == 16 );
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
132
	}
133
134
//	for( sqlite::query::iterator i = q2.begin(); i != q2.end(); i++ ) {
9 by edam
- added NEWS
135
//		for( unsigned int a = 0; a < q2.column_count(); a++ )
136
//			std::cout << q2.column_name( a ) << "[" <<
137
//			i->column< std::string >( a ) << "] ";
138
//		std::cout << "\n";
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
139
//	}
9 by edam
- added NEWS
140
141
	// - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
142
3 by edam
- rewrote transaction classes and added transaction_guard
143
	// test transactions
16 by edam
- renamed database to connection to better identify what it is (would database_connection be better though?)
144
	{
145
		sqlite::transaction_guard< sqlite::deferred_tranaction > t1( db );
146
	}
147
	{
148
		sqlite::transaction_guard< sqlite::immediate_transaction > t1( db );
149
	}
150
	{
151
		sqlite::transaction_guard< sqlite::exclusive_transaction > t1( db );
152
	}
153
	{
154
		sqlite::transaction_guard< sqlite::recursive_transaction > t1( db );
155
		sqlite::transaction_guard< sqlite::recursive_transaction > t2( db );
156
	}
3 by edam
- rewrote transaction classes and added transaction_guard
157
158
	// test transaction guard rollback
159
	{
160
		sqlite::transaction_guard< > t1( db );
161
		db.exec( "UPDATE pets SET age = 99" );
162
	}
163
	sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
164
		.step() >> age; assert( age == 123 );
165
166
	// text recursive transactions
167
	{
168
		sqlite::transaction_guard< sqlite::recursive_transaction > t1( db );
9 by edam
- added NEWS
169
		db.exec( "UPDATE pets SET age = 66" );
3 by edam
- rewrote transaction classes and added transaction_guard
170
171
		{
172
			sqlite::transaction_guard< sqlite::recursive_transaction > t2( db );
173
			db.exec( "UPDATE pets SET age = 1" );
174
		}
175
176
		sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
9 by edam
- added NEWS
177
			.step() >> age; assert( age == 66 );
3 by edam
- rewrote transaction classes and added transaction_guard
178
	}
179
	sqlite::query( db, "SELECT age FROM pets ORDER BY age DESC LIMIT 1" )
9 by edam
- added NEWS
180
		.step() >> age; assert( age == 123 );
3 by edam
- rewrote transaction classes and added transaction_guard
181
4 by edam
- print "ok" when test program is successful!
182
	// ok
1 by edam
- initial commit
183
	return 0;
184
}