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