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 use DBIC::SqlMakerTest;
   9
  10 my $schema = DBICTest->init_schema();
  11
  12 plan tests => 24;
  13
  14 my $rs = $schema->resultset('CD')->search({},
  15     {
  16         '+select'   => \ 'COUNT(*)',
  17         '+as'       => 'count'
  18     }
  19 );
  20 lives_ok(sub { $rs->first->get_column('count') }, 'additional count rscolumn present');
  21 dies_ok(sub { $rs->first->get_column('nonexistent_column') }, 'nonexistant column requests still throw exceptions');
  22
  23 $rs = $schema->resultset('CD')->search({},
  24     {
  25         '+select'   => [ \ 'COUNT(*)', 'title' ],
  26         '+as'       => [ 'count', 'addedtitle' ]
  27     }
  28 );
  29 lives_ok(sub { $rs->first->get_column('count') }, 'multiple +select/+as columns, 1st rscolumn present');
  30 lives_ok(sub { $rs->first->get_column('addedtitle') }, 'multiple +select/+as columns, 2nd rscolumn present');
  31
  32 # Tests a regression in ResultSetColumn wrt +select
  33 $rs = $schema->resultset('CD')->search(undef,
  34     {
  35         '+select'   => [ \'COUNT(*) AS year_count' ],
  36                 order_by => 'year_count'
  37         }
  38 );
  39 my @counts = $rs->get_column('cdid')->all;
  40 ok(scalar(@counts), 'got rows from ->all using +select');
  41
  42 $rs = $schema->resultset('CD')->search({},
  43     {
  44         '+select'   => [ \ 'COUNT(*)', 'title' ],
  45         '+as'       => [ 'count', 'addedtitle' ]
  46     }
  47 )->search({},
  48     {
  49         '+select'   => 'title',
  50         '+as'       => 'addedtitle2'
  51     }
  52 );
  53 lives_ok(sub { $rs->first->get_column('count') }, '+select/+as chained search 1st rscolumn present');
  54 lives_ok(sub { $rs->first->get_column('addedtitle') }, '+select/+as chained search 1st rscolumn present');
  55 lives_ok(sub { $rs->first->get_column('addedtitle2') }, '+select/+as chained search 3rd rscolumn present');
  56
  57
  58 # test the from search attribute (gets between the FROM and WHERE keywords, allows arbitrary subselects)
  59 # also shows that outer select attributes are ok (i.e. order_by)
  60 #
  61 # from doesn't seem to be useful without using a scalarref - there were no initial tests >:(
  62 #
  63 my $cds = $schema->resultset ('CD')->search ({}, { order_by => 'me.cdid'}); # make sure order is consistent
  64 cmp_ok ($cds->count, '>', 2, 'Initially populated with more than 2 CDs');
  65
  66 my $table = $cds->result_source->name;
  67 my $subsel = $cds->search ({}, {
  68     columns => [qw/cdid title/],
  69     from => \ "(SELECT cdid, title FROM $table LIMIT 2) me",
  70 });
  71
  72 is ($subsel->count, 2, 'Subselect correctly limited the rs to 2 cds');
  73 is ($subsel->next->title, $cds->next->title, 'First CD title match');
  74 is ($subsel->next->title, $cds->next->title, 'Second CD title match');
  75
  76 is($schema->resultset('CD')->current_source_alias, "me", '$rs->current_source_alias returns "me"');
  77
  78
  79
  80 $rs = $schema->resultset('CD')->search({},
  81     {
  82         'join' => 'artist',
  83         'columns' => ['cdid', 'title', 'artist.name'],
  84     }
  85 );
  86
  87 my ($sql, @bind) = @${$rs->as_query};
  88 is_same_sql_bind (
  89   $sql,
  90   \@bind,
  91   '(SELECT me.cdid, me.title, artist.name FROM cd me  JOIN artist artist ON artist.artistid = me.artist)',
  92   [],
  93   'Use of columns attribute results in proper sql'
  94 );
  95
  96 lives_ok(sub {
  97   $rs->first->get_column('cdid')
  98 }, 'columns 1st rscolumn present');
  99
 100 lives_ok(sub {
 101   $rs->first->get_column('title')
 102 }, 'columns 2nd rscolumn present');
 103
 104 lives_ok(sub {
 105   $rs->first->artist->get_column('name')
 106 }, 'columns 3rd rscolumn present');
 107
 108
 109
 110 $rs = $schema->resultset('CD')->search({},
 111     {
 112         'join' => 'artist',
 113         '+columns' => ['cdid', 'title', 'artist.name'],
 114     }
 115 );
 116
 117 ($sql, @bind) = @${$rs->as_query};
 118 is_same_sql_bind (
 119   $sql,
 120   \@bind,
 121   '(SELECT me.cdid, me.artist, me.title, me.year, me.genreid, me.single_track, me.cdid, me.title, artist.name FROM cd me  JOIN artist artist ON artist.artistid = me.artist)',
 122   [],
 123   'Use of columns attribute results in proper sql'
 124 );
 125
 126 lives_ok(sub {
 127   $rs->first->get_column('cdid')
 128 }, 'columns 1st rscolumn present');
 129
 130 lives_ok(sub {
 131   $rs->first->get_column('title')
 132 }, 'columns 2nd rscolumn present');
 133
 134 lives_ok(sub {
 135   $rs->first->artist->get_column('name')
 136 }, 'columns 3rd rscolumn present');
 137
 138
 139 $rs = $schema->resultset('CD')->search({'tracks.position' => { -in => [2] } },
 140   {
 141     join => 'tracks',
 142     columns => [qw/me.cdid me.title/],
 143     '+select' => ['tracks.position'],
 144     '+as' => ['track_position'],
 145
 146     # get a hashref of CD1 only (the first with a second track)
 147     result_class => 'DBIx::Class::ResultClass::HashRefInflator',
 148     order_by => 'cdid',
 149     rows => 1,
 150   }
 151 );
 152
 153 is_deeply (
 154   $rs->single,
 155   {
 156     cdid => 1,
 157     track_position => 2,
 158     title => 'Spoonful of bees',
 159   },
 160   'limited prefetch via column works on a multi-relationship',
 161 );
 162
 163 my $sub_rs = $rs->search ({},
 164   {
 165     columns => [qw/artist tracks.trackid/],    # columns should not be merged but override $rs columns
 166     '+select' => ['tracks.title'],
 167     '+as' => ['tracks.title'],
 168   }
 169 );
 170
 171 is_deeply (
 172   $sub_rs->single,
 173   {
 174     artist => 1,
 175     track_position => 2,
 176     tracks =>
 177       {
 178         trackid => 17,
 179         title => 'Apiary',
 180       },
 181   },
 182   'columns/select/as fold properly on sub-searches',
 183 );
 184
 185 TODO: {
 186   local $TODO = "Multi-collapsing still doesn't work right - HRI should be getting an arrayref, not an individual hash";
 187   is_deeply (
 188     $sub_rs->single,
 189     {
 190       artist => 1,
 191       track_position => 2,
 192       tracks => [
 193         {
 194           trackid => 17,
 195           title => 'Apiary',
 196         },
 197       ],
 198     },
 199     'columns/select/as fold properly on sub-searches',
 200   );
 201 }