fixed up sqlt tests a bit and tried fixing 2 failing sqlt tests
[dbsrgits/DBIx-Class-Historic.git] / t / 81transactions.t
1 use strict;
2 use warnings;  
3
4 use Test::More;
5 use lib qw(t/lib);
6 use DBICTest;
7
8 my $schema = DBICTest->init_schema();
9
10 plan tests => 39;
11
12 my $code = sub {
13   my ($artist, @cd_titles) = @_;
14   
15   $artist->create_related('cds', {
16     title => $_,
17     year => 2006,
18   }) foreach (@cd_titles);
19   
20   return $artist->cds->all;
21 };
22
23 # Test checking of parameters
24 {
25   eval {
26     (ref $schema)->txn_do(sub{});
27   };
28   like($@, qr/class method/, '$self parameter check ok');
29   eval {
30     $schema->txn_do('');
31   };
32   like($@, qr/must be a CODE reference/, '$coderef parameter check ok');
33 }
34
35 # Test successful txn_do() - scalar context
36 {
37   my @titles = map {'txn_do test CD ' . $_} (1..5);
38   my $artist = $schema->resultset('Artist')->find(1);
39   my $count_before = $artist->cds->count;
40   my $count_after = $schema->txn_do($code, $artist, @titles);
41   is($count_after, $count_before+5, 'successful txn added 5 cds');
42   is($artist->cds({
43     title => "txn_do test CD $_",
44   })->first->year, 2006, "new CD $_ year correct") for (1..5);
45 }
46
47 # Test successful txn_do() - list context
48 {
49   my @titles = map {'txn_do test CD ' . $_} (6..10);
50   my $artist = $schema->resultset('Artist')->find(1);
51   my $count_before = $artist->cds->count;
52   my @cds = $schema->txn_do($code, $artist, @titles);
53   is(scalar @cds, $count_before+5, 'added 5 CDs and returned in list context');
54   is($artist->cds({
55     title => "txn_do test CD $_",
56   })->first->year, 2006, "new CD $_ year correct") for (6..10);
57 }
58
59 # Test nested successful txn_do()
60 {
61   my $nested_code = sub {
62     my ($schema, $artist, $code) = @_;
63
64     my @titles1 = map {'nested txn_do test CD ' . $_} (1..5);
65     my @titles2 = map {'nested txn_do test CD ' . $_} (6..10);
66
67     $schema->txn_do($code, $artist, @titles1);
68     $schema->txn_do($code, $artist, @titles2);
69   };
70
71   my $artist = $schema->resultset('Artist')->find(2);
72   my $count_before = $artist->cds->count;
73
74   eval {
75     $schema->txn_do($nested_code, $schema, $artist, $code);
76   };
77
78   my $error = $@;
79
80   ok(!$error, 'nested txn_do succeeded');
81   is($artist->cds({
82     title => 'nested txn_do test CD '.$_,
83   })->first->year, 2006, qq{nested txn_do CD$_ year ok}) for (1..10);
84   is($artist->cds->count, $count_before+10, 'nested txn_do added all CDs');
85 }
86
87 my $fail_code = sub {
88   my ($artist) = @_;
89   $artist->create_related('cds', {
90     title => 'this should not exist',
91     year => 2005,
92   });
93   die "the sky is falling";
94 };
95
96 # Test failed txn_do()
97 {
98   my $artist = $schema->resultset('Artist')->find(3);
99
100   eval {
101     $schema->txn_do($fail_code, $artist);
102   };
103
104   my $error = $@;
105
106   like($error, qr/the sky is falling/, 'failed txn_do threw an exception');
107   my $cd = $artist->cds({
108     title => 'this should not exist',
109     year => 2005,
110   })->first;
111   ok(!defined($cd), q{failed txn_do didn't change the cds table});
112 }
113
114 # Test failed txn_do() with failed rollback
115 {
116   my $artist = $schema->resultset('Artist')->find(3);
117
118   # Force txn_rollback() to throw an exception
119   no warnings 'redefine';
120   no strict 'refs';
121   local *{"DBIx::Class::Schema::txn_rollback"} = sub{die 'FAILED'};
122
123   eval {
124     $schema->txn_do($fail_code, $artist);
125   };
126
127   my $error = $@;
128
129   like($error, qr/Rollback failed/, 'failed txn_do with a failed '.
130        'txn_rollback threw a rollback exception');
131   like($error, qr/the sky is falling/, 'failed txn_do with a failed '.
132        'txn_rollback included the original exception');
133
134   my $cd = $artist->cds({
135     title => 'this should not exist',
136     year => 2005,
137   })->first;
138   isa_ok($cd, 'DBICTest::CD', q{failed txn_do with a failed txn_rollback }.
139          q{changed the cds table});
140   $cd->delete; # Rollback failed
141   $cd = $artist->cds({
142     title => 'this should not exist',
143     year => 2005,
144   })->first;
145   ok(!defined($cd), q{deleted the failed txn's cd});
146   $schema->storage->{transaction_depth} = 0; # Must reset this or further tests
147                                              # will fail
148 }
149
150 # Test nested failed txn_do()
151 {
152   my $nested_fail_code = sub {
153     my ($schema, $artist, $code1, $code2) = @_;
154
155     my @titles = map {'nested txn_do test CD ' . $_} (1..5);
156
157     $schema->txn_do($code1, $artist, @titles); # successful txn
158     $schema->txn_do($code2, $artist);          # failed txn
159   };
160
161   my $artist = $schema->resultset('Artist')->find(3);
162
163   eval {
164     $schema->txn_do($nested_fail_code, $schema, $artist, $code, $fail_code);
165   };
166
167   my $error = $@;
168
169   like($error, qr/the sky is falling/, 'nested failed txn_do threw exception');
170   ok(!defined($artist->cds({
171     title => 'nested txn_do test CD '.$_,
172     year => 2006,
173   })->first), qq{failed txn_do didn't add first txn's cd $_}) for (1..5);
174   my $cd = $artist->cds({
175     title => 'this should not exist',
176     year => 2005,
177   })->first;
178   ok(!defined($cd), q{failed txn_do didn't add failed txn's cd});
179 }
180