Fixed the bind value column names in the SQL literal + bind examples.
[dbsrgits/DBIx-Class.git] / t / count / distinct.t
1 use strict;
2 use warnings;  
3
4 use Test::More;
5 use Test::Exception;
6
7 use lib qw(t/lib);
8
9 use DBICTest;
10 use DBIC::SqlMakerTest;
11
12 my $schema = DBICTest->init_schema();
13
14 # The tag Blue is assigned to cds 1 2 3 and 5
15 # The tag Cheesy is assigned to cds 2 4 and 5
16 #
17 # This combination should make some interesting group_by's
18 #
19 my $rs;
20 my $in_rs = $schema->resultset('Tag')->search({ tag => [ 'Blue', 'Cheesy' ] });
21
22 for my $get_count (
23   sub { shift->count },
24   sub { my $crs = shift->count_rs; isa_ok ($crs, 'DBIx::Class::ResultSetColumn'); $crs->next }
25 ) {
26   $rs = $schema->resultset('Tag')->search({ tag => 'Blue' });
27   is($get_count->($rs), 4, 'Count without DISTINCT');
28
29   $rs = $schema->resultset('Tag')->search({ tag => [ 'Blue', 'Cheesy' ] }, { group_by => 'tag' });
30   is($get_count->($rs), 2, 'Count with single column group_by');
31
32   $rs = $schema->resultset('Tag')->search({ tag => [ 'Blue', 'Cheesy' ] }, { group_by => 'cd' });
33   is($get_count->($rs), 5, 'Count with another single column group_by');
34
35   $rs = $schema->resultset('Tag')->search({ tag => 'Blue' }, { group_by => [ qw/tag cd/ ]});
36   is($get_count->($rs), 4, 'Count with multiple column group_by');
37
38   $rs = $schema->resultset('Tag')->search({ tag => 'Blue' }, { distinct => 1 });
39   is($get_count->($rs), 4, 'Count with single column distinct');
40
41   $rs = $schema->resultset('Tag')->search({ tag => { -in => $in_rs->get_column('tag')->as_query } });
42   is($get_count->($rs), 7, 'Count with IN subquery');
43
44   $rs = $schema->resultset('Tag')->search({ tag => { -in => $in_rs->get_column('tag')->as_query } }, { group_by => 'tag' });
45   is($get_count->($rs), 2, 'Count with IN subquery with outside group_by');
46
47   $rs = $schema->resultset('Tag')->search({ tag => { -in => $in_rs->get_column('tag')->as_query } }, { distinct => 1 });
48   is($get_count->($rs), 7, 'Count with IN subquery with outside distinct');
49
50   $rs = $schema->resultset('Tag')->search({ tag => { -in => $in_rs->get_column('tag')->as_query } }, { distinct => 1, select => 'tag' }), 
51   is($get_count->($rs), 2, 'Count with IN subquery with outside distinct on a single column');
52
53   $rs = $schema->resultset('Tag')->search({ tag => { -in => $in_rs->search({}, { group_by => 'tag' })->get_column('tag')->as_query } });
54   is($get_count->($rs), 7, 'Count with IN subquery with single group_by');
55
56   $rs = $schema->resultset('Tag')->search({ tag => { -in => $in_rs->search({}, { group_by => 'cd' })->get_column('tag')->as_query } });
57   is($get_count->($rs), 7, 'Count with IN subquery with another single group_by');
58
59   $rs = $schema->resultset('Tag')->search({ tag => { -in => $in_rs->search({}, { group_by => [ qw/tag cd/ ] })->get_column('tag')->as_query } });
60   is($get_count->($rs), 7, 'Count with IN subquery with multiple group_by');
61
62   $rs = $schema->resultset('Tag')->search({ tag => \"= 'Blue'" });
63   is($get_count->($rs), 4, 'Count without DISTINCT, using literal SQL');
64
65   $rs = $schema->resultset('Tag')->search({ tag => \" IN ('Blue', 'Cheesy')" }, { group_by => 'tag' });
66   is($get_count->($rs), 2, 'Count with literal SQL and single group_by');
67
68   $rs = $schema->resultset('Tag')->search({ tag => \" IN ('Blue', 'Cheesy')" }, { group_by => 'cd' });
69   is($get_count->($rs), 5, 'Count with literal SQL and another single group_by');
70
71   $rs = $schema->resultset('Tag')->search({ tag => \" IN ('Blue', 'Cheesy')" }, { group_by => [ qw/tag cd/ ] });
72   is($get_count->($rs), 7, 'Count with literal SQL and multiple group_by');
73
74   $rs = $schema->resultset('Tag')->search({ tag => 'Blue' }, { '+select' => { max => 'tagid' }, distinct => 1 });
75   is($get_count->($rs), 4, 'Count with +select aggreggate');
76
77   $rs = $schema->resultset('Tag')->search({}, { select => 'length(me.tag)', distinct => 1 });
78   is($get_count->($rs), 3, 'Count by distinct function result as select literal');
79 }
80
81 throws_ok(
82   sub { my $row = $schema->resultset('Tag')->search({}, { select => { distinct => [qw/tag cd/] } })->first },
83   qr/select => { distinct => \.\.\. } syntax is not supported for multiple columns/,
84   'throw on unsupported syntax'
85 );
86
87 # make sure distinct+func works
88 {
89   my $rs = $schema->resultset('Artist')->search(
90     {},
91     {
92       join => 'cds',
93       distinct => 1,
94       '+select' => [ { count => 'cds.cdid', -as => 'amount_of_cds' } ],
95       '+as' => [qw/num_cds/],
96       order_by => { -desc => 'amount_of_cds' },
97     }
98   );
99
100   is_same_sql_bind (
101     $rs->as_query,
102     '(
103       SELECT me.artistid, me.name, me.rank, me.charfield, COUNT( cds.cdid ) AS amount_of_cds
104         FROM artist me LEFT JOIN cd cds ON cds.artist = me.artistid
105       GROUP BY me.artistid, me.name, me.rank, me.charfield
106       ORDER BY amount_of_cds DESC
107     )',
108     [],
109   );
110
111   is ($rs->next->get_column ('num_cds'), 3, 'Function aliased correctly');
112 }
113
114 # These two rely on the database to throw an exception. This might not be the case one day. Please revise.
115 dies_ok(sub { my $count = $schema->resultset('Tag')->search({}, { '+select' => \'tagid AS tag_id', distinct => 1 })->count }, 'expecting to die');
116
117 done_testing;