[dbsrgits/DBIx-Class.git] / t / 96multi_create.t

use strict;
use warnings;

use Test::More qw(no_plan);
use lib qw(t/lib);
use DBICTest;

my $schema = DBICTest->init_schema();

my $cd2 = $schema->resultset('CD')->create({ artist => 
                                   { name => 'Fred Bloggs' },
                                   title => 'Some CD',
                                   year => 1996
                                 });

is(ref $cd2->artist, 'DBICTest::Artist', 'Created CD and Artist object');
is($cd2->artist->name, 'Fred Bloggs', 'Artist created correctly');

my $artist = $schema->resultset('Artist')->create({ name => 'Fred 2',
                                                     cds => [
                                                             { title => 'Music to code by',
                                                               year => 2007,
                                                               tags => [
                                                                 { 'tag' => 'rock' },
                                                                 { 'tag' => 'pop' },
                                                               ],
                                                             },
                                                             ],
                                                     });
is(ref $artist->cds->first, 'DBICTest::CD', 'Created Artist with CDs');
is($artist->cds->first->title, 'Music to code by', 'CD created correctly');
is($artist->cds->first->tags->count, 2, 'Correct tags count');

# Add a new CD
$artist->update({cds => [ $artist->cds, 
                          { title => 'Yet another CD',
                            year => 2006,
                          },
                        ],
                });
is(($artist->cds->search({}, { order_by => 'year' }))[0]->title, 'Yet another CD', 'Updated and added another CD');

my $newartist = $schema->resultset('Artist')->find_or_create({ name => 'Fred 2'});

is($newartist->name, 'Fred 2', 'Retrieved the artist');


my $newartist2 = $schema->resultset('Artist')->find_or_create({ name => 'Fred 3',
                                                                cds => [
                                                                        { title => 'Noah Act',
                                                                          year => 2007,
                                                                        },
                                                                       ],

                                                              });

is($newartist2->name, 'Fred 3', 'Created new artist with cds via find_or_create');

my $artist2 = $schema->resultset('Artist')->create({
                                                    name => 'Fred 3',
                                                     cds => [
                                                             {
                                                               title => 'Music to code by',
                                                               year => 2007,
                                                             },
                                                             ],
                                                    cds_unordered => [
                                                             {
                                                               title => 'Music to code by',
                                                               year => 2007,
                                                             },
                                                             ]
                                                     });

is($artist2->in_storage, 1, 'artist with duplicate rels inserted okay');

CREATE_RELATED1 :{

	my $artist = $schema->resultset('Artist')->first;
	
	my $cd_result = $artist->create_related('cds', {
	
		title => 'TestOneCD1',
		year => 2007,
		tracks => [
		
			{ position=>111,
			  title => 'TrackOne',
			},
			{ position=>112,
			  title => 'TrackTwo',
			}
		],

	});
	
	ok( $cd_result && ref $cd_result eq 'DBICTest::CD', "Got Good CD Class");
	ok( $cd_result->title eq "TestOneCD1", "Got Expected Title");
	
	my $tracks = $cd_result->tracks;
	
	ok( ref $tracks eq "DBIx::Class::ResultSet", "Got Expected Tracks ResultSet");
	
	foreach my $track ($tracks->all)
	{
		ok( $track && ref $track eq 'DBICTest::Track', 'Got Expected Track Class');
	}
}

CREATE_RELATED2 :{

	my $artist = $schema->resultset('Artist')->first;
	
	my $cd_result = $artist->create_related('cds', {
	
		title => 'TestOneCD2',
		year => 2007,
		tracks => [
		
			{ position=>111,
			  title => 'TrackOne',
			},
			{ position=>112,
			  title => 'TrackTwo',
			}
		],

    liner_notes => { notes => 'I can haz liner notes?' },

	});
	
	ok( $cd_result && ref $cd_result eq 'DBICTest::CD', "Got Good CD Class");
	ok( $cd_result->title eq "TestOneCD2", "Got Expected Title");
  ok( $cd_result->notes eq 'I can haz liner notes?', 'Liner notes');
	
	my $tracks = $cd_result->tracks;
	
	ok( ref $tracks eq "DBIx::Class::ResultSet", "Got Expected Tracks ResultSet");
	
	foreach my $track ($tracks->all)
	{
		ok( $track && ref $track eq 'DBICTest::Track', 'Got Expected Track Class');
	}
}

my $cdp = $schema->resultset('CD_to_Producer')->create({
            cd => { artist => 1, title => 'foo', year => 2000 },
            producer => { name => 'jorge' }
          });

ok($cdp, 'join table record created ok');

SPECIAL_CASE: {
  my $kurt_cobain = { name => 'Kurt Cobain' };

  my $in_utero = $schema->resultset('CD')->new({
      title => 'In Utero',
      year  => 1993
    });

  $kurt_cobain->{cds} = [ $in_utero ];


  $schema->resultset('Artist')->populate([ $kurt_cobain ]); # %)
  $a = $schema->resultset('Artist')->find({name => 'Kurt Cobain'});

  is($a->name, 'Kurt Cobain', 'Artist insertion ok');
  is($a->cds && $a->cds->first && $a->cds->first->title, 
		  'In Utero', 'CD insertion ok');
}

SPECIAL_CASE2: {
  my $pink_floyd = { name => 'Pink Floyd' };

  my $the_wall = { title => 'The Wall', year  => 1979 };

  $pink_floyd->{cds} = [ $the_wall ];


  $schema->resultset('Artist')->populate([ $pink_floyd ]); # %)
  $a = $schema->resultset('Artist')->find({name => 'Pink Floyd'});

  is($a->name, 'Pink Floyd', 'Artist insertion ok');
  is($a->cds && $a->cds->first->title, 'The Wall', 'CD insertion ok');
}

## Create foreign key col obj including PK
## See test 20 in 66relationships.t
my $new_cd_hashref = { 
              cdid => 27, 
               title => 'Boogie Woogie', 
              year => '2007', 
              artist => { artistid => 17, name => 'king luke' }
             };

my $cd = $schema->resultset("CD")->find(1);

is($cd->artist->id, 1, 'rel okay');

my $new_cd = $schema->resultset("CD")->create($new_cd_hashref);
is($new_cd->artist->id, 17, 'new id retained okay');


# Test find or create related functionality
my $new_artist = $schema->resultset("Artist")->create({ artistid => 18, name => 'larry' });

eval {
	$schema->resultset("CD")->create({ 
              cdid => 28, 
              title => 'Boogie Wiggle', 
              year => '2007', 
              artist => { artistid => 18, name => 'larry' }
             });
};
is($@, '', 'new cd created without clash on related artist');

# Make sure exceptions from errors in created rels propogate
eval {
    my $t = $schema->resultset("Track")->new({ cd => { artist => undef } });
    #$t->cd($t->new_related('cd', { artist => undef } ) );
    #$t->{_rel_in_storage} = 0;
    $t->insert;
};
like($@, qr/cd.artist may not be NULL/, "Exception propogated properly");

# Test multi create over many_to_many
$schema->resultset('CD')->create ({
    artist => $new_artist,
    title => 'Warble Marble',
    year => '2009',
    cd_to_producer => [
        { producer => { name => 'Cowboy Neal' } },
    ],
});

my $m2m_cd = $schema->resultset('CD')->search ({ title => 'Warble Marble'});
is ($m2m_cd->count, 1, 'One CD row created via M2M create');
is ($m2m_cd->first->producers->count, 1, 'CD row created with one producer');
is ($m2m_cd->first->producers->first->name, 'Cowboy Neal', 'Correct producer row created');

# and some insane multicreate 
# (should work, despite the fact that no one will probably use it this way)

# first count how many rows do we have

my $counts;
$counts->{$_} = $schema->resultset($_)->count for qw/Artist CD Genre Producer/;

# do the crazy create
$schema->resultset('CD')->create ({
    artist => $new_artist,
    title => 'Greatest hits 1',
    year => '2012',
    genre => {
      name => '"Greatest" collections',
    },
    cd_to_producer => [
      {
        producer => {
          name => 'Dirty Harry',
          producer_to_cd => [
            {
              cd => { 
                artist => {
                  name => 'Dirty Harry himself',
                },
                title => 'Greatest hits 2',
                year => 2012,
                genre => {
                  name => '"Greatest" collections',
                },
              },
            },
            {
              cd => { 
                artist => {
                  name => 'Dirty Harry himself',
                },
                title => 'Greatest hits 3',
                year => 2012,
                genre => {
                  name => '"Greatest" collections',
                },
              },
            },
            {
              cd => { 
                artist => $new_artist,
                title => 'Greatest hits 4',
                year => 2012,
              },
            },
          ],
        },
      },
    ],
});

is ($schema->resultset ('Artist')->count, $counts->{Artist} + 1, 'One new artists created');  # even though the 'name' is not uniquely constrained find_or_create will arguably DWIM
is ($schema->resultset ('Genre')->count, $counts->{Genre} + 1, 'One additional genre created');
is ($schema->resultset ('Producer')->count, $counts->{Producer} + 1, 'One new producer');
is ($schema->resultset ('CD')->count, $counts->{CD} + 4, '4 new CDs');

my $harry_cds = $schema->resultset ('Artist')->single ({name => 'Dirty Harry himself'})->cds;
is ($harry_cds->count, 2, 'Two CDs created by Harry');
ok ($harry_cds->single ({title => 'Greatest hits 2'}), 'First CD name correct');
ok ($harry_cds->single ({title => 'Greatest hits 3'}), 'Second CD name correct');

my $harry_productions = $schema->resultset ('Producer')->single ({name => 'Dirty Harry'})
    ->search_related ('producer_to_cd', {})->search_related ('cd', {});
is ($harry_productions->count, 4, 'All 4 CDs are produced by Harry');
is ($harry_productions->search ({ year => 2012 })->count, 4, 'All 4 CDs have the correct year');

my $hits_genre = $schema->resultset ('Genre')->single ({name => '"Greatest" collections'});
ok ($hits_genre, 'New genre row found');
is ($hits_genre->cds->count, 3, 'Three of the new CDs fall into the new genre');
Commit	Line	Data
33dd4e80	1	use strict;
af2d42c0	2	use warnings;
33dd4e80	3
9c6d6d93	4	use Test::More qw(no_plan);
33dd4e80	5	use lib qw(t/lib);
	6	use DBICTest;
	7
	8	my $schema = DBICTest->init_schema();
	9
33dd4e80	10	my $cd2 = $schema->resultset('CD')->create({ artist =>
	11	{ name => 'Fred Bloggs' },
	12	title => 'Some CD',
	13	year => 1996
	14	});
	15
ac8e89d7	16	is(ref $cd2->artist, 'DBICTest::Artist', 'Created CD and Artist object');
	17	is($cd2->artist->name, 'Fred Bloggs', 'Artist created correctly');
	18
	19	my $artist = $schema->resultset('Artist')->create({ name => 'Fred 2',
	20	cds => [
	21	{ title => 'Music to code by',
	22	year => 2007,
7b7b6c88	23	tags => [
	24	{ 'tag' => 'rock' },
	25	{ 'tag' => 'pop' },
	26	],
ac8e89d7	27	},
	28	],
	29	});
	30	is(ref $artist->cds->first, 'DBICTest::CD', 'Created Artist with CDs');
	31	is($artist->cds->first->title, 'Music to code by', 'CD created correctly');
7b7b6c88	32	is($artist->cds->first->tags->count, 2, 'Correct tags count');
af2d42c0	33
	34	# Add a new CD
	35	$artist->update({cds => [ $artist->cds,
	36	{ title => 'Yet another CD',
	37	year => 2006,
	38	},
	39	],
	40	});
	41	is(($artist->cds->search({}, { order_by => 'year' }))[0]->title, 'Yet another CD', 'Updated and added another CD');
	42
	43	my $newartist = $schema->resultset('Artist')->find_or_create({ name => 'Fred 2'});
	44
	45	is($newartist->name, 'Fred 2', 'Retrieved the artist');
	46
3d8ee6ab	47
af2d42c0	48	my $newartist2 = $schema->resultset('Artist')->find_or_create({ name => 'Fred 3',
	49	cds => [
	50	{ title => 'Noah Act',
	51	year => 2007,
	52	},
	53	],
	54
	55	});
	56
	57	is($newartist2->name, 'Fred 3', 'Created new artist with cds via find_or_create');
3d8ee6ab	58
370f2ba2	59	my $artist2 = $schema->resultset('Artist')->create({
c193d1d2	60	name => 'Fred 3',
c193d1d2	61	cds => [
370f2ba2	62	{
c193d1d2	63	title => 'Music to code by',
	64	year => 2007,
	65	},
	66	],
	67	cds_unordered => [
370f2ba2	68	{
c193d1d2	69	title => 'Music to code by',
	70	year => 2007,
	71	},
	72	]
	73	});
	74
	75	is($artist2->in_storage, 1, 'artist with duplicate rels inserted okay');
3d8ee6ab	76
	77	CREATE_RELATED1 :{
	78
	79	my $artist = $schema->resultset('Artist')->first;
	80
	81	my $cd_result = $artist->create_related('cds', {
	82
	83	title => 'TestOneCD1',
	84	year => 2007,
	85	tracks => [
	86
	87	{ position=>111,
	88	title => 'TrackOne',
	89	},
	90	{ position=>112,
	91	title => 'TrackTwo',
	92	}
	93	],
	94
	95	});
	96
	97	ok( $cd_result && ref $cd_result eq 'DBICTest::CD', "Got Good CD Class");
	98	ok( $cd_result->title eq "TestOneCD1", "Got Expected Title");
	99
	100	my $tracks = $cd_result->tracks;
	101
	102	ok( ref $tracks eq "DBIx::Class::ResultSet", "Got Expected Tracks ResultSet");
	103
	104	foreach my $track ($tracks->all)
	105	{
	106	ok( $track && ref $track eq 'DBICTest::Track', 'Got Expected Track Class');
	107	}
	108	}
	109
	110	CREATE_RELATED2 :{
	111
2ec8e594	112	my $artist = $schema->resultset('Artist')->first;
3d8ee6ab	113
2ec8e594	114	my $cd_result = $artist->create_related('cds', {
3d8ee6ab	115
2ec8e594	116	title => 'TestOneCD2',
3d8ee6ab	117	year => 2007,
	118	tracks => [
	119
	120	{ position=>111,
	121	title => 'TrackOne',
	122	},
	123	{ position=>112,
	124	title => 'TrackTwo',
	125	}
	126	],
	127
9c6d6d93	128	liner_notes => { notes => 'I can haz liner notes?' },
9c6d6d93	129
3d8ee6ab	130	});
	131
	132	ok( $cd_result && ref $cd_result eq 'DBICTest::CD', "Got Good CD Class");
2ec8e594	133	ok( $cd_result->title eq "TestOneCD2", "Got Expected Title");
9c6d6d93	134	ok( $cd_result->notes eq 'I can haz liner notes?', 'Liner notes');
3d8ee6ab	135
	136	my $tracks = $cd_result->tracks;
	137
	138	ok( ref $tracks eq "DBIx::Class::ResultSet", "Got Expected Tracks ResultSet");
	139
	140	foreach my $track ($tracks->all)
	141	{
	142	ok( $track && ref $track eq 'DBICTest::Track', 'Got Expected Track Class');
	143	}
	144	}
e5dddc05	145
	146	my $cdp = $schema->resultset('CD_to_Producer')->create({
	147	cd => { artist => 1, title => 'foo', year => 2000 },
	148	producer => { name => 'jorge' }
	149	});
	150
	151	ok($cdp, 'join table record created ok');
2bc3c81e	152
	153	SPECIAL_CASE: {
	154	my $kurt_cobain = { name => 'Kurt Cobain' };
	155
	156	my $in_utero = $schema->resultset('CD')->new({
	157	title => 'In Utero',
	158	year => 1993
	159	});
	160
	161	$kurt_cobain->{cds} = [ $in_utero ];
	162
	163
	164	$schema->resultset('Artist')->populate([ $kurt_cobain ]); # %)
	165	$a = $schema->resultset('Artist')->find({name => 'Kurt Cobain'});
	166
	167	is($a->name, 'Kurt Cobain', 'Artist insertion ok');
	168	is($a->cds && $a->cds->first && $a->cds->first->title,
	169	'In Utero', 'CD insertion ok');
	170	}
	171
	172	SPECIAL_CASE2: {
	173	my $pink_floyd = { name => 'Pink Floyd' };
	174
	175	my $the_wall = { title => 'The Wall', year => 1979 };
	176
	177	$pink_floyd->{cds} = [ $the_wall ];
	178
	179
	180	$schema->resultset('Artist')->populate([ $pink_floyd ]); # %)
	181	$a = $schema->resultset('Artist')->find({name => 'Pink Floyd'});
	182
	183	is($a->name, 'Pink Floyd', 'Artist insertion ok');
	184	is($a->cds && $a->cds->first->title, 'The Wall', 'CD insertion ok');
	185	}
e02b9964	186
	187	## Create foreign key col obj including PK
	188	## See test 20 in 66relationships.t
	189	my $new_cd_hashref = {
	190	cdid => 27,
	191	title => 'Boogie Woogie',
	192	year => '2007',
	193	artist => { artistid => 17, name => 'king luke' }
	194	};
	195
	196	my $cd = $schema->resultset("CD")->find(1);
	197
e02b9964	198	is($cd->artist->id, 1, 'rel okay');
	199
	200	my $new_cd = $schema->resultset("CD")->create($new_cd_hashref);
29cfcca2	201	is($new_cd->artist->id, 17, 'new id retained okay');
f10ac17d	202
f10ac17d	203
38c03c20	204	# Test find or create related functionality
38c03c20	205	my $new_artist = $schema->resultset("Artist")->create({ artistid => 18, name => 'larry' });
6ede9177	206
	207	eval {
	208	$schema->resultset("CD")->create({
38c03c20	209	cdid => 28,
370f2ba2	210	title => 'Boogie Wiggle',
38c03c20	211	year => '2007',
38c03c20	212	artist => { artistid => 18, name => 'larry' }
6ede9177	213	});
38c03c20	214	};
6ede9177	215	is($@, '', 'new cd created without clash on related artist');
38c03c20	216
f10ac17d	217	# Make sure exceptions from errors in created rels propogate
f10ac17d	218	eval {
370f2ba2	219	my $t = $schema->resultset("Track")->new({ cd => { artist => undef } });
	220	#$t->cd($t->new_related('cd', { artist => undef } ) );
	221	#$t->{_rel_in_storage} = 0;
f10ac17d	222	$t->insert;
	223	};
	224	like($@, qr/cd.artist may not be NULL/, "Exception propogated properly");
bbbf67eb	225
bbbf67eb	226	# Test multi create over many_to_many
76b8cf98	227	$schema->resultset('CD')->create ({
	228	artist => $new_artist,
	229	title => 'Warble Marble',
	230	year => '2009',
	231	cd_to_producer => [
	232	{ producer => { name => 'Cowboy Neal' } },
	233	],
	234	});
	235
	236	my $m2m_cd = $schema->resultset('CD')->search ({ title => 'Warble Marble'});
	237	is ($m2m_cd->count, 1, 'One CD row created via M2M create');
	238	is ($m2m_cd->first->producers->count, 1, 'CD row created with one producer');
	239	is ($m2m_cd->first->producers->first->name, 'Cowboy Neal', 'Correct producer row created');
	240
	241	# and some insane multicreate
	242	# (should work, despite the fact that no one will probably use it this way)
	243
	244	# first count how many rows do we have
	245
	246	my $counts;
	247	$counts->{$_} = $schema->resultset($_)->count for qw/Artist CD Genre Producer/;
	248
	249	# do the crazy create
	250	$schema->resultset('CD')->create ({
	251	artist => $new_artist,
	252	title => 'Greatest hits 1',
	253	year => '2012',
	254	genre => {
	255	name => '"Greatest" collections',
	256	},
	257	cd_to_producer => [
	258	{
	259	producer => {
	260	name => 'Dirty Harry',
	261	producer_to_cd => [
	262	{
	263	cd => {
	264	artist => {
	265	name => 'Dirty Harry himself',
	266	},
	267	title => 'Greatest hits 2',
	268	year => 2012,
	269	genre => {
	270	name => '"Greatest" collections',
	271	},
	272	},
	273	},
	274	{
	275	cd => {
	276	artist => {
	277	name => 'Dirty Harry himself',
	278	},
	279	title => 'Greatest hits 3',
	280	year => 2012,
	281	genre => {
	282	name => '"Greatest" collections',
	283	},
	284	},
	285	},
	286	{
	287	cd => {
	288	artist => $new_artist,
	289	title => 'Greatest hits 4',
	290	year => 2012,
291	},
292	},
293	],
294	},
295	},
296	],
297	});
298
299	is ($schema->resultset ('Artist')->count, $counts->{Artist} + 1, 'One new artists created'); # even though the 'name' is not uniquely constrained find_or_create will arguably DWIM
300	is ($schema->resultset ('Genre')->count, $counts->{Genre} + 1, 'One additional genre created');
301	is ($schema->resultset ('Producer')->count, $counts->{Producer} + 1, 'One new producer');
302	is ($schema->resultset ('CD')->count, $counts->{CD} + 4, '4 new CDs');
303
304	my $harry_cds = $schema->resultset ('Artist')->single ({name => 'Dirty Harry himself'})->cds;
305	is ($harry_cds->count, 2, 'Two CDs created by Harry');
306	ok ($harry_cds->single ({title => 'Greatest hits 2'}), 'First CD name correct');
307	ok ($harry_cds->single ({title => 'Greatest hits 3'}), 'Second CD name correct');
308
309	my $harry_productions = $schema->resultset ('Producer')->single ({name => 'Dirty Harry'})
310	->search_related ('producer_to_cd', {})->search_related ('cd', {});
311	is ($harry_productions->count, 4, 'All 4 CDs are produced by Harry');
312	is ($harry_productions->search ({ year => 2012 })->count, 4, 'All 4 CDs have the correct year');
313
314	my $hits_genre = $schema->resultset ('Genre')->single ({name => '"Greatest" collections'});
315	ok ($hits_genre, 'New genre row found');
316	is ($hits_genre->cds->count, 3, 'Three of the new CDs fall into the new genre');