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;