t/66-postgres-dbi-parser.t

   1 #!/usr/bin/perl
   2 # vim: set ft=perl:
   3
   4 use strict;
   5 use Test::More;
   6 use SQL::Translator;
   7 use SQL::Translator::Schema::Constants;
   8 use Test::SQL::Translator qw(maybe_plan table_ok);
   9
  10 BEGIN {
  11     maybe_plan(49, 'SQL::Translator::Parser::DBI::PostgreSQL');
  12     SQL::Translator::Parser::DBI::PostgreSQL->import('parse');
  13 }
  14
  15 use_ok('SQL::Translator::Parser::DBI::PostgreSQL');
  16
  17 my @dsn =
  18   $ENV{DBICTEST_PG_DSN} ? @ENV{ map { "DBICTEST_PG_$_" } qw/DSN USER PASS/ }
  19 : $ENV{DBI_DSN} ? @ENV{ map { "DBI_$_" } qw/DSN USER PASS/ }
  20 : ( "dbi:Pg:dbname=postgres", '', '' );
  21
  22 my $dbh = eval {
  23   DBI->connect(@dsn, {AutoCommit => 1, RaiseError=>1,PrintError => 1} );
  24 };
  25
  26 SKIP: {
  27     if (my $err = ($@ || $DBI::err )) {
  28       chomp $err;
  29       skip "No connection to test db. DBI says '$err'", 48;
  30     }
  31
  32     ok($dbh, "dbh setup correctly");
  33     $dbh->do('SET client_min_messages=WARNING');
  34
  35 my $t   = SQL::Translator->new( trace => 0 );
  36 my $sql = q[
  37     drop table if exists sqlt_test2;
  38     drop table if exists sqlt_test1;
  39     drop table if exists sqlt_products_1;
  40
  41     create table sqlt_test1 (
  42         f_serial serial NOT NULL primary key,
  43         f_varchar character varying (255),
  44         f_text text default 'FOO'
  45     );
  46
  47
  48     create table sqlt_test2 (
  49         f_id integer NOT NULL,
  50         f_int smallint,
  51         primary key (f_id),
  52         f_fk1 integer NOT NULL references sqlt_test1 (f_serial)
  53     );
  54
  55     CREATE TABLE sqlt_products_1 (
  56         product_no integer,
  57         name text,
  58         price numeric
  59     );
  60 ];
  61
  62 $| = 1;
  63
  64 $dbh->do($sql);
  65
  66 my $data   = SQL::Translator::Parser::DBI::PostgreSQL::parse( $t, $dbh );
  67 my $schema = $t->schema;
  68
  69 isa_ok( $schema, 'SQL::Translator::Schema', 'Schema object' );
  70 my @tables = $schema->get_tables;
  71
  72 my $t1 = $schema->get_table("sqlt_test1");
  73 is( $t1->name, 'sqlt_test1', 'Table sqlt_test1 exists' );
  74
  75 my @t1_fields = $t1->get_fields;
  76 is( scalar @t1_fields, 3, '3 fields in sqlt_test1' );
  77
  78 my $f1 = shift @t1_fields;
  79 is( $f1->name, 'f_serial', 'First field is "f_serial"' );
  80 #FIXME: it should better be 'INTEGER' instead of 'int4'
  81 is( $f1->data_type, 'int4', 'Field is an integer' );
  82 is( $f1->is_nullable, 0, 'Field cannot be null' );
  83 is( $f1->default_value, "nextval('sqlt_test1_f_serial_seq'::regclass)", 'Default value is nextval()' );
  84 is( $f1->is_primary_key, 1, 'Field is PK' );
  85 #FIXME: not set to auto-increment? maybe we can guess auto-increment behavior by looking at the default_value (i.e. it call function nextval() )
  86 #is( $f1->is_auto_increment, 1, 'Field is auto increment' );
  87
  88 my $f2 = shift @t1_fields;
  89 is( $f2->name, 'f_varchar', 'Second field is "f_varchar"' );
  90 is( $f2->data_type, 'varchar', 'Field is a varchar' );
  91 is( $f2->is_nullable, 1, 'Field can be null' );
  92 #FIXME: should not be 255?
  93 is( $f2->size, 259, 'Size is "259"' );
  94 is( $f2->default_value, undef, 'Default value is undefined' );
  95 is( $f2->is_primary_key, 0, 'Field is not PK' );
  96 is( $f2->is_auto_increment, 0, 'Field is not auto increment' );
  97
  98 my $f3 = shift @t1_fields;
  99 is( $f3->name, 'f_text', 'Third field is "f_text"' );
 100 is( $f3->data_type, 'text', 'Field is a text' );
 101 is( $f3->is_nullable, 1, 'Field can be null' );
 102 is( $f3->size, 0, 'Size is 0' );
 103 is( $f3->default_value, "'FOO'::text", 'Default value is "FOO"' );
 104 is( $f3->is_primary_key, 0, 'Field is not PK' );
 105 is( $f3->is_auto_increment, 0, 'Field is not auto increment' );
 106
 107 #TODO: no 'NOT NULL' constraint not set
 108
 109 my $t2 = $schema->get_table("sqlt_test2");
 110 is( $t2->name, 'sqlt_test2', 'Table sqlt_test2 exists' );
 111
 112 my @t2_fields = $t2->get_fields;
 113 is( scalar @t2_fields, 3, '3 fields in sqlt_test2' );
 114
 115 my $t2_f1 = shift @t2_fields;
 116 is( $t2_f1->name, 'f_id', 'First field is "f_id"' );
 117 is( $t2_f1->data_type, 'int4', 'Field is an integer' );
 118 is( $t2_f1->is_nullable, 0, 'Field cannot be null' );
 119 is( $t2_f1->size, 0, 'Size is "0"' );
 120 is( $t2_f1->default_value, undef, 'Default value is undefined' );
 121 is( $t2_f1->is_primary_key, 1, 'Field is PK' );
 122
 123 my $t2_f2= shift @t2_fields;
 124 is( $t2_f2->name, 'f_int', 'Third field is "f_int"' );
 125 is( $t2_f2->data_type, 'int2', 'Field is an integer' );
 126 is( $t2_f2->is_nullable, 1, 'Field can be null' );
 127 is( $t2_f2->size, 0, 'Size is "0"' );
 128 is( $t2_f2->default_value, undef, 'Default value is undefined' );
 129 is( $t2_f2->is_primary_key, 0, 'Field is not PK' );
 130
 131 my $t2_f3 = shift @t2_fields;
 132 is( $t2_f3->name, 'f_fk1', 'Third field is "f_fk1"' );
 133 is( $t2_f3->data_type, 'int4', 'Field is an integer' );
 134 is( $t2_f3->is_nullable, 0, 'Field cannot be null' );
 135 is( $t2_f3->size, 0, 'Size is "0"' );
 136 is( $t2_f3->default_value, undef, 'Default value is undefined' );
 137 is( $t2_f3->is_primary_key, 0, 'Field is not PK' );
 138 is( $t2_f3->is_foreign_key, 1, 'Field is a FK' );
 139 my $fk_ref1 = $t2_f3->foreign_key_reference;
 140 isa_ok( $fk_ref1, 'SQL::Translator::Schema::Constraint', 'FK' );
 141 is( $fk_ref1->reference_table, 'sqlt_test1', 'FK is to "sqlt_test1" table' );
 142
 143 my @t2_constraints = $t2->get_constraints;
 144 is( scalar @t2_constraints, 1, "One constraint on table" );
 145
 146 my $t2_c1 = shift @t2_constraints;
 147 is( $t2_c1->type, FOREIGN_KEY, "Constraint is a FK" );
 148
 149 $dbh->disconnect;
 150 } # end of SKIP block
 151
 152 END {
 153   if ($dbh) {
 154     for (
 155       'drop table if exists sqlt_test2',
 156       'drop table if exists sqlt_test1',
 157       'drop table if exists sqlt_products_1',
 158     ) {
 159       eval { $dbh->do($_) };
 160     }
 161   }
 162 }