use Test::More;
use Test::Exception;
use Test::SQL::Translator qw(maybe_plan);
-
+use SQL::Translator::Schema::Constants;
use Data::Dumper;
use FindBin qw/$Bin/;
#=============================================================================
BEGIN {
- maybe_plan(22,
+ maybe_plan(undef,
'SQL::Translator::Producer::PostgreSQL',
'Test::Differences',
)
my $PRODUCER = \&SQL::Translator::Producer::PostgreSQL::create_field;
+{
+ my $table = SQL::Translator::Schema::Table->new( name => 'foo.bar' );
+ my $field = SQL::Translator::Schema::Field->new( name => 'baz',
+ table => $table,
+ data_type => 'VARCHAR',
+ size => 10,
+ default_value => 'quux',
+ is_auto_increment => 0,
+ is_nullable => 0,
+ is_foreign_key => 0,
+ is_unique => 0 );
+ $table->add_field($field);
+ my ($create, $fks) = SQL::Translator::Producer::PostgreSQL::create_table($table, { quote_table_names => q{"} });
+ is($table->name, 'foo.bar');
+
+ my $expected = "--\n-- Table: foo.bar\n--\nCREATE TABLE \"foo\".\"bar\" (\n \"baz\" character varying(10) DEFAULT 'quux' NOT NULL\n)";
+ is($create, $expected);
+}
+
my $table = SQL::Translator::Schema::Table->new( name => 'mytable');
my $field1 = SQL::Translator::Schema::Field->new( name => 'myfield',
is($field1_sql, 'myfield character varying(10)', 'Create field works');
+my $field_array = SQL::Translator::Schema::Field->new( name => 'myfield',
+ table => $table,
+ data_type => 'character varying[]',
+ size => 10,
+ default_value => undef,
+ is_auto_increment => 0,
+ is_nullable => 1,
+ is_foreign_key => 0,
+ is_unique => 0 );
+
+my $field_array_sql = SQL::Translator::Producer::PostgreSQL::create_field($field_array);
+
+is($field_array_sql, 'myfield character varying(10)[]', 'Create field works');
+
my $field2 = SQL::Translator::Schema::Field->new( name => 'myfield',
table => $table,
data_type => 'VARCHAR',
is_foreign_key => 0,
is_unique => 0 );
+my $pk_constraint = SQL::Translator::Schema::Constraint->new(
+ table => $table,
+ name => 'foo',
+ fields => [qw(myfield)],
+ type => 'PRIMARY_KEY',
+);
+
+my ($pk_constraint_def_ref, $pk_constraint_fk_ref ) = SQL::Translator::Producer::PostgreSQL::create_constraint($pk_constraint);
+ok(@{$pk_constraint_def_ref} == 1 && @{$pk_constraint_fk_ref} == 0, 'precheck of create_Primary Key constraint');
+is($pk_constraint_def_ref->[0], 'CONSTRAINT foo PRIMARY KEY (myfield)', 'Create Primary Key Constraint works');
+
+my $alter_pk_constraint = SQL::Translator::Producer::PostgreSQL::alter_drop_constraint($pk_constraint);
+is($alter_pk_constraint, 'ALTER TABLE mytable DROP CONSTRAINT foo', 'Alter drop Primary Key constraint works');
+
+my $table2 = SQL::Translator::Schema::Table->new( name => 'mytable2');
+
+my $field1_2 = SQL::Translator::Schema::Field->new( name => 'myfield_2',
+ table => $table,
+ data_type => 'VARCHAR',
+ size => 10,
+ default_value => undef,
+ is_auto_increment => 0,
+ is_nullable => 1,
+ is_foreign_key => 0,
+ is_unique => 0 );
+
+# check named, and unnamed foreign keys
+for my $name ( 'foo', undef ) {
+ my $fk_constraint = SQL::Translator::Schema::Constraint->new(
+ table => $table,
+ name => $name,
+ fields => [qw(myfield)],
+ type => 'FOREIGN_KEY',
+ reference_table => $table2,
+ reference_fields => [qw(myfield_2)],
+ );
+ my $fk_constraint_2 = SQL::Translator::Schema::Constraint->new(
+ table => $table,
+ name => $name,
+ fields => [qw(myfield)],
+ type => 'FOREIGN_KEY',
+ reference_table => $table2,
+ reference_fields => [qw(myfield_2)],
+ );
+
+ my ($fk_constraint_def_ref, $fk_constraint_fk_ref ) = SQL::Translator::Producer::PostgreSQL::create_constraint($fk_constraint);
+
+ ok(@{$fk_constraint_def_ref} == 0 && @{$fk_constraint_fk_ref} == 1, 'precheck of create_Foreign Key constraint');
+
+ if ( $name ) {
+ is($fk_constraint_fk_ref->[0], "ALTER TABLE mytable ADD CONSTRAINT $name FOREIGN KEY (myfield)
+ REFERENCES mytable2 (myfield_2) DEFERRABLE", 'Create Foreign Key Constraint works');
+
+ # ToDo: may we should check if the constraint name was valid, or if next
+ # unused_name created has choosen a different one
+ my $alter_fk_constraint = SQL::Translator::Producer::PostgreSQL::alter_drop_constraint($fk_constraint);
+ is($alter_fk_constraint, "ALTER TABLE mytable DROP CONSTRAINT $name", 'Alter drop Foreign Key constraint works');
+ }
+ else {
+ is($fk_constraint_fk_ref->[0], 'ALTER TABLE mytable ADD FOREIGN KEY (myfield)
+ REFERENCES mytable2 (myfield_2) DEFERRABLE', 'Create un-named Foreign Key Constraint works');
+
+ my $alter_fk_constraint = SQL::Translator::Producer::PostgreSQL::alter_drop_constraint($fk_constraint);
+ is($alter_fk_constraint, 'ALTER TABLE mytable DROP CONSTRAINT mytable_myfield_fkey', 'Alter drop un-named Foreign Key constraint works');
+ }
+}
+
+# check named, and unnamed primary keys
+for my $name ( 'foo', undef ) {
+ my $pk_constraint = SQL::Translator::Schema::Constraint->new(
+ table => $table,
+ name => $name,
+ fields => [qw(myfield)],
+ type => 'PRIMARY_KEY',
+ );
+ my $pk_constraint_2 = SQL::Translator::Schema::Constraint->new(
+ table => $table,
+ name => $name,
+ fields => [qw(myfield)],
+ type => 'PRIMARY_KEY',
+ );
+
+ my ($pk_constraint_def_ref, $pk_constraint_pk_ref ) = SQL::Translator::Producer::PostgreSQL::create_constraint($pk_constraint);
+
+ if ( $name ) {
+ is($pk_constraint_def_ref->[0], "CONSTRAINT $name PRIMARY KEY (myfield)", 'Create Primary Key Constraint works');
+
+ # ToDo: may we should check if the constraint name was valid, or if next
+ # unused_name created has choosen a different one
+ my $alter_pk_constraint = SQL::Translator::Producer::PostgreSQL::alter_drop_constraint($pk_constraint);
+ is($alter_pk_constraint, "ALTER TABLE mytable DROP CONSTRAINT $name", 'Alter drop Primary Key constraint works');
+ }
+ else {
+ is($pk_constraint_def_ref->[0], 'PRIMARY KEY (myfield)', 'Create un-named Primary Key Constraint works');
+
+ my $alter_pk_constraint = SQL::Translator::Producer::PostgreSQL::alter_drop_constraint($pk_constraint);
+ is($alter_pk_constraint, 'ALTER TABLE mytable DROP CONSTRAINT mytable_pkey', 'Alter drop un-named Foreign Key constraint works');
+ }
+}
+
my $alter_field = SQL::Translator::Producer::PostgreSQL::alter_field($field1,
$field2);
-is($alter_field, qq[ALTER TABLE mytable ALTER COLUMN myfield SET NOT NULL
+is($alter_field, qq[ALTER TABLE mytable ALTER COLUMN myfield SET NOT NULL;
ALTER TABLE mytable ALTER COLUMN myfield TYPE character varying(25)],
'Alter field works');
+my $field1_complex = SQL::Translator::Schema::Field->new(
+ name => 'my_complex_field',
+ table => $table,
+ data_type => 'VARCHAR',
+ size => 10,
+ default_value => undef,
+ is_auto_increment => 0,
+ is_nullable => 1,
+ is_foreign_key => 0,
+ is_unique => 0
+);
+
+my $field2_complex = SQL::Translator::Schema::Field->new(
+ name => 'my_altered_field',
+ table => $table,
+ data_type => 'VARCHAR',
+ size => 60,
+ default_value => 'whatever',
+ is_auto_increment => 0,
+ is_nullable => 1,
+ is_foreign_key => 0,
+ is_unique => 0
+);
+
+my $alter_field_complex = SQL::Translator::Producer::PostgreSQL::alter_field($field1_complex, $field2_complex);
+is(
+ $alter_field_complex,
+ q{ALTER TABLE mytable RENAME COLUMN my_complex_field TO my_altered_field;
+ALTER TABLE mytable ALTER COLUMN my_altered_field TYPE character varying(60);
+ALTER TABLE mytable ALTER COLUMN my_altered_field SET DEFAULT 'whatever'},
+ 'Complex Alter field works'
+);
+
$field1->name('field3');
my $add_field = SQL::Translator::Producer::PostgreSQL::add_field($field1);
is($field3_sql, 'time_field time NOT NULL', 'Create time field works');
+my $field3_datetime_with_TZ = SQL::Translator::Schema::Field->new(
+ name => 'datetime_with_TZ',
+ table => $table,
+ data_type => 'timestamp with time zone',
+ size => 7,
+);
+
+my $field3_datetime_with_TZ_sql =
+ SQL::Translator::Producer::PostgreSQL::create_field(
+ $field3_datetime_with_TZ
+ );
+
+is(
+ $field3_datetime_with_TZ_sql,
+ 'datetime_with_TZ timestamp(6) with time zone',
+ 'Create time field with time zone and size, works'
+);
+
+my $field3_time_without_TZ = SQL::Translator::Schema::Field->new(
+ name => 'time_without_TZ',
+ table => $table,
+ data_type => 'time without time zone',
+ size => 2,
+);
+
+my $field3_time_without_TZ_sql
+ = SQL::Translator::Producer::PostgreSQL::create_field(
+ $field3_time_without_TZ
+ );
+
+is(
+ $field3_time_without_TZ_sql,
+ 'time_without_TZ time(2) without time zone',
+ 'Create time field without time zone but with size, works'
+);
+
+my $field_num = SQL::Translator::Schema::Field->new( name => 'num',
+ table => $table,
+ data_type => 'numeric',
+ size => [10,2],
+ );
+my $fieldnum_sql = SQL::Translator::Producer::PostgreSQL::create_field($field_num);
+
+is($fieldnum_sql, 'num numeric(10,2)', 'Create numeric field works');
+
+
my $field4 = SQL::Translator::Schema::Field->new( name => 'bytea_field',
table => $table,
data_type => 'bytea',
my $field5 = SQL::Translator::Schema::Field->new( name => 'enum_field',
table => $table,
data_type => 'enum',
- extra => { list => [ 'Foo', 'Bar' ] },
+ extra => { list => [ 'Foo', 'Bar', 'Ba\'z' ] },
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
-my $field5_sql = SQL::Translator::Producer::PostgreSQL::create_field($field5,{ postgres_version => 8.3 });
+my $field5_types = {};
+my $field5_sql = SQL::Translator::Producer::PostgreSQL::create_field(
+ $field5,
+ {
+ postgres_version => 8.3,
+ type_defs => $field5_types,
+ }
+);
is($field5_sql, 'enum_field mytable_enum_field_type NOT NULL', 'Create real enum field works');
-
-
-
+is_deeply(
+ $field5_types,
+ { mytable_enum_field_type =>
+ "DROP TYPE IF EXISTS mytable_enum_field_type CASCADE;\n" .
+ "CREATE TYPE mytable_enum_field_type AS ENUM ('Foo', 'Bar', 'Ba''z')"
+ },
+ 'Create real enum type works'
+);
my $field6 = SQL::Translator::Schema::Field->new(
name => 'character',
is($field12_sql, 'time_field timestamp NOT NULL', 'time with precision');
+my $field13 = SQL::Translator::Schema::Field->new( name => 'enum_field_with_type_name',
+ table => $table,
+ data_type => 'enum',
+ extra => { list => [ 'Foo', 'Bar', 'Ba\'z' ],
+ custom_type_name => 'real_enum_type' },
+ is_auto_increment => 0,
+ is_nullable => 0,
+ is_foreign_key => 0,
+ is_unique => 0 );
+
+my $field13_types = {};
+my $field13_sql = SQL::Translator::Producer::PostgreSQL::create_field(
+ $field13,
+ {
+ postgres_version => 8.3,
+ type_defs => $field13_types,
+ }
+);
+
+is($field13_sql, 'enum_field_with_type_name real_enum_type NOT NULL', 'Create real enum field works');
+is_deeply(
+ $field13_types,
+ { real_enum_type =>
+ "DROP TYPE IF EXISTS real_enum_type CASCADE;\n" .
+ "CREATE TYPE real_enum_type AS ENUM ('Foo', 'Bar', 'Ba''z')"
+ },
+ 'Create real enum type works'
+);
+
{
# let's test default values! -- rjbs, 2008-09-30
SELECT id, name FROM thing
WITH CASCADED CHECK OPTION";
is($view2_sql1, $view2_sql_replace, 'correct "CREATE OR REPLACE VIEW" SQL 2');
+
+{
+ my $table = SQL::Translator::Schema::Table->new( name => 'foobar', fields => [qw( foo bar )] );
+ my $quote = { quote_table_names => '"' };
+
+ {
+ my $index = $table->add_index(name => 'myindex', fields => ['foo']);
+ my ($def) = SQL::Translator::Producer::PostgreSQL::create_index($index);
+ is($def, "CREATE INDEX myindex on foobar (foo)", 'index created');
+ ($def) = SQL::Translator::Producer::PostgreSQL::create_index($index, $quote);
+ is($def, 'CREATE INDEX "myindex" on "foobar" ("foo")', 'index created w/ quotes');
+ }
+
+ {
+ my $index = $table->add_index(name => 'myindex', fields => ['lower(foo)']);
+ my ($def) = SQL::Translator::Producer::PostgreSQL::create_index($index);
+ is($def, "CREATE INDEX myindex on foobar (lower(foo))", 'index created');
+ ($def) = SQL::Translator::Producer::PostgreSQL::create_index($index, $quote);
+ is($def, 'CREATE INDEX "myindex" on "foobar" (lower(foo))', 'index created w/ quotes');
+ }
+
+ {
+ my $index = $table->add_index(name => 'myindex', fields => ['bar', 'lower(foo)']);
+ my ($def) = SQL::Translator::Producer::PostgreSQL::create_index($index);
+ is($def, "CREATE INDEX myindex on foobar (bar, lower(foo))", 'index created');
+ ($def) = SQL::Translator::Producer::PostgreSQL::create_index($index, $quote);
+ is($def, 'CREATE INDEX "myindex" on "foobar" ("bar", lower(foo))', 'index created w/ quotes');
+ }
+
+ {
+ my $constr = $table->add_constraint(name => 'constr', type => UNIQUE, fields => ['foo']);
+ my ($def) = SQL::Translator::Producer::PostgreSQL::create_constraint($constr);
+ is($def->[0], 'CONSTRAINT constr UNIQUE (foo)', 'constraint created');
+ ($def) = SQL::Translator::Producer::PostgreSQL::create_constraint($constr, $quote);
+ is($def->[0], 'CONSTRAINT "constr" UNIQUE ("foo")', 'constraint created w/ quotes');
+ }
+
+ {
+ my $constr = $table->add_constraint(name => 'constr', type => UNIQUE, fields => ['lower(foo)']);
+ my ($def) = SQL::Translator::Producer::PostgreSQL::create_constraint($constr);
+ is($def->[0], 'CONSTRAINT constr UNIQUE (lower(foo))', 'constraint created');
+ ($def) = SQL::Translator::Producer::PostgreSQL::create_constraint($constr, $quote);
+ is($def->[0], 'CONSTRAINT "constr" UNIQUE (lower(foo))', 'constraint created w/ quotes');
+ }
+
+ {
+ my $constr = $table->add_constraint(name => 'constr', type => UNIQUE, fields => ['bar', 'lower(foo)']);
+ my ($def) = SQL::Translator::Producer::PostgreSQL::create_constraint($constr);
+ is($def->[0], 'CONSTRAINT constr UNIQUE (bar, lower(foo))', 'constraint created');
+ ($def) = SQL::Translator::Producer::PostgreSQL::create_constraint($constr, $quote);
+ is($def->[0], 'CONSTRAINT "constr" UNIQUE ("bar", lower(foo))', 'constraint created w/ quotes');
+ }
+}
+
+my $drop_view_opts1 = { add_drop_view => 1, no_comments => 1, postgres_version => 8.001 };
+my $drop_view_8_1_produced = SQL::Translator::Producer::PostgreSQL::create_view($view1, $drop_view_opts1);
+
+my $drop_view_8_1_expected = "DROP VIEW view_foo;
+CREATE VIEW view_foo ( id, name ) AS
+ SELECT id, name FROM thing
+";
+
+is($drop_view_8_1_produced, $drop_view_8_1_expected, "My DROP VIEW statement for 8.1 is correct");
+
+my $drop_view_opts2 = { add_drop_view => 1, no_comments => 1, postgres_version => 9.001 };
+my $drop_view_9_1_produced = SQL::Translator::Producer::PostgreSQL::create_view($view1, $drop_view_opts2);
+
+my $drop_view_9_1_expected = "DROP VIEW IF EXISTS view_foo;
+CREATE VIEW view_foo ( id, name ) AS
+ SELECT id, name FROM thing
+";
+
+is($drop_view_9_1_produced, $drop_view_9_1_expected, "My DROP VIEW statement for 9.1 is correct");
+
+done_testing;
projects / dbsrgits/SQL-Translator.git / blobdiff