Added support for subqueries in the select and +select sections
[dbsrgits/DBIx-Class.git] / t / search / subquery.t
1 #!/usr/bin/perl
2
3 use strict;
4 use warnings FATAL => 'all';
5
6 use Data::Dumper;
7
8 use Test::More;
9
10 BEGIN {
11     eval "use SQL::Abstract 1.49";
12     plan $@
13         ? ( skip_all => "Needs SQLA 1.49+" )
14         : ( tests => 7 );
15 }
16
17 use lib qw(t/lib);
18 use DBICTest;
19 use DBIC::SqlMakerTest;
20
21 my $schema = DBICTest->init_schema();
22 my $art_rs = $schema->resultset('Artist');
23 my $cdrs = $schema->resultset('CD');
24
25 {
26   my $cdrs2 = $cdrs->search({
27     artist_id => { 'in' => $art_rs->search({}, { rows => 1 })->get_column( 'id' )->as_query },
28   });
29
30   my $arr = $cdrs2->as_query;
31   my ($query, @bind) = @{$$arr};
32   is_same_sql_bind(
33     $query, \@bind,
34     "SELECT me.cdid,me.artist,me.title,me.year,me.genreid,me.single_track FROM cd me WHERE artist_id IN ( SELECT id FROM artist me LIMIT 1 )",
35     [],
36   );
37 }
38
39 {
40   my $rs = $art_rs->search(
41     {},
42     {
43       'select' => [
44         $cdrs->search({}, { rows => 1 })->get_column('id')->as_query,
45       ],
46     },
47   );
48
49   my $arr = $rs->as_query;
50   my ($query, @bind) = @{$$arr};
51   is_same_sql_bind(
52     $query, \@bind,
53     "SELECT (SELECT id FROM cd me LIMIT 1) FROM artist me",
54     [],
55   );
56 }
57
58 {
59   my $rs = $art_rs->search(
60     {},
61     {
62       '+select' => [
63         $cdrs->search({}, { rows => 1 })->get_column('id')->as_query,
64       ],
65     },
66   );
67
68   my $arr = $rs->as_query;
69   my ($query, @bind) = @{$$arr};
70   is_same_sql_bind(
71     $query, \@bind,
72     "SELECT me.artistid, me.name, me.rank, me.charfield, (SELECT id FROM cd me LIMIT 1) FROM artist me",
73     [],
74   );
75 }
76
77 # simple from
78 {
79   my $rs = $cdrs->search(
80     {},
81     {
82       alias => 'cd2',
83       from => [
84         { cd2 => $cdrs->search({ id => { '>' => 20 } })->as_query },
85       ],
86     },
87   );
88
89   my $arr = $rs->as_query;
90   my ($query, @bind) = @{$$arr};
91   is_same_sql_bind(
92     $query, \@bind,
93     "SELECT cd2.cdid, cd2.artist, cd2.title, cd2.year, cd2.genreid, cd2.single_track FROM (SELECT me.cdid,me.artist,me.title,me.year,me.genreid,me.single_track FROM cd me WHERE id > 20) cd2",
94     [],
95   );
96 }
97
98 # nested from
99 {
100   my $art_rs2 = $schema->resultset('Artist')->search({}, 
101   {
102     from => [ { 'me' => 'artist' }, 
103       [ { 'cds' => $cdrs->search({},{ 'select' => [\'me.artist as cds_artist' ]})->as_query },
104       { 'me.artistid' => 'cds_artist' } ] ]
105   });
106
107   my $arr = $art_rs2->as_query;
108   my ($query, @bind) = @{$$arr};
109   is_same_sql_bind(
110     $query, \@bind,
111     "SELECT me.artistid, me.name, me.rank, me.charfield FROM artist me JOIN (SELECT me.artist as cds_artist FROM cd me) cds ON me.artistid = cds_artist", []
112   );
113
114
115 }
116
117 # nested subquery in from
118 {
119   my $rs = $cdrs->search(
120     {},
121     {
122       alias => 'cd2',
123       from => [
124         { cd2 => $cdrs->search(
125             { id => { '>' => 20 } }, 
126             { 
127                 alias => 'cd3',
128                 from => [ 
129                 { cd3 => $cdrs->search( { id => { '<' => 40 } } )->as_query }
130                 ],
131             }, )->as_query },
132       ],
133     },
134   );
135
136   my $arr = $rs->as_query;
137   my ($query, @bind) = @{$$arr};
138   is_same_sql_bind(
139     $query, \@bind,
140     "SELECT cd2.cdid, cd2.artist, cd2.title, cd2.year, cd2.genreid, cd2.single_track 
141       FROM 
142         (SELECT cd3.cdid,cd3.artist,cd3.title,cd3.year,cd3.genreid,cd3.single_track 
143           FROM 
144             (SELECT me.cdid,me.artist,me.title,me.year,me.genreid,me.single_track 
145               FROM cd me WHERE id < 40) cd3
146           WHERE id > 20) cd2",
147     [],
148   );
149
150 }
151
152 {
153   my $rs = $cdrs->search({
154     year => {
155       '=' => $cdrs->search(
156         { artistid => { '=' => \'me.artistid' } },
157         { alias => 'inner' }
158       )->get_column('year')->max_rs->as_query,
159     },
160   });
161   my $arr = $rs->as_query;
162   my ($query, @bind) = @{$$arr};
163   is_same_sql_bind(
164     $query, \@bind,
165     "SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track FROM cd me WHERE year = (SELECT MAX(inner.year) FROM cd inner WHERE artistid = me.artistid)",
166     [],
167   );
168 }
169
170 __END__