Add support for USING and WHERE on indexes in PostgreSQL producer

[dbsrgits/SQL-Translator.git] / t / 07p_args.t

diff --git a/t/07p_args.t b/t/07p_args.t
index 76b922e..7aa2f06 100644 (file)
--- a/t/07p_args.t
+++ b/t/07p_args.t
@@ -6,7 +6,7 @@
 use strict;
 
 use SQL::Translator;
-use Test::More tests => 6;
+use Test::More tests => 9;
 
 sub silly_parser {
     my ($tr, $data) = @_;
@@ -14,7 +14,13 @@ sub silly_parser {
 
     my @fields = split /$pargs->{'delimiter'}/, $data;
 
-    return \@fields;
+    my $schema = $tr->schema;
+    my $table  = $schema->add_table( name => 'foo') or die $schema->error;
+    for my $value ( @fields ) {
+        my $field = $table->add_field( name => $value ) or die $table->error;
+    }
+
+    return 1;
 }
 
 # The "data" to be parsed
@@ -26,37 +32,50 @@ my $tr = SQL::Translator->new;
 $tr->parser(\&silly_parser);
 $tr->parser_args(delimiter => '\|');
 
-my $pargs = $tr->parser_args;
-my $parsed = $tr->translate(\$data);
+my $pargs  = $tr->parser_args;
+$tr->translate(\$data);
+my $schema = $tr->schema;
 
-is($pargs->{'delimiter'}, '\|',
-    "parser_args works when called directly");
-is(scalar @{$parsed}, 4,
-    "right number of fields");
+is($pargs->{'delimiter'}, '\|', "parser_args works when called directly");
+my @tables = $schema->get_tables;
+is(scalar @tables, 1, "right number of tables");
+my $table = shift @tables;
+my @fields = $table->get_fields;
+is(scalar @fields, 4, "right number of fields");
+
+#
+# Blow away the existing schema object.
+#
+$tr->reset;
 
 # Now, pass parser_args indirectly...
 $tr->parser(\&silly_parser, { delimiter => "\t" });
 $data =~ s/\|/\t/g;
 
 $pargs = $tr->parser_args;
-$parsed = $tr->translate(\$data);
+$tr->translate(\$data);
 
 is($pargs->{'delimiter'}, "\t",
     "parser_args works when called indirectly");
 
-is(scalar @{$parsed}, 4,
-    "right number of fields with new delimiter");
+@tables = $schema->get_tables;
+is(scalar @tables, 1, "right number of tables");
+$table = shift @tables;
+@fields = $table->get_fields;
+is(scalar @fields, 4, "right number of fields");
 
 undef $tr;
 $tr = SQL::Translator->new(parser => \&silly_parser,
                            parser_args => { delimiter => ":" });
 $data =~ s/\t/:/g;
 $pargs = $tr->parser_args;
-$parsed = $tr->translate(\$data);
+$tr->translate(\$data);
 
 is($pargs->{'delimiter'}, ":",
     "parser_args works when called as constructor arg");
 
-is(scalar @{$parsed}, 4,
-    "right number of fields with new delimiter");
-
+@tables = $schema->get_tables;
+is(scalar @tables, 1, "right number of tables");
+$table = shift @tables;
+@fields = $table->get_fields;
+is(scalar @fields, 4, "right number of fields with new delimiter");