t/76select.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
   9 my $schema = DBICTest->init_schema();
  10
  11 plan tests => 13;
  12
  13 my $rs = $schema->resultset('CD')->search({},
  14     {
  15         '+select'   => \ 'COUNT(*)',
  16         '+as'       => 'count'
  17     }
  18 );
  19 lives_ok(sub { $rs->first->get_column('count') }, 'additional count rscolumn present');
  20 dies_ok(sub { $rs->first->get_column('nonexistent_column') }, 'nonexistant column requests still throw exceptions');
  21
  22 $rs = $schema->resultset('CD')->search({},
  23     {
  24         '+select'   => [ \ 'COUNT(*)', 'title' ],
  25         '+as'       => [ 'count', 'addedtitle' ]
  26     }
  27 );
  28 lives_ok(sub { $rs->first->get_column('count') }, 'multiple +select/+as columns, 1st rscolumn present');
  29 lives_ok(sub { $rs->first->get_column('addedtitle') }, 'multiple +select/+as columns, 2nd rscolumn present');
  30
  31 # Tests a regression in ResultSetColumn wrt +select
  32 $rs = $schema->resultset('CD')->search(undef,
  33     {
  34         '+select'   => [ \'COUNT(*) AS year_count' ],
  35                 order_by => 'year_count'
  36         }
  37 );
  38 my @counts = $rs->get_column('cdid')->all;
  39 ok(scalar(@counts), 'got rows from ->all using +select');
  40
  41 $rs = $schema->resultset('CD')->search({},
  42     {
  43         '+select'   => [ \ 'COUNT(*)', 'title' ],
  44         '+as'       => [ 'count', 'addedtitle' ]
  45     }
  46 )->search({},
  47     {
  48         '+select'   => 'title',
  49         '+as'       => 'addedtitle2'
  50     }
  51 );
  52 lives_ok(sub { $rs->first->get_column('count') }, '+select/+as chained search 1st rscolumn present');
  53 lives_ok(sub { $rs->first->get_column('addedtitle') }, '+select/+as chained search 1st rscolumn present');
  54 lives_ok(sub { $rs->first->get_column('addedtitle2') }, '+select/+as chained search 3rd rscolumn present');
  55
  56
  57 # test the from search attribute (gets between the FROM and WHERE keywords, allows arbitrary subselects)
  58 # also shows that outer select attributes are ok (i.e. order_by)
  59 #
  60 # from doesn't seem to be useful without using a scalarref - there were no initial tests >:(
  61 #
  62 my $cds = $schema->resultset ('CD')->search ({}, { order_by => 'me.cdid'}); # make sure order is consistent
  63 cmp_ok ($cds->count, '>', 2, 'Initially populated with more than 2 CDs');
  64
  65 my $table = $cds->result_source->name;
  66 my $subsel = $cds->search ({}, {
  67     columns => [qw/cdid title/],
  68     from => \ "(SELECT cdid, title FROM $table LIMIT 2) me",
  69 });
  70
  71 is ($subsel->count, 2, 'Subselect correctly limited the rs to 2 cds');
  72 is ($subsel->next->title, $cds->next->title, 'First CD title match');
  73 is ($subsel->next->title, $cds->next->title, 'Second CD title match');
  74
  75 is($schema->resultset('CD')->current_source_alias, "me", '$rs->current_source_alias returns "me"');