>table(\"foo") now works

[dbsrgits/DBIx-Class.git] / t / 99dbic_sqlt_parser.t

diff --git a/t/99dbic_sqlt_parser.t b/t/99dbic_sqlt_parser.t
index 900e124..d95f698 100644 (file)
--- a/t/99dbic_sqlt_parser.t
+++ b/t/99dbic_sqlt_parser.t
@@ -5,24 +5,33 @@ use Test::More;
 use lib qw(t/lib);
 use DBICTest;
 
+
 BEGIN {
-    eval "use DBD::mysql; use SQL::Translator 0.09;";
-    plan $@
-        ? ( skip_all => 'needs SQL::Translator 0.09 for testing' )
-        : ( tests => 99 );
+    eval "use SQL::Translator 0.09003;";
+    if ($@) {
+        plan skip_all => 'needs SQL::Translator 0.09003 for testing';
+    }
 }
 
 my $schema = DBICTest->init_schema();
+# Dummy was yanked out by the sqlt hook test
+# YearXXXXCDs are views
+
+my @sources = grep { $_ ne 'Dummy' && $_ !~ /^Year\d{4}CDs$/ } 
+                $schema->sources;
+
+plan tests => ( @sources * 3);
 
 { 
        my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { } } });
 
-       foreach my $source ($schema->sources) {
-               my $table = $sqlt_schema->get_table($schema->source($source)->from);
+       foreach my $source (@sources) {
+               my $table = get_table($sqlt_schema, $schema, $source);
 
                my $fk_count = scalar(grep { $_->type eq 'FOREIGN KEY' } $table->get_constraints);
                my @indices = $table->get_indices;
                my $index_count = scalar(@indices);
+    $index_count++ if ($source eq 'TwoKeys'); # TwoKeys has the index turned off on the rel def
                is($index_count, $fk_count, "correct number of indices for $source with no args");
        }
 }
@@ -30,12 +39,13 @@ my $schema = DBICTest->init_schema();
 { 
        my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { add_fk_index => 1 } } });
 
-       foreach my $source ($schema->sources) {
-               my $table = $sqlt_schema->get_table($schema->source($source)->from);
+       foreach my $source (@sources) {
+               my $table = get_table($sqlt_schema, $schema, $source);
 
                my $fk_count = scalar(grep { $_->type eq 'FOREIGN KEY' } $table->get_constraints);
                my @indices = $table->get_indices;
                my $index_count = scalar(@indices);
+    $index_count++ if ($source eq 'TwoKeys'); # TwoKeys has the index turned off on the rel def
                is($index_count, $fk_count, "correct number of indices for $source with add_fk_index => 1");
        }
 }
@@ -43,8 +53,8 @@ my $schema = DBICTest->init_schema();
 { 
        my $sqlt_schema = create_schema({ schema => $schema, args => { parser_args => { add_fk_index => 0 } } });
 
-       foreach my $source ($schema->sources) {
-               my $table = $sqlt_schema->get_table($schema->source($source)->from);
+       foreach my $source (@sources) {
+               my $table = get_table($sqlt_schema, $schema, $source);
 
                my @indices = $table->get_indices;
                my $index_count = scalar(@indices);
@@ -70,3 +80,12 @@ sub create_schema {
        $sqlt->parser('SQL::Translator::Parser::DBIx::Class');
        return $sqlt->translate({ data => $schema }) or die $sqlt->error;
 }
+
+sub get_table {
+    my ($sqlt_schema, $schema, $source) = @_;
+
+    my $table_name = $schema->source($source)->from;
+    $table_name    = $$table_name if ref $table_name;
+
+    return $sqlt_schema->get_table($table_name);
+}