X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2F80unique.t;h=6108f289983d6c46a5704d95f522eefbbc7128c9;hb=af2d42c04edc254f8b83010a8feadc31d0ac3860;hp=9d7634cd098f275fdd02217d4cffe44e3132dfd4;hpb=d6915f449e2d68ac184d6bc616043fd605913757;p=dbsrgits%2FDBIx-Class.git

diff --git a/t/80unique.t b/t/80unique.t
index 9d7634c..6108f28 100644
--- a/t/80unique.t
+++ b/t/80unique.t
@@ -5,9 +5,26 @@ use Test::More;
 use lib qw(t/lib);
 use DBICTest;
 
-my $schema = DBICTest::init_schema();
+my $schema = DBICTest->init_schema();
 
-plan tests => 34;
+plan tests => 45;
+
+# Check the defined unique constraints
+is_deeply(
+  [ sort $schema->source('CD')->unique_constraint_names ],
+  [ qw/cd_artist_title primary/ ],
+  'CD source has an automatically named unique constraint'
+);
+is_deeply(
+  [ sort $schema->source('Producer')->unique_constraint_names ],
+  [ qw/primary prod_name/ ],
+  'Producer source has a named unique constraint'
+);
+is_deeply(
+  [ sort $schema->source('Track')->unique_constraint_names ],
+  [ qw/primary track_cd_position track_cd_title/ ],
+  'Track source has three unique constraints'
+);
 
 my $artistid = 1;
 my $title    = 'UNIQUE Constraint';
@@ -23,14 +40,14 @@ my $cd2 = $schema->resultset('CD')->find(
     artist => $artistid,
     title  => $title,
   },
-  { key => 'artist_title' }
+  { key => 'cd_artist_title' }
 );
 
 is($cd2->get_column('artist'), $cd1->get_column('artist'), 'find by specific key: artist is correct');
 is($cd2->title, $cd1->title, 'title is correct');
 is($cd2->year, $cd1->year, 'year is correct');
 
-my $cd3 = $schema->resultset('CD')->find($artistid, $title, { key => 'artist_title' });
+my $cd3 = $schema->resultset('CD')->find($artistid, $title, { key => 'cd_artist_title' });
 
 is($cd3->get_column('artist'), $cd1->get_column('artist'), 'find by specific key, ordered columns: artist is correct');
 is($cd3->title, $cd1->title, 'title is correct');
@@ -56,7 +73,7 @@ my $cd5 = $schema->resultset('CD')->update_or_create(
     title  => $title,
     year   => 2007,
   },
-  { key => 'artist_title' }
+  { key => 'cd_artist_title' }
 );
 
 ok(! $cd5->is_changed, 'update_or_create by specific key: row is clean');
@@ -87,7 +104,7 @@ my $cd7 = $schema->resultset('CD')->find_or_create(
     title  => $title,
     year   => 2010,
   },
-  { key => 'artist_title' }
+  { key => 'cd_artist_title' }
 );
 
 is($cd7->cdid, $cd1->cdid, 'find_or_create by specific key: cdid is correct');
@@ -98,11 +115,10 @@ is($cd7->year, $cd1->year, 'year is correct');
 my $artist = $schema->resultset('Artist')->find($artistid);
 my $cd8 = $artist->find_or_create_related('cds',
   {
-    artist => $artistid,
     title  => $title,
     year   => 2020,
   },
-  { key => 'artist_title' }
+  { key => 'cd_artist_title' }
 );
 
 is($cd8->cdid, $cd1->cdid, 'find_or_create related by specific key: cdid is correct');
@@ -110,13 +126,13 @@ is($cd8->get_column('artist'), $cd1->get_column('artist'), 'artist is correct');
 is($cd8->title, $cd1->title, 'title is correct');
 is($cd8->year, $cd1->year, 'year is correct');
 
-my $cd9 = $artist->update_or_create_related('cds',
+my $cd9 = $artist->cds->update_or_create(
   {
-    artist => $artistid,
+    cdid   => $cd1->cdid,
     title  => $title,
     year   => 2021,
   },
-  { key => 'artist_title' }
+  { key => 'cd_artist_title' }
 );
 
 ok(! $cd9->is_changed, 'update_or_create by specific key: row is clean');
@@ -125,3 +141,45 @@ is($cd9->get_column('artist'), $cd1->get_column('artist'), 'artist is correct');
 is($cd9->title, $cd1->title, 'title is correct');
 is($cd9->year, 2021, 'year is correct');
 
+# Table with two unique constraints, and we're satisying one of them
+my $track = $schema->resultset('Track')->find(
+  {
+    cd       => 1,
+    position => 3,
+  },
+  { order_by => 'position' }
+);
+
+is($track->get_column('cd'), 1, 'track cd is correct');
+is($track->get_column('position'), 3, 'track position is correct');
+
+# Test a table with a unique constraint but no primary key
+my $row = $schema->resultset('NoPrimaryKey')->update_or_create(
+  {
+    foo => 1,
+    bar => 2,
+    baz => 3,
+  },
+  { key => 'foo_bar' }
+);
+
+ok(! $row->is_changed, 'update_or_create on table without primary key: row is clean');
+is($row->foo, 1, 'foo is correct');
+is($row->bar, 2, 'bar is correct');
+is($row->baz, 3, 'baz is correct');
+
+# Test a unique condition with extra information in the where attr
+{
+  my $artist = $schema->resultset('Artist')->find({ artistid => 1 });
+  my $cd = $artist->cds->find_or_new(
+    {
+      cdid  => 1,
+      title => 'Not The Real Title',
+      year  => 3000,
+    },
+    { key => 'primary' }
+  );
+
+  ok($cd->in_storage, 'find correctly grepped the key across a relationship');
+  is($cd->cdid, 1, 'cdid is correct');
+}