Merge 'trunk' into 'replication_dedux'
[dbsrgits/DBIx-Class.git] / t / 83cache.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 my $queries;
11 $schema->storage->debugcb( sub{ $queries++ } );
12
13 eval "use DBD::SQLite";
14 plan skip_all => 'needs DBD::SQLite for testing' if $@;
15 plan tests => 23;
16
17 my $rs = $schema->resultset("Artist")->search(
18   { artistid => 1 }
19 );
20
21 my $artist = $rs->first;
22
23 ok( !defined($rs->get_cache), 'cache is not populated without cache attribute' );
24
25 $rs = $schema->resultset('Artist')->search( undef, { cache => 1 } );
26 my $artists = [ $rs->all ];
27
28 is( scalar @{$rs->get_cache}, 3, 'all() populates cache for search with cache attribute' );
29
30 $rs->clear_cache;
31
32 ok( !defined($rs->get_cache), 'clear_cache is functional' );
33
34 $rs->next;
35
36 is( scalar @{$rs->get_cache}, 3, 'next() populates cache for search with cache attribute' );
37
38 pop( @$artists );
39 $rs->set_cache( $artists );
40
41 is( scalar @{$rs->get_cache}, 2, 'set_cache() is functional' );
42
43 my $cd = $schema->resultset('CD')->find(1);
44
45 $rs->clear_cache;
46
47 $queries = 0;
48 $schema->storage->debug(1);
49
50 $rs = $schema->resultset('Artist')->search( undef, { cache => 1 } );
51 while( $artist = $rs->next ) {}
52 $artist = $rs->first();
53
54 is( $queries, 1, 'revisiting a row does not issue a query when cache => 1' );
55
56 $schema->storage->debug(0);
57
58 my @a = $schema->resultset("Artist")->search(
59   { },
60   {
61     join => [ qw/ cds /],
62     prefetch => [qw/ cds /],
63   }
64 );
65
66 is(scalar @a, 3, 'artist with cds: count parent objects');
67
68 $rs = $schema->resultset("Artist")->search(
69   { 'artistid' => 1 },
70   {
71     join => [ qw/ cds /],
72     prefetch => [qw/ cds /],
73   }
74 );
75
76 use Data::Dumper; $Data::Dumper::Deparse = 1;
77
78 # start test for prefetch SELECT count
79 $queries = 0;
80 $schema->storage->debug(1);
81
82 $artist = $rs->first;
83 $rs->reset();
84
85 # make sure artist contains a related resultset for cds
86 is( ref $artist->{related_resultsets}->{cds}, 'DBIx::Class::ResultSet', 'artist has a related_resultset for cds' );
87
88 # check if $artist->cds->get_cache is populated
89 is( scalar @{$artist->cds->get_cache}, 3, 'cache for artist->cds contains correct number of records');
90
91 # ensure that $artist->cds returns correct number of objects
92 is( scalar ($artist->cds), 3, 'artist->cds returns correct number of objects' );
93
94 # ensure that $artist->cds->count returns correct value
95 is( $artist->cds->count, 3, 'artist->cds->count returns correct value' );
96
97 # ensure that $artist->count_related('cds') returns correct value
98 is( $artist->count_related('cds'), 3, 'artist->count_related returns correct value' );
99
100 is($queries, 1, 'only one SQL statement executed');
101
102 $schema->storage->debug(0);
103
104 # make sure related_resultset is deleted after object is updated
105 $artist->set_column('name', 'New Name');
106 $artist->update();
107
108 is( scalar keys %{$artist->{related_resultsets}}, 0, 'related resultsets deleted after update' );
109
110 # todo: make sure caching works with nested prefetch e.g. $artist->cds->tracks
111 $rs = $schema->resultset("Artist")->search(
112   { artistid => 1 },
113   {
114     join => { cds => 'tags' },
115     prefetch => {
116       cds => 'tags'
117     },
118   }
119 );
120 {
121 my $artist_count_before = $schema->resultset('Artist')->count;
122 $schema->resultset("Artist")->create({artistid=>4,name=>qq{Humoungous Hamsters}});
123 is($schema->resultset('Artist')->count, $artist_count_before + 1, 'count() reflects new artist');
124 my $artist = $schema->resultset("Artist")->search(
125   { artistid => 4 },{prefetch=>[qw/cds/]}
126 )->first;
127
128 is($artist->cds, 0, 'No cds for this artist');
129 }
130
131 # SELECT count for nested has_many prefetch
132 $queries = 0;
133 $schema->storage->debug(1);
134
135 $artist = ($rs->all)[0];
136
137 is($queries, 1, 'only one SQL statement executed');
138
139 $schema->storage->debug(0);
140
141 my @objs;
142 #$artist = $rs->find(1);
143
144 $queries = 0;
145 $schema->storage->debug(1);
146
147 my $cds = $artist->cds;
148 my $tags = $cds->next->tags;
149 while( my $tag = $tags->next ) {
150   push @objs, $tag->tagid; #warn "tag:", $tag->ID, " => ", $tag->tag;
151 }
152
153 is_deeply( \@objs, [ 3 ], 'first cd has correct tags' );
154
155 $tags = $cds->next->tags;
156 @objs = ();
157 while( my $tag = $tags->next ) {
158   push @objs, $tag->id; #warn "tag: ", $tag->ID;
159 }
160
161 is_deeply( \@objs, [ 1 ], 'second cd has correct tags' );
162
163 $tags = $cds->next->tags;
164 @objs = ();
165 while( my $tag = $tags->next ) {
166   push @objs, $tag->id; #warn "tag: ", $tag->ID;
167 }
168
169 is_deeply( \@objs, [ 2, 5, 8 ], 'third cd has correct tags' );
170
171 is( $queries, 0, 'no additional SQL statements while checking nested data' );
172
173 # start test for prefetch SELECT count
174 $queries = 0;
175
176 $artist = $schema->resultset('Artist')->find(1, { prefetch => [qw/cds/] });
177
178 is( $queries, 1, 'only one select statement on find with inline has_many prefetch' );
179
180 # start test for prefetch SELECT count
181 $queries = 0;
182
183 $rs = $schema->resultset('Artist')->search(undef, { prefetch => [qw/cds/] });
184 $artist = $rs->find(1);
185
186 is( $queries, 1, 'only one select statement on find with has_many prefetch on resultset' );
187
188 $schema->storage->debug(0);
189