1use strict;
2use warnings;
3
4use Test::More;
5use Test::Exception;
6
7use lib qw(t/lib);
8use DBICTest;
9
10my $schema = DBICTest->init_schema();
11
12lives_ok ( sub {
13  my $no_prefetch = $schema->resultset('Track')->search_related(cd =>
14    {
15      'cd.year' => "2000",
16    },
17    {
18      join => 'tags',
19      order_by => 'me.trackid',
20      rows => 1,
21    }
22  );
23
24  my $use_prefetch = $no_prefetch->search(
25    {},
26    {
27      prefetch => 'tags',
28    }
29  );
30
31  is($use_prefetch->count, $no_prefetch->count, 'counts with and without prefetch match');
32  is(
33    scalar ($use_prefetch->all),
34    scalar ($no_prefetch->all),
35    "Amount of returned rows is right"
36  );
37
38}, 'search_related prefetch with order_by works');
39
40TODO: { local $TODO = 'Unqualified columns in where clauses can not be fixed without an SQLA rewrite' if SQL::Abstract->VERSION < 2;
41lives_ok ( sub {
42  my $no_prefetch = $schema->resultset('Track')->search_related(cd =>
43    {
44      'cd.year' => "2000",
45      'tagid' => 1,
46    },
47    {
48      join => 'tags',
49      rows => 1,
50    }
51  );
52
53  my $use_prefetch = $no_prefetch->search(
54    undef,
55    {
56      prefetch => 'tags',
57    }
58  );
59
60  is(
61    scalar ($use_prefetch->all),
62    scalar ($no_prefetch->all),
63    "Amount of returned rows is right"
64  );
65  is($use_prefetch->count, $no_prefetch->count, 'counts with and without prefetch match');
66
67}, 'search_related prefetch with condition referencing unqualified column of a joined table works');
68}
69
70
71lives_ok (sub {
72    my $rs = $schema->resultset("Artwork")->search(undef, {distinct => 1})
73              ->search_related('artwork_to_artist')->search_related('artist',
74                undef,
75                { prefetch => 'cds' },
76              );
77    is($rs->all, 0, 'prefetch without WHERE (objects)');
78    is($rs->count, 0, 'prefetch without WHERE (count)');
79
80    $rs = $schema->resultset("Artwork")->search(undef, {distinct => 1})
81              ->search_related('artwork_to_artist')->search_related('artist',
82                { 'cds.title' => 'foo' },
83                { prefetch => 'cds' },
84              );
85    is($rs->all, 0, 'prefetch with WHERE (objects)');
86    is($rs->count, 0, 'prefetch with WHERE (count)');
87
88
89# test where conditions at the root of the related chain
90    my $artist_rs = $schema->resultset("Artist")->search({artistid => 2});
91    my $artist = $artist_rs->next;
92    $artist->create_related ('cds', $_) for (
93      {
94        year => 1999, title => 'vague cd', genre => { name => 'vague genre' }
95      },
96      {
97        year => 1999, title => 'vague cd2', genre => { name => 'vague genre' }
98      },
99    );
100
101    $rs = $artist_rs->search_related('cds')->search_related('genre',
102                    { 'genre.name' => 'vague genre' },
103                    { prefetch => 'cds' },
104                 );
105    is($rs->all, 1, 'base without distinct (objects)');
106    is($rs->count, 1, 'base without distinct (count)');
107    # artist -> 2 cds -> 2 genres -> 2 cds for each genre = 4
108    is($rs->search_related('cds')->all, 4, 'prefetch without distinct (objects)');
109    is($rs->search_related('cds')->count, 4, 'prefetch without distinct (count)');
110
111
112    $rs = $artist_rs->search_related('cds', {}, { distinct => 1})->search_related('genre',
113                    { 'genre.name' => 'vague genre' },
114                 );
115    is($rs->all, 2, 'distinct does not propagate over search_related (objects)');
116    is($rs->count, 2, 'distinct does not propagate over search_related (count)');
117
118    $rs = $rs->search ({}, { distinct => 1} );
119    is($rs->all, 1, 'distinct without prefetch (objects)');
120    is($rs->count, 1, 'distinct without prefetch (count)');
121
122
123    $rs = $artist_rs->search_related('cds')->search_related('genre',
124                    { 'genre.name' => 'vague genre' },
125                    { prefetch => 'cds', distinct => 1 },
126                 );
127    is($rs->all, 1, 'distinct with prefetch (objects)');
128    is($rs->count, 1, 'distinct with prefetch (count)');
129
130  TODO: {
131    local $TODO = "This makes another 2 trips to the database, it can't be right";
132    # artist -> 2 cds -> 2 genres -> 2 cds for each genre + distinct = 2
133    is($rs->search_related('cds')->all, 2, 'prefetched distinct with prefetch (objects)');
134    is($rs->search_related('cds')->count, 2, 'prefetched distinct with prefetch (count)');
135  }
136
137}, 'distinct generally works with prefetch on deep search_related chains');
138
139done_testing;
140