Some cleanup
[dbsrgits/DBIx-Class.git] / t / 100extra_source.t
1 use strict;
2 use warnings;  
3
4 use Test::More;
5 use lib qw(t/lib);
6 use DBICTest;
7
8 {
9     package DBICTest::ResultSource::OtherSource;
10     use strict;
11     use warnings;
12     use base qw/DBIx::Class::ResultSource::Table/;
13 }
14
15 plan tests => 4;
16
17 my $schema = DBICTest->init_schema();
18 my $artist_source = $schema->source('Artist');
19
20 my $new_source = DBICTest::ResultSource::OtherSource->new({
21   %$artist_source,
22   name           => 'artist_preview',
23   _relationships => Storable::dclone( $artist_source->_relationships ),
24 });
25
26 $new_source->add_column('other_col' => { data_type => 'integer', default_value => 1 });
27
28 my $warn = '';
29 local $SIG{__WARN__} = sub { $warn = shift };
30
31 {
32   $schema->register_extra_source( 'artist->extra' => $new_source );
33
34   my $source = $schema->source('DBICTest::Artist');
35   is($source->source_name, 'Artist', 'original source still primary source');
36 }
37
38 {
39   my $source = $schema->source('DBICTest::Artist');
40   $schema->register_source($source->source_name, $source);
41   is($warn, '', "re-registering an existing source under the same name causes no errors");
42 }
43
44 {
45   my $new_source_name = 'Artist->preview(artist_preview)';
46   $schema->register_source( $new_source_name => $new_source );
47
48   ok(($warn =~ /DBICTest::Artist already has a source, use register_extra_source for additional sources/), 'registering extra source causes errors');
49   
50   my $source = $schema->source('DBICTest::Artist');
51   is($source->source_name, $new_source_name, 'original source still primary source');
52 }
53
54 1;