From: Toby Corkindale <tjc@cpan.org>
Date: Thu, 11 Jun 2009 07:21:20 +0000 (+0000)
Subject: Alter register_extra_source() to _unregister_source() and register_source()
X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=commitdiff_plain;h=refs%2Fheads%2Fabandoned%2Fruntime_added_relships;p=dbsrgits%2FDBIx-Class-Historic.git

Alter register_extra_source() to _unregister_source() and register_source()
---

diff --git a/t/relationship/afterthought.t b/t/relationship/afterthought.t
index f4942a3..7965b2a 100644
--- a/t/relationship/afterthought.t
+++ b/t/relationship/afterthought.t
@@ -21,9 +21,13 @@ my $class = $schema->class('Artist');
 $class->belongs_to('rank' => $schema->class('Lyrics'));
 
 # Re-register the source:
-$schema->register_extra_source(
-    Artist => $schema->class('Artist')->new->result_source_instance
-);
+{
+    my $class = $schema->class('Artist');
+    $schema->_unregister_source('Artist');
+    $schema->register_source(
+        Artist => $class->new->result_source_instance
+    );
+}
 
 # Now check we have the relationship:
 my $source = $schema->source('Artist');