11 my $db_orig = DBICTest->_sqlite_dbfilename;
12 my $db_tmp = "$db_orig.tmp";
14 # Set up the "usual" sqlite for DBICTest
15 my $schema = DBICTest->init_schema( sqlite_use_file => 1 );
17 # Make sure we're connected by doing something
18 my @art = $schema->resultset("Artist")->search({ }, { order_by => 'name DESC'});
19 cmp_ok(@art, '==', 3, "Three artists returned");
21 # Disconnect the dbh, and be sneaky about it
22 # Also test if DBD::SQLite finaly knows how to ->disconnect properly
25 local $SIG{__WARN__} = sub { $w = shift };
26 $schema->storage->_dbh->disconnect;
27 ok ($w !~ /active statement handles/, 'SQLite can disconnect properly');
30 # Try the operation again - What should happen here is:
31 # 1. S::DBI blindly attempts the SELECT, which throws an exception
32 # 2. It catches the exception, checks ->{Active}/->ping, sees the disconnected state...
33 # 3. Reconnects, and retries the operation
35 my @art_two = $schema->resultset("Artist")->search({ }, { order_by => 'name DESC'});
36 cmp_ok(@art_two, '==', 3, "Three artists returned");
38 ### Now, disconnect the dbh, and move the db file;
39 # create a new one and chmod 000 to prevent SQLite from connecting.
40 $schema->storage->_dbh->disconnect;
41 move( $db_orig, $db_tmp )
42 or die "failed to move $db_orig to $db_tmp: $!";
43 open DBFILE, '>', $db_orig;
44 print DBFILE 'THIS IS NOT A REAL DATABASE';
48 ### Try the operation again... it should fail, since there's no db
50 # Catch the DBI connection error
51 local $SIG{__WARN__} = sub {};
53 my @art_three = $schema->resultset("Artist")->search( {}, { order_by => 'name DESC' } );
54 } 'The operation failed';
57 # otherwise win32 can't unlink the fake db file
58 # as root sqlite will have the file open and won't reconnect
59 $schema->storage->_dbh->disconnect if $^O eq 'MSWin32' or $> == 0;
61 ### Now, move the db file back to the correct name
62 unlink($db_orig) or die "could not delete $db_orig: $!";
63 move( $db_tmp, $db_orig )
64 or die "could not move $db_tmp to $db_orig: $!";
66 ### Try the operation again... this time, it should succeed
69 @art_four = $schema->resultset("Artist")->search( {}, { order_by => 'name DESC' } );
70 } 'The operation succeeded';
71 cmp_ok( @art_four, '==', 3, "Three artists returned" );
73 # check that reconnection contexts are preserved in txn_do / dbh_do
79 invoke => sub { shift->(); 1 },
83 invoke => sub { my $foo = shift->() },
87 invoke => sub { my @foo = shift->() },
92 for my $ctx (keys %$ctx_map) {
94 # start disconnected and then connected
95 $schema->storage->disconnect;
99 $ctx_map->{$ctx}{invoke}->(sub { $schema->txn_do(sub {
100 is_deeply (\@_, $args, 'Args propagated correctly' );
102 is (wantarray(), $ctx_map->{$ctx}{wa}, "Correct $ctx context");
104 # this will cause a retry
105 $schema->storage->_dbh->disconnect unless $disarmed++;
107 isa_ok ($schema->resultset('Artist')->next, 'DBICTest::Artist');