Merge 'trunk' into 'table_name_ref'
[dbsrgits/DBIx-Class.git] / t / 98savepoints.t
1 use strict;
2 use warnings;
3
4 use Test::More;
5 use lib qw(t/lib);
6 use DBICTest;
7 use DBICTest::Stats;
8
9 my ($create_sql, $dsn, $user, $pass);
10
11 if (exists $ENV{DBICTEST_PG_DSN}) {
12   ($dsn, $user, $pass) = @ENV{map { "DBICTEST_PG_${_}" } qw/DSN USER PASS/};
13
14   $create_sql = "CREATE TABLE artist (artistid serial PRIMARY KEY, name VARCHAR(100), rank INTEGER NOT NULL DEFAULT '13', charfield CHAR(10))";
15 } elsif (exists $ENV{DBICTEST_MYSQL_DSN}) {
16   ($dsn, $user, $pass) = @ENV{map { "DBICTEST_MYSQL_${_}" } qw/DSN USER PASS/};
17
18   $create_sql = "CREATE TABLE artist (artistid INTEGER NOT NULL AUTO_INCREMENT PRIMARY KEY, name VARCHAR(100), rank INTEGER NOT NULL DEFAULT '13', charfield CHAR(10)) ENGINE=InnoDB";
19 } else {
20   plan skip_all => 'Set DBICTEST_(PG|MYSQL)_DSN _USER and _PASS if you want to run savepoint tests';
21 }
22
23 plan tests => 16;
24
25 my $schema = DBICTest::Schema->connect ($dsn,$user,$pass,{ auto_savepoint => 1 });
26
27 my $stats = DBICTest::Stats->new;
28
29 $schema->storage->debugobj($stats);
30
31 $schema->storage->debug(1);
32
33 {
34     local $SIG{__WARN__} = sub {};
35     $schema->storage->dbh->do ('DROP TABLE IF EXISTS artist');
36     $schema->storage->dbh->do ($create_sql);
37 }
38
39 $schema->resultset('Artist')->create({ name => 'foo' });
40
41 $schema->txn_begin;
42
43 my $arty = $schema->resultset('Artist')->find(1);
44
45 my $name = $arty->name;
46
47 # First off, test a generated savepoint name
48 $schema->svp_begin;
49
50 cmp_ok($stats->{'SVP_BEGIN'}, '==', 1, 'Statistics svp_begin tickled');
51
52 $arty->update({ name => 'Jheephizzy' });
53
54 $arty->discard_changes;
55
56 cmp_ok($arty->name, 'eq', 'Jheephizzy', 'Name changed');
57
58 # Rollback the generated name
59 # Active: 0
60 $schema->svp_rollback;
61
62 cmp_ok($stats->{'SVP_ROLLBACK'}, '==', 1, 'Statistics svp_rollback tickled');
63
64 $arty->discard_changes;
65
66 cmp_ok($arty->name, 'eq', $name, 'Name rolled back');
67
68 $arty->update({ name => 'Jheephizzy'});
69
70 # Active: 0 1
71 $schema->svp_begin('testing1');
72
73 $arty->update({ name => 'yourmom' });
74
75 # Active: 0 1 2
76 $schema->svp_begin('testing2');
77
78 $arty->update({ name => 'gphat' });
79 $arty->discard_changes;
80 cmp_ok($arty->name, 'eq', 'gphat', 'name changed');
81 # Active: 0 1 2
82 # Rollback doesn't DESTROY the savepoint, it just rolls back to the value
83 # at it's conception
84 $schema->svp_rollback('testing2');
85 $arty->discard_changes;
86 cmp_ok($arty->name, 'eq', 'yourmom', 'testing2 reverted');
87
88 # Active: 0 1 2 3
89 $schema->svp_begin('testing3');
90 $arty->update({ name => 'coryg' });
91 # Active: 0 1 2 3 4
92 $schema->svp_begin('testing4');
93 $arty->update({ name => 'watson' });
94
95 # Release 3, which implicitly releases 4
96 # Active: 0 1 2
97 $schema->svp_release('testing3');
98 $arty->discard_changes;
99 cmp_ok($arty->name, 'eq', 'watson', 'release left data');
100 # This rolls back savepoint 2
101 # Active: 0 1 2
102 $schema->svp_rollback;
103 $arty->discard_changes;
104 cmp_ok($arty->name, 'eq', 'yourmom', 'rolled back to 2');
105
106 # Rollback the original savepoint, taking us back to the beginning, implicitly
107 # rolling back savepoint 1 and 2
108 $schema->svp_rollback('savepoint_0');
109 $arty->discard_changes;
110 cmp_ok($arty->name, 'eq', 'foo', 'rolled back to start');
111
112 $schema->txn_commit;
113
114 # And now to see if txn_do will behave correctly
115
116 $schema->txn_do (sub {
117     $schema->txn_do (sub {
118         $arty->name ('Muff');
119
120         $arty->update;
121       });
122
123     eval {
124       $schema->txn_do (sub {
125           $arty->name ('Moff');
126
127           $arty->update;
128
129           $arty->discard_changes;
130
131           is($arty->name,'Moff','Value updated in nested transaction');
132
133           $schema->storage->dbh->do ("GUARANTEED TO PHAIL");
134         });
135     };
136
137     ok ($@,'Nested transaction failed (good)');
138
139     $arty->discard_changes;
140
141     is($arty->name,'Muff','auto_savepoint rollback worked');
142
143     $arty->name ('Miff');
144
145     $arty->update;
146   });
147
148 $arty->discard_changes;
149
150 is($arty->name,'Miff','auto_savepoint worked');
151
152 cmp_ok($stats->{'SVP_BEGIN'},'==',7,'Correct number of savepoints created');
153
154 cmp_ok($stats->{'SVP_RELEASE'},'==',3,'Correct number of savepoints released');
155
156 cmp_ok($stats->{'SVP_ROLLBACK'},'==',5,'Correct number of savepoint rollbacks');
157
158 END { $schema->storage->dbh->do ("DROP TABLE artist") if defined $schema }
159