X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=blobdiff_plain;f=t%2F14postgres-parser.t;h=1ba550ad82be7a46df073c84aa9d685a65fdde2d;hb=668c50396c6e55829fee2a6151b0eb6500fd9389;hp=217a2c99fa09e4705cc06f6550914fc31dc707a7;hpb=f4eb75f36c82859ee7e004d9856e2aeeb0095ebf;p=dbsrgits%2FSQL-Translator.git

diff --git a/t/14postgres-parser.t b/t/14postgres-parser.t
index 217a2c9..1ba550a 100644
--- a/t/14postgres-parser.t
+++ b/t/14postgres-parser.t
@@ -1,21 +1,21 @@
 #!/usr/bin/perl
 
 use strict;
-use Data::Dumper;
-use Test::More 'no_plan'; # plans => 1;
+use Test::More tests => 117;
 use SQL::Translator;
+use SQL::Translator::Schema::Constants;
 use SQL::Translator::Parser::PostgreSQL qw(parse);
 
 my $t   = SQL::Translator->new( trace => 0 );
 my $sql = q[
     create table t_test1 (
-        f_serial serial NOT NULL primary key,
+        f_serial serial NOT NULL default '0' primary key,
         f_varchar character varying (255),
         f_double double precision,
-        f_bigint bigint,
+        f_bigint bigint not null,
         f_char character(10),
         f_bool boolean,
-        f_bin binary data(100),
+        f_bin bytea,
         f_tz timestamp,
         f_text text,
         f_fk1 integer not null references t_test2 (f_id),
@@ -25,65 +25,193 @@ my $sql = q[
     create table t_test2 (
         f_id integer NOT NULL,
         f_varchar varchar(25),
-        primary key (f_id)
+        f_int smallint,
+        primary key (f_id),
+        check (f_int between 1 and 5)
     );
 
     alter table only t_test1 add constraint c_u1 unique (f_varchar);
 
-    alter table only t_test1 add constraint "$1" foreign key (f_fk2)
+    alter table only t_test1 add constraint "c_fk2" foreign key (f_fk2)
     references t_test2 (f_id) on update no action on delete cascade;
 ];
 
 $| = 1;
 
-my $data = parse( $t, $sql );
-#print Dumper($data),"\n";
-
-is( ref $data, 'HASH', 'Data is a hashref' );
-is( scalar keys %{ $data || {} }, 2, 'Two tables' );
-
-ok( defined $data->{'t_test1'}, 'Table t_test1 exists' );
-ok( defined $data->{'t_test2'}, 'Table t_test2 exists' );
-
-my $t1 = $data->{'t_test1'};
-my $t1_fields = $t1->{'fields'};
-is( scalar keys %{ $t1_fields }, 11, '11 fields in t_test1' );
-is( $t1_fields->{'f_serial'}{'data_type'}, 'integer', 'Field is an integer' );
-is( $t1_fields->{'f_varchar'}{'data_type'}, 'varchar', 'Field is a varchar' );
-is( $t1_fields->{'f_double'}{'data_type'}, 'float', 'Field is a float' );
-is( $t1_fields->{'f_bigint'}{'data_type'}, 'integer', 'Field is an integer' );
-is( $t1_fields->{'f_char'}{'data_type'}, 'char', 'Field is char' );
-is( $t1_fields->{'f_bool'}{'data_type'}, 'boolean', 'Field is a boolean' );
-is( $t1_fields->{'f_bin'}{'data_type'}, 'binary', 'Field is binary' );
-is( $t1_fields->{'f_tz'}{'data_type'}, 'timestamp', 'Field is a timestamp' );
-is( $t1_fields->{'f_text'}{'data_type'}, 'text', 'Field is text' );
-is( $t1_fields->{'f_fk1'}{'data_type'}, 'integer', 'Field is an integer' );
-is( $t1_fields->{'f_fk2'}{'data_type'}, 'integer', 'Field is an integer' );
-
-is( $t1_fields->{'f_serial'}{'is_primary_key'}, 1, 
-    'Field "f_serial" is primary key' );
-
-my $t1_constraints = $t1->{'constraints'};
-#print Dumper($t1_constraints),"\n";
-is( scalar @{ $t1_constraints || [] }, 6, '6 constraints on t_test1' );
-is( $t1_constraints->[-2]{'type'}, 'unique', 'Constraint is unique' );
-is( $t1_constraints->[-2]{'fields'}[0], 'f_varchar', 
-    'Constraint is on field "f_varchar"' );
-
-is( $t1_constraints->[-1]{'type'}, 'foreign_key', 'Constraint is foreign key' );
-is( $t1_constraints->[-1]{'fields'}[0], 'f_fk2', 
-    'Constraint is on field "f_fk2"' );
-is( $t1_constraints->[-1]{'reference_table'}, 't_test2', 
-    'Constraint is to table "t_test2"' );
-is( $t1_constraints->[-1]{'reference_fields'}[0], 'f_id', 
-    'Constraint is to field "f_id"' );
-is( $t1_constraints->[-1]{'on_update_do'}, 'no_action', 
-    'No action on update' );
-is( $t1_constraints->[-1]{'on_delete_do'}, 'cascade', 
-    'Cascade on delete' );
-
-my $t2 = $data->{'t_test2'};
-my $t2_fields = $t2->{'fields'};
-is( scalar keys %{ $t2_fields }, 2, '2 fields in t_test2' );
-is( $t2_fields->{'f_id'}{'data_type'}, 'integer', 'Field is an integer' );
-is( $t2_fields->{'f_varchar'}{'data_type'}, 'varchar', 'Field is an varchar' );
+my $data   = parse( $t, $sql );
+my $schema = $t->schema;
+
+isa_ok( $schema, 'SQL::Translator::Schema', 'Schema object' );
+my @tables = $schema->get_tables;
+is( scalar @tables, 2, 'Two tables' );
+
+my $t1 = shift @tables;
+is( $t1->name, 't_test1', 'Table t_test1 exists' );
+
+my @t1_fields = $t1->get_fields;
+is( scalar @t1_fields, 11, '11 fields in t_test1' );
+
+my $f1 = shift @t1_fields;
+is( $f1->name, 'f_serial', 'First field is "f_serial"' );
+is( $f1->data_type, 'integer', 'Field is an integer' );
+is( $f1->is_nullable, 0, 'Field cannot be null' );
+is( $f1->size, 11, 'Size is "11"' );
+is( $f1->default_value, '0', 'Default value is "0"' );
+is( $f1->is_primary_key, 1, 'Field is PK' );
+is( $f1->is_auto_increment, 1, 'Field is auto increment' );
+
+my $f2 = shift @t1_fields;
+is( $f2->name, 'f_varchar', 'Second field is "f_varchar"' );
+is( $f2->data_type, 'varchar', 'Field is a varchar' );
+is( $f2->is_nullable, 1, 'Field can be null' );
+is( $f2->size, 255, 'Size is "255"' );
+is( $f2->default_value, undef, 'Default value is undefined' );
+is( $f2->is_primary_key, 0, 'Field is not PK' );
+is( $f2->is_auto_increment, 0, 'Field is not auto increment' );
+
+my $f3 = shift @t1_fields;
+is( $f3->name, 'f_double', 'Third field is "f_double"' );
+is( $f3->data_type, 'float', 'Field is a float' );
+is( $f3->is_nullable, 1, 'Field can be null' );
+is( $f3->size, 20, 'Size is "20"' );
+is( $f3->default_value, undef, 'Default value is undefined' );
+is( $f3->is_primary_key, 0, 'Field is not PK' );
+
+my $f4 = shift @t1_fields;
+is( $f4->name, 'f_bigint', 'Fourth field is "f_bigint"' );
+is( $f4->data_type, 'integer', 'Field is an integer' );
+is( $f4->is_nullable, 0, 'Field cannot be null' );
+is( $f4->size, 20, 'Size is "20"' );
+is( $f4->default_value, undef, 'Default value is undefined' );
+is( $f4->is_primary_key, 0, 'Field is not PK' );
+
+my $f5 = shift @t1_fields;
+is( $f5->name, 'f_char', 'Fifth field is "f_char"' );
+is( $f5->data_type, 'char', 'Field is char' );
+is( $f5->is_nullable, 1, 'Field can be null' );
+is( $f5->size, 10, 'Size is "10"' );
+is( $f5->default_value, undef, 'Default value is undefined' );
+is( $f5->is_primary_key, 0, 'Field is not PK' );
+
+my $f6 = shift @t1_fields;
+is( $f6->name, 'f_bool', 'Sixth field is "f_bool"' );
+is( $f6->data_type, 'boolean', 'Field is a boolean' );
+is( $f6->is_nullable, 1, 'Field can be null' );
+is( $f6->size, 0, 'Size is "0"' );
+is( $f6->default_value, undef, 'Default value is undefined' );
+is( $f6->is_primary_key, 0, 'Field is not PK' );
+
+my $f7 = shift @t1_fields;
+is( $f7->name, 'f_bin', 'Seventh field is "f_bin"' );
+is( $f7->data_type, 'bytea', 'Field is bytea' );
+is( $f7->is_nullable, 1, 'Field can be null' );
+is( $f7->size, 0, 'Size is "0"' );
+is( $f7->default_value, undef, 'Default value is undefined' );
+is( $f7->is_primary_key, 0, 'Field is not PK' );
+
+my $f8 = shift @t1_fields;
+is( $f8->name, 'f_tz', 'Eighth field is "f_tz"' );
+is( $f8->data_type, 'timestamp', 'Field is a timestamp' );
+is( $f8->is_nullable, 1, 'Field can be null' );
+is( $f8->size, 0, 'Size is "0"' );
+is( $f8->default_value, undef, 'Default value is undefined' );
+is( $f8->is_primary_key, 0, 'Field is not PK' );
+
+my $f9 = shift @t1_fields;
+is( $f9->name, 'f_text', 'Ninth field is "f_text"' );
+is( $f9->data_type, 'text', 'Field is text' );
+is( $f9->is_nullable, 1, 'Field can be null' );
+is( $f9->size, 64000, 'Size is "64,000"' );
+is( $f9->default_value, undef, 'Default value is undefined' );
+is( $f9->is_primary_key, 0, 'Field is not PK' );
+
+my $f10 = shift @t1_fields;
+is( $f10->name, 'f_fk1', 'Tenth field is "f_fk1"' );
+is( $f10->data_type, 'integer', 'Field is an integer' );
+is( $f10->is_nullable, 0, 'Field cannot be null' );
+is( $f10->size, 10, 'Size is "10"' );
+is( $f10->default_value, undef, 'Default value is undefined' );
+is( $f10->is_primary_key, 0, 'Field is not PK' );
+is( $f10->is_foreign_key, 1, 'Field is a FK' );
+my $fk_ref1 = $f10->foreign_key_reference;
+isa_ok( $fk_ref1, 'SQL::Translator::Schema::Constraint', 'FK' );
+is( $fk_ref1->reference_table, 't_test2', 'FK is to "t_test2" table' );
+
+my $f11 = shift @t1_fields;
+is( $f11->name, 'f_fk2', 'Eleventh field is "f_fk2"' );
+is( $f11->data_type, 'integer', 'Field is an integer' );
+is( $f11->is_nullable, 1, 'Field can be null' );
+is( $f11->size, 10, 'Size is "10"' );
+is( $f11->default_value, undef, 'Default value is undefined' );
+is( $f11->is_primary_key, 0, 'Field is not PK' );
+is( $f11->is_foreign_key, 1, 'Field is a FK' );
+my $fk_ref2 = $f11->foreign_key_reference;
+isa_ok( $fk_ref2, 'SQL::Translator::Schema::Constraint', 'FK' );
+is( $fk_ref2->reference_table, 't_test2', 'FK is to "t_test2" table' );
+
+my @t1_constraints = $t1->get_constraints;
+is( scalar @t1_constraints, 7, '7 constraints on t_test1' );
+
+my $c1 = $t1_constraints[0];
+is( $c1->type, PRIMARY_KEY, 'First constraint is PK' );
+is( join(',', $c1->fields), 'f_serial', 'Constraint is on field "f_serial"' );
+
+my $c2 = $t1_constraints[4];
+is( $c2->type, FOREIGN_KEY, 'Second constraint is foreign key' );
+is( join(',', $c2->fields), 'f_fk1', 'Constraint is on field "f_fk1"' );
+is( $c2->reference_table, 't_test2', 'Constraint is to table "t_test2"' );
+is( join(',', $c2->reference_fields), 'f_id', 'Constraint is to field "f_id"' );
+
+my $c3 = $t1_constraints[5];
+is( $c3->type, UNIQUE, 'Third constraint is unique' );
+is( join(',', $c3->fields), 'f_varchar', 'Constraint is on field "f_varchar"' );
+
+my $c4 = $t1_constraints[6];
+is( $c4->type, FOREIGN_KEY, 'Fourth constraint is foreign key' );
+is( join(',', $c4->fields), 'f_fk2', 'Constraint is on field "f_fk2"' );
+is( $c4->reference_table, 't_test2', 'Constraint is to table "t_test2"' );
+is( join(',', $c4->reference_fields), 'f_id', 'Constraint is to field "f_id"' );
+is( $c4->on_delete, 'cascade', 'On delete: cascade' );
+is( $c4->on_update, 'no_action', 'On delete: no action' );
+
+my $t2 = shift @tables;
+is( $t2->name, 't_test2', 'Table t_test2 exists' );
+
+my @t2_fields = $t2->get_fields;
+is( scalar @t2_fields, 3, '3 fields in t_test2' );
+
+my $t2_f1 = shift @t2_fields;
+is( $t2_f1->name, 'f_id', 'First field is "f_id"' );
+is( $t2_f1->data_type, 'integer', 'Field is an integer' );
+is( $t2_f1->is_nullable, 0, 'Field cannot be null' );
+is( $t2_f1->size, 10, 'Size is "10"' );
+is( $t2_f1->default_value, undef, 'Default value is undefined' );
+is( $t2_f1->is_primary_key, 1, 'Field is PK' );
+
+my $t2_f2 = shift @t2_fields;
+is( $t2_f2->name, 'f_varchar', 'Second field is "f_varchar"' );
+is( $t2_f2->data_type, 'varchar', 'Field is an varchar' );
+is( $t2_f2->is_nullable, 1, 'Field can be null' );
+is( $t2_f2->size, 25, 'Size is "25"' );
+is( $t2_f2->default_value, undef, 'Default value is undefined' );
+is( $t2_f2->is_primary_key, 0, 'Field is not PK' );
+
+my $t2_f3 = shift @t2_fields;
+is( $t2_f3->name, 'f_int', 'Third field is "f_int"' );
+is( $t2_f3->data_type, 'integer', 'Field is an integer' );
+is( $t2_f3->is_nullable, 1, 'Field can be null' );
+is( $t2_f3->size, 5, 'Size is "5"' );
+is( $t2_f3->default_value, undef, 'Default value is undefined' );
+is( $t2_f3->is_primary_key, 0, 'Field is not PK' );
+
+my @t2_constraints = $t2->get_constraints;
+is( scalar @t2_constraints, 3, "Three constraints on table" );
+
+my $t2_c1 = shift @t2_constraints;
+is( $t2_c1->type, NOT_NULL, "Constraint is NOT NULL" );
+
+my $t2_c2 = shift @t2_constraints;
+is( $t2_c2->type, PRIMARY_KEY, "Constraint is a PK" );
+
+my $t2_c3 = shift @t2_constraints;
+is( $t2_c3->type, CHECK_C, "Constraint is a 'CHECK'" );