Fix RT#58554 (make sure FOR X is always the last part of a select)
[dbsrgits/DBIx-Class.git] / t / 71mysql.t
1 use strict;
2 use warnings;  
3
4 use Test::More;
5 use Test::Exception;
6 use lib qw(t/lib);
7 use DBICTest;
8 use DBI::Const::GetInfoType;
9 use DBIC::SqlMakerTest;
10
11 my ($dsn, $user, $pass) = @ENV{map { "DBICTEST_MYSQL_${_}" } qw/DSN USER PASS/};
12
13 #warn "$dsn $user $pass";
14
15 plan skip_all => 'Set $ENV{DBICTEST_MYSQL_DSN}, _USER and _PASS to run this test'
16   unless ($dsn && $user);
17
18 my $schema = DBICTest::Schema->connect($dsn, $user, $pass);
19
20 my $dbh = $schema->storage->dbh;
21
22 $dbh->do("DROP TABLE IF EXISTS artist;");
23
24 $dbh->do("CREATE TABLE artist (artistid INTEGER NOT NULL AUTO_INCREMENT PRIMARY KEY, name VARCHAR(100), rank INTEGER NOT NULL DEFAULT '13', charfield CHAR(10));");
25
26 $dbh->do("DROP TABLE IF EXISTS cd;");
27
28 $dbh->do("CREATE TABLE cd (cdid INTEGER NOT NULL AUTO_INCREMENT PRIMARY KEY, artist INTEGER, title TEXT, year DATE, genreid INTEGER, single_track INTEGER);");
29
30 $dbh->do("DROP TABLE IF EXISTS producer;");
31
32 $dbh->do("CREATE TABLE producer (producerid INTEGER NOT NULL AUTO_INCREMENT PRIMARY KEY, name TEXT);");
33
34 $dbh->do("DROP TABLE IF EXISTS cd_to_producer;");
35
36 $dbh->do("CREATE TABLE cd_to_producer (cd INTEGER,producer INTEGER);");
37
38 $dbh->do("DROP TABLE IF EXISTS owners;");
39
40 $dbh->do("CREATE TABLE owners (id INTEGER NOT NULL AUTO_INCREMENT PRIMARY KEY, name VARCHAR(100) NOT NULL);");
41
42 $dbh->do("DROP TABLE IF EXISTS books;");
43
44 $dbh->do("CREATE TABLE books (id INTEGER NOT NULL AUTO_INCREMENT PRIMARY KEY, source VARCHAR(100) NOT NULL, owner integer NOT NULL, title varchar(100) NOT NULL,  price integer);");
45
46 #'dbi:mysql:host=localhost;database=dbic_test', 'dbic_test', '');
47
48 # make sure sqlt_type overrides work (::Storage::DBI::mysql does this) 
49 {
50   my $schema = DBICTest::Schema->connect($dsn, $user, $pass);
51
52   ok (!$schema->storage->_dbh, 'definitely not connected');
53   is ($schema->storage->sqlt_type, 'MySQL', 'sqlt_type correct pre-connection');
54 }
55
56 # This is in Core now, but it's here just to test that it doesn't break
57 $schema->class('Artist')->load_components('PK::Auto');
58
59 # test primary key handling
60 my $new = $schema->resultset('Artist')->create({ name => 'foo' });
61 ok($new->artistid, "Auto-PK worked");
62
63 # test LIMIT support
64 for (1..6) {
65     $schema->resultset('Artist')->create({ name => 'Artist ' . $_ });
66 }
67 my $it = $schema->resultset('Artist')->search( {},
68     { rows => 3,
69       offset => 2,
70       order_by => 'artistid' }
71 );
72 is( $it->count, 3, "LIMIT count ok" );  # ask for 3 rows out of 7 artists
73 is( $it->next->name, "Artist 2", "iterator->next ok" );
74 $it->next;
75 $it->next;
76 is( $it->next, undef, "next past end of resultset ok" );
77
78 # Limit with select-lock
79 lives_ok {
80   $schema->txn_do (sub {
81     isa_ok (
82       $schema->resultset('Artist')->find({artistid => 1}, {for => 'update', rows => 1}),
83       'DBICTest::Schema::Artist',
84     );
85   });
86 } 'Limited FOR UPDATE select works';
87
88 my $test_type_info = {
89     'artistid' => {
90         'data_type' => 'INT',
91         'is_nullable' => 0,
92         'size' => 11,
93         'default_value' => undef,
94     },
95     'name' => {
96         'data_type' => 'VARCHAR',
97         'is_nullable' => 1,
98         'size' => 100,
99         'default_value' => undef,
100     },
101     'rank' => {
102         'data_type' => 'INT',
103         'is_nullable' => 0,
104         'size' => 11,
105         'default_value' => 13,
106     },
107     'charfield' => {
108         'data_type' => 'CHAR',
109         'is_nullable' => 1,
110         'size' => 10,
111         'default_value' => undef,
112     },
113 };
114
115 $schema->populate ('Owners', [
116   [qw/id  name  /],
117   [qw/1   wiggle/],
118   [qw/2   woggle/],
119   [qw/3   boggle/],
120 ]);
121
122 $schema->populate ('BooksInLibrary', [
123   [qw/source  owner title   /],
124   [qw/Library 1     secrets1/],
125   [qw/Eatery  1     secrets2/],
126   [qw/Library 2     secrets3/],
127 ]);
128
129 #
130 # try a distinct + prefetch on tables with identically named columns 
131 # (mysql doesn't seem to like subqueries with equally named columns)
132 #
133
134 {
135   # try a ->has_many direction (due to a 'multi' accessor the select/group_by group is collapsed)
136   my $owners = $schema->resultset ('Owners')->search (
137     { 'books.id' => { '!=', undef }},
138     { prefetch => 'books', distinct => 1 }
139   );
140   my $owners2 = $schema->resultset ('Owners')->search ({ id => { -in => $owners->get_column ('me.id')->as_query }});
141   for ($owners, $owners2) {
142     is ($_->all, 2, 'Prefetched grouped search returns correct number of rows');
143     is ($_->count, 2, 'Prefetched grouped search returns correct count');
144   }
145
146   # try a ->belongs_to direction (no select collapse)
147   my $books = $schema->resultset ('BooksInLibrary')->search (
148     { 'owner.name' => 'wiggle' },
149     { prefetch => 'owner', distinct => 1 }
150   );
151   my $books2 = $schema->resultset ('BooksInLibrary')->search ({ id => { -in => $books->get_column ('me.id')->as_query }});
152   for ($books, $books2) {
153     is ($_->all, 1, 'Prefetched grouped search returns correct number of rows');
154     is ($_->count, 1, 'Prefetched grouped search returns correct count');
155   }
156 }
157
158 SKIP: {
159     my $mysql_version = $dbh->get_info( $GetInfoType{SQL_DBMS_VER} );
160     skip "Cannot determine MySQL server version", 1 if !$mysql_version;
161
162     my ($v1, $v2, $v3) = $mysql_version =~ /^(\d+)\.(\d+)(?:\.(\d+))?/;
163     skip "Cannot determine MySQL server version", 1 if !$v1 || !defined($v2);
164
165     $v3 ||= 0;
166
167     if( ($v1 < 5) || ($v1 == 5 && $v2 == 0 && $v3 <= 3) ) {
168         $test_type_info->{charfield}->{data_type} = 'VARCHAR';
169     }
170
171     my $type_info = $schema->storage->columns_info_for('artist');
172     is_deeply($type_info, $test_type_info, 'columns_info_for - column data types');
173 }
174
175 my $cd = $schema->resultset ('CD')->create ({});
176 my $producer = $schema->resultset ('Producer')->create ({});
177 lives_ok { $cd->set_producers ([ $producer ]) } 'set_relationship doesnt die';
178
179 {
180   my $artist = $schema->resultset('Artist')->next;
181   my $cd = $schema->resultset('CD')->next;
182   $cd->set_from_related ('artist', $artist);
183   $cd->update;
184
185   my $rs = $schema->resultset('CD')->search ({}, { prefetch => 'artist' });
186
187   lives_ok sub {
188     my $cd = $rs->next;
189     is ($cd->artist->name, $artist->name, 'Prefetched artist');
190   }, 'join does not throw (mysql 3 test)';
191
192   # induce a jointype override, make sure it works even if we don't have mysql3
193   local $schema->storage->sql_maker->{_default_jointype} = 'inner';
194   is_same_sql_bind (
195     $rs->as_query,
196     '(
197       SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track,
198              artist.artistid, artist.name, artist.rank, artist.charfield
199         FROM cd me
200         INNER JOIN artist artist ON artist.artistid = me.artist
201     )',
202     [],
203     'overriden default join type works',
204   );
205 }
206
207 {
208   # Test support for straight joins
209   my $cdsrc = $schema->source('CD');
210   my $artrel_info = $cdsrc->relationship_info ('artist');
211   $cdsrc->add_relationship(
212     'straight_artist',
213     $artrel_info->{class},
214     $artrel_info->{cond},
215     { %{$artrel_info->{attrs}}, join_type => 'straight' },
216   );
217   is_same_sql_bind (
218     $cdsrc->resultset->search({}, { prefetch => 'straight_artist' })->as_query,
219     '(
220       SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track,
221              straight_artist.artistid, straight_artist.name, straight_artist.rank, straight_artist.charfield
222         FROM cd me
223         STRAIGHT_JOIN artist straight_artist ON straight_artist.artistid = me.artist
224     )',
225     [],
226     'straight joins correctly supported for mysql'
227   );
228 }
229
230 ## Can we properly deal with the null search problem?
231 ##
232 ## Only way is to do a SET SQL_AUTO_IS_NULL = 0; on connect
233 ## But I'm not sure if we should do this or not (Ash, 2008/06/03)
234 #
235 # There is now a built-in function to do this, test that everything works
236 # with it (ribasushi, 2009/07/03)
237
238 NULLINSEARCH: {
239     my $ansi_schema = DBICTest::Schema->connect ($dsn, $user, $pass, { on_connect_call => 'set_strict_mode' });
240
241     $ansi_schema->resultset('Artist')->create ({ name => 'last created artist' });
242
243     ok my $artist1_rs = $ansi_schema->resultset('Artist')->search({artistid=>6666})
244       => 'Created an artist resultset of 6666';
245
246     is $artist1_rs->count, 0
247       => 'Got no returned rows';
248
249     ok my $artist2_rs = $ansi_schema->resultset('Artist')->search({artistid=>undef})
250       => 'Created an artist resultset of undef';
251
252     is $artist2_rs->count, 0
253       => 'got no rows';
254
255     my $artist = $artist2_rs->single;
256
257     is $artist => undef
258       => 'Nothing Found!';
259 }
260
261 # check for proper grouped counts
262 {
263   my $ansi_schema = DBICTest::Schema->connect ($dsn, $user, $pass, {
264     on_connect_call => 'set_strict_mode',
265     quote_char => '`',
266     name_sep => '.'
267   });
268   my $rs = $ansi_schema->resultset('CD');
269
270   my $years;
271   $years->{$_->year|| scalar keys %$years}++ for $rs->all;  # NULL != NULL, thus the keys eval
272
273   lives_ok ( sub {
274     is (
275       $rs->search ({}, { group_by => 'year'})->count,
276       scalar keys %$years,
277       'grouped count correct',
278     );
279   }, 'Grouped count does not throw');
280
281   lives_ok( sub {
282     $ansi_schema->resultset('Owners')->search({}, {
283       join => 'books', group_by => [ 'me.id', 'books.id' ]
284     })->count();
285   }, 'count on grouped columns with the same name does not throw');
286
287
288 }
289
290 ZEROINSEARCH: {
291   my $cds_per_year = {
292     2001 => 2,
293     2002 => 1,
294     2005 => 3,
295   };
296
297   my $rs = $schema->resultset ('CD');
298   $rs->delete;
299   for my $y (keys %$cds_per_year) {
300     for my $c (1 .. $cds_per_year->{$y} ) {
301       $rs->create ({ title => "CD $y-$c", artist => 1, year => "$y-01-01" });
302     }
303   }
304
305   is ($rs->count, 6, 'CDs created successfully');
306
307   $rs = $rs->search ({}, {
308     select => [ \ 'YEAR(year)' ], as => ['y'], distinct => 1,
309   });
310
311   is_deeply (
312     [ sort ($rs->get_column ('y')->all) ],
313     [ sort keys %$cds_per_year ],
314     'Years group successfully',
315   );
316
317   $rs->create ({ artist => 1, year => '0-1-1', title => 'Jesus Rap' });
318
319   is_deeply (
320     [ sort $rs->get_column ('y')->all ],
321     [ 0, sort keys %$cds_per_year ],
322     'Zero-year groups successfully',
323   );
324
325   # convoluted search taken verbatim from list 
326   my $restrict_rs = $rs->search({ -and => [
327     year => { '!=', 0 },
328     year => { '!=', undef }
329   ]});
330
331   is_deeply (
332     [ $restrict_rs->get_column('y')->all ],
333     [ $rs->get_column ('y')->all ],
334     'Zero year was correctly excluded from resultset',
335   );
336 }
337
338 ## If find() is the first query after connect()
339 ## DBI::Storage::sql_maker() will be called before
340 ## _determine_driver() and so the ::SQLHacks class for MySQL
341 ## will not be used
342
343 my $schema2 = DBICTest::Schema->connect($dsn, $user, $pass);
344 $schema2->resultset("Artist")->find(4);
345 isa_ok($schema2->storage->sql_maker, 'DBIx::Class::SQLMaker::MySQL');
346
347 done_testing;