1use strict;
2use warnings;
3
4use Test::More;
5use lib qw(t/lib);
6use DBICTest;
7use DBIC::SqlMakerTest;
8use DBIC::DebugObj;
9
10my $schema = DBICTest->init_schema();
11
12# Check the defined unique constraints
13is_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);
18is_deeply(
19  [ sort $schema->source('Producer')->unique_constraint_names ],
20  [ qw/primary prod_name/ ],
21  'Producer source has a named unique constraint'
22);
23is_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
29my $artistid = 1;
30my $title    = 'UNIQUE Constraint';
31
32my $cd1 = $schema->resultset('CD')->find_or_create({
33  artist => $artistid,
34  title  => $title,
35  year   => 2005,
36});
37
38my $cd2 = $schema->resultset('CD')->find(
39  {
40    artist => $artistid,
41    title  => $title,
42  },
43  { key => 'cd_artist_title' }
44);
45
46is($cd2->get_column('artist'), $cd1->get_column('artist'), 'find by specific key: artist is correct');
47is($cd2->title, $cd1->title, 'title is correct');
48is($cd2->year, $cd1->year, 'year is correct');
49
50my $cd3 = $schema->resultset('CD')->find($artistid, $title, { key => 'cd_artist_title' });
51
52is($cd3->get_column('artist'), $cd1->get_column('artist'), 'find by specific key, ordered columns: artist is correct');
53is($cd3->title, $cd1->title, 'title is correct');
54is($cd3->year, $cd1->year, 'year is correct');
55
56my $cd4 = $schema->resultset('CD')->update_or_create(
57  {
58    artist => $artistid,
59    title  => $title,
60    year   => 2007,
61  },
62);
63
64ok(! $cd4->is_changed, 'update_or_create without key: row is clean');
65is($cd4->cdid, $cd2->cdid, 'cdid is correct');
66is($cd4->get_column('artist'), $cd2->get_column('artist'), 'artist is correct');
67is($cd4->title, $cd2->title, 'title is correct');
68is($cd4->year, 2007, 'updated year is correct');
69
70my $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
79ok(! $cd5->is_changed, 'update_or_create by specific key: row is clean');
80is($cd5->cdid, $cd2->cdid, 'cdid is correct');
81is($cd5->get_column('artist'), $cd2->get_column('artist'), 'artist is correct');
82is($cd5->title, $cd2->title, 'title is correct');
83is($cd5->year, 2007, 'updated year is correct');
84
85my $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
95ok(! $cd6->is_changed, 'update_or_create by PK: row is clean');
96is($cd6->cdid, $cd2->cdid, 'cdid is correct');
97is($cd6->get_column('artist'), $cd2->get_column('artist'), 'artist is correct');
98is($cd6->title, $cd2->title, 'title is correct');
99is($cd6->year, 2005, 'updated year is correct');
100
101my $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
110is($cd7->cdid, $cd1->cdid, 'find_or_create by specific key: cdid is correct');
111is($cd7->get_column('artist'), $cd1->get_column('artist'), 'artist is correct');
112is($cd7->title, $cd1->title, 'title is correct');
113is($cd7->year, $cd1->year, 'year is correct');
114
115my $artist = $schema->resultset('Artist')->find($artistid);
116my $cd8 = $artist->find_or_create_related('cds',
117  {
118    title  => $title,
119    year   => 2020,
120  },
121  { key => 'cd_artist_title' }
122);
123
124is($cd8->cdid, $cd1->cdid, 'find_or_create related by specific key: cdid is correct');
125is($cd8->get_column('artist'), $cd1->get_column('artist'), 'artist is correct');
126is($cd8->title, $cd1->title, 'title is correct');
127is($cd8->year, $cd1->year, 'year is correct');
128
129my $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
138ok(! $cd9->is_changed, 'update_or_create by specific key: row is clean');
139is($cd9->cdid, $cd1->cdid, 'cdid is correct');
140is($cd9->get_column('artist'), $cd1->get_column('artist'), 'artist is correct');
141is($cd9->title, $cd1->title, 'title is correct');
142is($cd9->year, 2021, 'year is correct');
143
144# Table with two unique constraints, and we're satisying one of them
145my $track = $schema->resultset('Track')->find(
146  {
147    cd       => 1,
148    position => 3,
149  },
150  { order_by => 'position' }
151);
152
153is($track->get_column('cd'), 1, 'track cd is correct');
154is($track->get_column('position'), 3, 'track position is correct');
155
156# Test a table with a unique constraint but no primary key
157my $row = $schema->resultset('NoPrimaryKey')->update_or_create(
158  {
159    foo => 1,
160    bar => 2,
161    baz => 3,
162  },
163  { key => 'foo_bar' }
164);
165
166ok(! $row->is_changed, 'update_or_create on table without primary key: row is clean');
167is($row->foo, 1, 'foo is correct');
168is($row->bar, 2, 'bar is correct');
169is($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
235done_testing;
236