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