From: Justin Hunter Date: Sat, 5 Sep 2009 16:34:41 +0000 (-0700) Subject: PostgreSQL DDL parser test X-Git-Url: http://git.shadowcat.co.uk/gitweb/gitweb.cgi?a=commitdiff_plain;h=5c2813d9dcd09027448a8df61eb1f7f882ecd43b;p=dbsrgits%2FSQL-Translator-2.0-ish.git PostgreSQL DDL parser test --- diff --git a/t/14postgres-parser.t b/t/14postgres-parser.t new file mode 100644 index 0000000..d159c58 --- /dev/null +++ b/t/14postgres-parser.t @@ -0,0 +1,280 @@ +#!/usr/bin/perl +# vim: set ft=perl: + +use strict; +use Test::More; +use SQL::Translator; +use SQL::Translator::Constants qw(:sqlt_types :sqlt_constants); +use Test::SQL::Translator qw(maybe_plan); + +BEGIN { + maybe_plan(120, 'SQL::Translator::Parser::DDL::PostgreSQL'); +# SQL::Translator::Parser::PostgreSQL->import('parse'); +} + +my $t = SQL::Translator->new( trace => 0, from => 'PostgreSQL' ); +my $sql = q[ + -- comment on t_test1 + create table t_test1 ( + -- this is the primary key + f_serial serial NOT NULL default '0' primary key, + f_varchar character varying (255), + f_double double precision, + f_bigint bigint not null, + f_char character(10) default 'FOO', + f_bool boolean, + f_bin bytea, + f_tz timestamp, + f_text text, + f_fk1 integer not null references t_test2 (f_id), + f_dropped text, + f_timestamp timestamp(0) with time zone, + f_timestamp2 timestamp without time zone + ); + + create table t_test2 ( + f_id integer NOT NULL, + f_varchar varchar(25), + f_int smallint, + primary key (f_id), + check (f_int between 1 and 5) + ); + + CREATE TABLE products_1 ( + product_no integer, + name text, + price numeric + ); + + CREATE TEMP TABLE products_2 ( + product_no integer, + name text, + price numeric + ); + + CREATE TEMPORARY TABLE products_3 ( + product_no integer, + name text, + price numeric + ); + + alter table t_test1 add f_fk2 integer; + + alter table only t_test1 add constraint c_u1 unique (f_varchar); + + alter table t_test1 add constraint "c_fk2" foreign key (f_fk2) + references t_test2 (f_id) on update no action on delete cascade; + + alter table t_test1 drop column f_dropped restrict; + + alter table t_test1 alter column f_fk2 set default 'FOO'; + + alter table t_test1 alter column f_char drop default; + + -- The following are allowed by the grammar + -- but won\'t do anything... - ky + + alter table t_text1 alter column f_char set not null; + + alter table t_text1 alter column f_char drop not null; + + alter table t_test1 alter f_char set statistics 10; + + alter table t_test1 alter f_text set storage extended; + + alter table t_test1 rename column f_text to foo; + + alter table t_test1 rename to foo; + + alter table only t_test1 drop constraint foo cascade; + + alter table t_test1 owner to foo; +]; + +$| = 1; + +my $data = $t->parse( $sql ); +my $schema = $t->schema; + +isa_ok( $schema, 'SQL::Translator::Object::Schema', 'Schema object' ); +my @tables = $schema->get_tables; +is( scalar @tables, 5, 'Five tables' ); + +my $t1 = shift @tables; +is( $t1->name, 't_test1', 'Table t_test1 exists' ); + +is( $t1->comments, 'comment on t_test1', 'Table comment exists' ); + +my @t1_fields = $t1->get_fields; +is( scalar @t1_fields, 13, '13 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->comments, 'this is the primary key', 'Comment' ); +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::Object::ForeignKey', 'FK' ); +is( $fk_ref1->reference_table, 't_test2', 'FK is to "t_test2" table' ); + +my $f11 = shift @t1_fields; +is( $f11->name, 'f_timestamp', 'Eleventh field is "f_timestamp"' ); +is( $f11->data_type, 'timestamp', 'Field is a timestamp' ); +is( $f11->is_nullable, 1, 'Field can be null' ); +is( $f11->size, 0, 'Size is "0"' ); +is( $f11->default_value, undef, 'Default value is "undef"' ); +is( $f11->is_primary_key, 0, 'Field is not PK' ); +is( $f11->is_foreign_key, 0, 'Field is not FK' ); +# my $fk_ref2 = $f11->foreign_key_reference; +# isa_ok( $fk_ref2, 'SQL::Translator::Object::ForeignKey', 'FK' ); +# is( $fk_ref2->reference_table, 't_test2', 'FK is to "t_test2" table' ); + +my @t1_constraints = $t1->get_constraints; +is( scalar @t1_constraints, 8, '8 constraints on t_test1' ); + +my $c1 = $t1_constraints[0]; +is( $c1->type, NOT_NULL, 'First constraint is NOT NULL' ); +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'" ); + +# test temporary tables +is( exists $schema->get_table('products_1')->extra()->{'temporary'}, "", "Table is NOT temporary"); +is( $schema->get_table('products_2')->extra('temporary'), 1,"Table is TEMP"); +is( $schema->get_table('products_3')->extra('temporary'), 1,"Table is TEMPORARY");