Pg version check for can_insert_returning
[dbsrgits/DBIx-Class-Historic.git] / t / 80unique.t
1 use strict;
2 use warnings;
3
4 use Test::More;
5 use lib qw(t/lib);
6 use DBICTest;
7 use DBIC::SqlMakerTest;
8 use DBIC::DebugObj;
9
10 my $schema = DBICTest->init_schema();
11
12 # Check the defined unique constraints
13 is_deeply(
14   [ sort $schema->source('CD')->unique_constraint_names ],
15   [ qw/cd_artist_title primary/ ],
16   'CD source has an automatically named unique constraint'
17 );
18 is_deeply(
19   [ sort $schema->source('Producer')->unique_constraint_names ],
20   [ qw/primary prod_name/ ],
21   'Producer source has a named unique constraint'
22 );
23 is_deeply(
24   [ sort $schema->source('Track')->unique_constraint_names ],
25   [ qw/primary track_cd_position track_cd_title/ ],
26   'Track source has three unique constraints'
27 );
28
29 my $artistid = 1;
30 my $title    = 'UNIQUE Constraint';
31
32 my $cd1 = $schema->resultset('CD')->find_or_create({
33   artist => $artistid,
34   title  => $title,
35   year   => 2005,
36 });
37
38 my $cd2 = $schema->resultset('CD')->find(
39   {
40     artist => $artistid,
41     title  => $title,
42   },
43   { key => 'cd_artist_title' }
44 );
45
46 is($cd2->get_column('artist'), $cd1->get_column('artist'), 'find by specific key: artist is correct');
47 is($cd2->title, $cd1->title, 'title is correct');
48 is($cd2->year, $cd1->year, 'year is correct');
49
50 my $cd3 = $schema->resultset('CD')->find($artistid, $title, { key => 'cd_artist_title' });
51
52 is($cd3->get_column('artist'), $cd1->get_column('artist'), 'find by specific key, ordered columns: artist is correct');
53 is($cd3->title, $cd1->title, 'title is correct');
54 is($cd3->year, $cd1->year, 'year is correct');
55
56 my $cd4 = $schema->resultset('CD')->update_or_create(
57   {
58     artist => $artistid,
59     title  => $title,
60     year   => 2007,
61   },
62 );
63
64 ok(! $cd4->is_changed, 'update_or_create without key: row is clean');
65 is($cd4->cdid, $cd2->cdid, 'cdid is correct');
66 is($cd4->get_column('artist'), $cd2->get_column('artist'), 'artist is correct');
67 is($cd4->title, $cd2->title, 'title is correct');
68 is($cd4->year, 2007, 'updated year is correct');
69
70 my $cd5 = $schema->resultset('CD')->update_or_create(
71   {
72     artist => $artistid,
73     title  => $title,
74     year   => 2007,
75   },
76   { key => 'cd_artist_title' }
77 );
78
79 ok(! $cd5->is_changed, 'update_or_create by specific key: row is clean');
80 is($cd5->cdid, $cd2->cdid, 'cdid is correct');
81 is($cd5->get_column('artist'), $cd2->get_column('artist'), 'artist is correct');
82 is($cd5->title, $cd2->title, 'title is correct');
83 is($cd5->year, 2007, 'updated year is correct');
84
85 my $cd6 = $schema->resultset('CD')->update_or_create(
86   {
87     cdid   => $cd2->cdid,
88     artist => 1,
89     title  => $cd2->title,
90     year   => 2005,
91   },
92   { key => 'primary' }
93 );
94
95 ok(! $cd6->is_changed, 'update_or_create by PK: row is clean');
96 is($cd6->cdid, $cd2->cdid, 'cdid is correct');
97 is($cd6->get_column('artist'), $cd2->get_column('artist'), 'artist is correct');
98 is($cd6->title, $cd2->title, 'title is correct');
99 is($cd6->year, 2005, 'updated year is correct');
100
101 my $cd7 = $schema->resultset('CD')->find_or_create(
102   {
103     artist => $artistid,
104     title  => $title,
105     year   => 2010,
106   },
107   { key => 'cd_artist_title' }
108 );
109
110 is($cd7->cdid, $cd1->cdid, 'find_or_create by specific key: cdid is correct');
111 is($cd7->get_column('artist'), $cd1->get_column('artist'), 'artist is correct');
112 is($cd7->title, $cd1->title, 'title is correct');
113 is($cd7->year, $cd1->year, 'year is correct');
114
115 my $artist = $schema->resultset('Artist')->find($artistid);
116 my $cd8 = $artist->find_or_create_related('cds',
117   {
118     title  => $title,
119     year   => 2020,
120   },
121   { key => 'cd_artist_title' }
122 );
123
124 is($cd8->cdid, $cd1->cdid, 'find_or_create related by specific key: cdid is correct');
125 is($cd8->get_column('artist'), $cd1->get_column('artist'), 'artist is correct');
126 is($cd8->title, $cd1->title, 'title is correct');
127 is($cd8->year, $cd1->year, 'year is correct');
128
129 my $cd9 = $artist->cds->update_or_create(
130   {
131     cdid   => $cd1->cdid,
132     title  => $title,
133     year   => 2021,
134   },
135   { key => 'cd_artist_title' }
136 );
137
138 ok(! $cd9->is_changed, 'update_or_create by specific key: row is clean');
139 is($cd9->cdid, $cd1->cdid, 'cdid is correct');
140 is($cd9->get_column('artist'), $cd1->get_column('artist'), 'artist is correct');
141 is($cd9->title, $cd1->title, 'title is correct');
142 is($cd9->year, 2021, 'year is correct');
143
144 # Table with two unique constraints, and we're satisying one of them
145 my $track = $schema->resultset('Track')->find(
146   {
147     cd       => 1,
148     position => 3,
149   },
150   { order_by => 'position' }
151 );
152
153 is($track->get_column('cd'), 1, 'track cd is correct');
154 is($track->get_column('position'), 3, 'track position is correct');
155
156 # Test a table with a unique constraint but no primary key
157 my $row = $schema->resultset('NoPrimaryKey')->update_or_create(
158   {
159     foo => 1,
160     bar => 2,
161     baz => 3,
162   },
163   { key => 'foo_bar' }
164 );
165
166 ok(! $row->is_changed, 'update_or_create on table without primary key: row is clean');
167 is($row->foo, 1, 'foo is correct');
168 is($row->bar, 2, 'bar is correct');
169 is($row->baz, 3, 'baz is correct');
170
171 # Test a unique condition with extra information in the where attr
172 {
173   my $artist = $schema->resultset('Artist')->find({ artistid => 1 });
174   my $cd = $artist->cds->find_or_new(
175     {
176       cdid  => 1,
177       title => 'Not The Real Title',
178       year  => 3000,
179     },
180     { key => 'primary' }
181   );
182
183   ok($cd->in_storage, 'find correctly grepped the key across a relationship');
184   is($cd->cdid, 1, 'cdid is correct');
185 }
186
187 # Test update_or_new
188 {
189     my $cd1 = $schema->resultset('CD')->update_or_new(
190       {
191         artist => $artistid,
192         title  => "SuperHits $$",
193         year   => 2007,
194       },
195       { key => 'cd_artist_title' }
196     );
197
198     is($cd1->in_storage, 0, 'CD is not in storage yet after update_or_new');
199     $cd1->insert;
200     ok($cd1->in_storage, 'CD got added to strage after update_or_new && insert');
201
202     my $cd2 = $schema->resultset('CD')->update_or_new(
203       {
204         artist => $artistid,
205         title  => "SuperHits $$",
206         year   => 2008,
207       },
208       { key => 'cd_artist_title' }
209     );
210     ok($cd2->in_storage, 'Updating year using update_or_new was successful');
211     is($cd2->id, $cd1->id, 'Got the same CD using update_or_new');
212 }
213
214 # make sure the ident condition is assembled sanely
215 {
216   my $artist = $schema->resultset('Artist')->next;
217
218   my ($sql, @bind);
219   $schema->storage->debugobj(DBIC::DebugObj->new(\$sql, \@bind)),
220   $schema->storage->debug(1);
221
222   $artist->discard_changes;
223
224   is_same_sql_bind (
225     $sql,
226     \@bind,
227     'SELECT me.artistid, me.name, me.rank, me.charfield FROM artist me WHERE me.artistid = ?',
228     [qw/'1'/],
229   );
230
231   $schema->storage->debug(0);
232   $schema->storage->debugobj(undef);
233 }
234
235 done_testing;