Applied patch sent in by Daniel Westermann-Clark on Oct 11 2006.

[dbsrgits/SQL-Translator.git] / t / lib / Producer / BaseTest.pm

diff --git a/t/lib/Producer/BaseTest.pm b/t/lib/Producer/BaseTest.pm
index a0e76c2..57cf39c 100644 (file)
--- a/t/lib/Producer/BaseTest.pm
+++ b/t/lib/Producer/BaseTest.pm
@@ -9,7 +9,8 @@ use base qw/SQL::Translator::Producer::TT::Base/;
 # Make sure we use our new class as the producer
 sub produce { return __PACKAGE__->new( translator => shift )->run; };
 
-sub tt_schema { local $/ = undef; \<DATA>; }
+# Note: we don't need to impliment tt_schema as the default will use the DATA
+# section by default.
 
 sub tt_vars { ( foo => "bar" ); }
 
@@ -19,9 +20,10 @@ sub tt_config { ( INTERPOLATE => 1 ); }
 
 __DATA__
 Hello World
-Tables: [% schema.get_tables %]
-[% table = schema.get_tables.first -%]
+Tables: [% schema.get_tables.join(', ') %]
+[% FOREACH table IN schema.get_tables -%]
 
 $table
 ------
 Fields: $table.field_names.join
+[% END %]