#!/usr/bin/perl
use strict;
use warnings;
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/;
# Testing 1,2,3,4...
#=============================================================================
BEGIN {
maybe_plan(53,
'SQL::Translator::Producer::PostgreSQL',
'Test::Differences',
)
}
use Test::Differences;
use SQL::Translator;
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',
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 $field1_sql = SQL::Translator::Producer::PostgreSQL::create_field($field1);
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',
size => 25,
default_value => undef,
is_auto_increment => 0,
is_nullable => 0,
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 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 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;
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($add_field, 'ALTER TABLE mytable ADD COLUMN field3 character varying(10)', 'Add field works');
my $drop_field = SQL::Translator::Producer::PostgreSQL::drop_field($field2);
is($drop_field, 'ALTER TABLE mytable DROP COLUMN myfield', 'Drop field works');
my $field3 = SQL::Translator::Schema::Field->new( name => 'time_field',
table => $table,
data_type => 'TIME',
default_value => undef,
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
my $field3_sql = SQL::Translator::Producer::PostgreSQL::create_field($field3);
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',
size => '16777215',
default_value => undef,
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
my $field4_sql = SQL::Translator::Producer::PostgreSQL::create_field($field4);
is($field4_sql, 'bytea_field bytea NOT NULL', 'Create bytea field works');
my $field5 = SQL::Translator::Schema::Field->new( name => 'enum_field',
table => $table,
data_type => 'enum',
extra => { list => [ 'Foo', 'Bar' ] },
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 });
is($field5_sql, 'enum_field mytable_enum_field_type NOT NULL', 'Create real enum field works');
my $field6 = SQL::Translator::Schema::Field->new(
name => 'character',
table => $table,
data_type => 'character',
size => '123',
default_value => 'foobar',
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0);
my $field7 = SQL::Translator::Schema::Field->new(
name => 'character',
table => $table,
data_type => 'character',
size => '123',
default_value => undef,
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0);
$alter_field = SQL::Translator::Producer::PostgreSQL::alter_field($field6,
$field7);
is($alter_field, q(ALTER TABLE mytable ALTER COLUMN character DROP DEFAULT), 'DROP DEFAULT');
$field7->default_value(q(foo'bar'));
$alter_field = SQL::Translator::Producer::PostgreSQL::alter_field($field6,
$field7);
is($alter_field, q(ALTER TABLE mytable ALTER COLUMN character SET DEFAULT 'foo''bar'''), 'DEFAULT with escaping');
$field7->default_value(\q(foobar));
$alter_field = SQL::Translator::Producer::PostgreSQL::alter_field($field6,
$field7);
is($alter_field, q(ALTER TABLE mytable ALTER COLUMN character SET DEFAULT foobar), 'DEFAULT unescaped if scalarref');
$field7->is_nullable(1);
$field7->default_value(q(foobar));
$alter_field = SQL::Translator::Producer::PostgreSQL::alter_field($field6,
$field7);
is($alter_field, q(ALTER TABLE mytable ALTER COLUMN character DROP NOT NULL), 'DROP NOT NULL');
my $field8 = SQL::Translator::Schema::Field->new( name => 'ts_field',
table => $table,
data_type => 'timestamp with time zone',
size => 6,
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
my $field8_sql = SQL::Translator::Producer::PostgreSQL::create_field($field8,{ postgres_version => 8.3 });
is($field8_sql, 'ts_field timestamp(6) with time zone NOT NULL', 'timestamp with precision');
my $field9 = SQL::Translator::Schema::Field->new( name => 'time_field',
table => $table,
data_type => 'time with time zone',
size => 6,
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
my $field9_sql = SQL::Translator::Producer::PostgreSQL::create_field($field9,{ postgres_version => 8.3 });
is($field9_sql, 'time_field time(6) with time zone NOT NULL', 'time with precision');
my $field10 = SQL::Translator::Schema::Field->new( name => 'interval_field',
table => $table,
data_type => 'interval',
size => 6,
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
my $field10_sql = SQL::Translator::Producer::PostgreSQL::create_field($field10,{ postgres_version => 8.3 });
is($field10_sql, 'interval_field interval(6) NOT NULL', 'time with precision');
my $field11 = SQL::Translator::Schema::Field->new( name => 'time_field',
table => $table,
data_type => 'time without time zone',
size => 6,
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
my $field11_sql = SQL::Translator::Producer::PostgreSQL::create_field($field11,{ postgres_version => 8.3 });
is($field11_sql, 'time_field time(6) without time zone NOT NULL', 'time with precision');
my $field12 = SQL::Translator::Schema::Field->new( name => 'time_field',
table => $table,
data_type => 'timestamp',
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
my $field12_sql = SQL::Translator::Producer::PostgreSQL::create_field($field12,{ postgres_version => 8.3 });
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' ],
custom_type_name => 'real_enum_type' },
is_auto_increment => 0,
is_nullable => 0,
is_foreign_key => 0,
is_unique => 0 );
my $field13_sql = SQL::Translator::Producer::PostgreSQL::create_field($field13,{ postgres_version => 8.3 });
is($field13_sql, 'enum_field_with_type_name real_enum_type NOT NULL', 'Create real enum field works');
{
# let's test default values! -- rjbs, 2008-09-30
my %field = (
table => $table,
data_type => 'VARCHAR',
size => 10,
is_auto_increment => 0,
is_nullable => 1,
is_foreign_key => 0,
is_unique => 0,
);
{
my $simple_default = SQL::Translator::Schema::Field->new(
%field,
name => 'str_default',
default_value => 'foo',
);
is(
$PRODUCER->($simple_default),
q{str_default character varying(10) DEFAULT 'foo'},
'default str',
);
}
{
my $null_default = SQL::Translator::Schema::Field->new(
%field,
name => 'null_default',
default_value => \'NULL',
);
is(
$PRODUCER->($null_default),
q{null_default character varying(10) DEFAULT NULL},
'default null',
);
}
{
my $null_default = SQL::Translator::Schema::Field->new(
%field,
name => 'null_default_2',
default_value => 'NULL', # XXX: this should go away
);
is(
$PRODUCER->($null_default),
q{null_default_2 character varying(10) DEFAULT NULL},
'default null from special cased string',
);
}
{
my $func_default = SQL::Translator::Schema::Field->new(
%field,
name => 'func_default',
default_value => \'func(funky)',
);
is(
$PRODUCER->($func_default),
q{func_default character varying(10) DEFAULT func(funky)},
'unquoted default from scalar ref',
);
}
}
my $view1 = SQL::Translator::Schema::View->new(
name => 'view_foo',
fields => [qw/id name/],
sql => 'SELECT id, name FROM thing',
);
my $create_opts = { add_replace_view => 1, no_comments => 1 };
my $view1_sql1 = SQL::Translator::Producer::PostgreSQL::create_view($view1, $create_opts);
my $view_sql_replace = "CREATE VIEW view_foo ( id, name ) AS
SELECT id, name FROM thing
";
is($view1_sql1, $view_sql_replace, 'correct "CREATE OR REPLACE VIEW" SQL');
my $view2 = SQL::Translator::Schema::View->new(
name => 'view_foo2',
sql => 'SELECT id, name FROM thing',
extra => {
'temporary' => '1',
'check_option' => 'cascaded',
},
);
my $create2_opts = { add_replace_view => 1, no_comments => 1 };
my $view2_sql1 = SQL::Translator::Producer::PostgreSQL::create_view($view2, $create2_opts);
my $view2_sql_replace = "CREATE TEMPORARY VIEW view_foo2 AS
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 => '"', quote_field_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");