t/14postgres-parser.t

   1 use strict;
   2 use warnings;
   3 use Test::More;
   4 use SQL::Translator;
   5 use SQL::Translator::Constants qw(:sqlt_types :sqlt_constants);
   6
   7 my $t   = SQL::Translator->new( trace => 0, from => 'PostgreSQL' );
   8 my $sql = q[
   9     -- comment on t_test1
  10     create table t_test1 (
  11         -- this is the primary key
  12         f_serial serial NOT NULL default '0' primary key,
  13         f_varchar character varying (255),
  14         f_double double precision,
  15         f_bigint bigint not null,
  16         f_char character(10) default 'FOO',
  17         f_bool boolean,
  18         f_bin bytea,
  19         f_tz timestamp,
  20         f_text text,
  21         f_fk1 integer not null references t_test2 (f_id),
  22         f_dropped text,
  23         f_timestamp timestamp(0) with time zone,
  24         f_timestamp2 timestamp without time zone
  25     );
  26
  27     create table t_test2 (
  28         f_id integer NOT NULL,
  29         f_varchar varchar(25),
  30         f_int smallint,
  31         primary key (f_id),
  32         check (f_int between 1 and 5)
  33     );
  34
  35     CREATE TABLE products_1 (
  36         product_no integer,
  37         name text,
  38         price numeric
  39     );
  40
  41     CREATE TEMP TABLE products_2 (
  42         product_no integer,
  43         name text,
  44         price numeric
  45     );
  46
  47     CREATE TEMPORARY TABLE products_3 (
  48         product_no integer,
  49         name text,
  50         price numeric
  51     );
  52
  53     alter table t_test1 add f_fk2 integer;
  54
  55     alter table only t_test1 add constraint c_u1 unique (f_varchar);
  56
  57     alter table t_test1 add constraint "c_fk2" foreign key (f_fk2)
  58     references t_test2 (f_id) match simple
  59     on update no action on delete cascade deferrable;
  60
  61
  62     alter table t_test1 drop column f_dropped restrict;
  63
  64     alter table t_test1 alter column f_fk2 set default 'FOO';
  65
  66     alter table t_test1 alter column f_char drop default;
  67
  68     -- The following are allowed by the grammar
  69     -- but won\'t do anything... - ky
  70
  71     alter table t_text1 alter column f_char set not null;
  72
  73     alter table t_text1 alter column f_char drop not null;
  74
  75     alter table t_test1 alter f_char set statistics 10;
  76
  77     alter table t_test1 alter f_text set storage extended;
  78
  79     alter table t_test1 rename column f_text to foo;
  80
  81     alter table t_test1 rename to foo;
  82
  83     alter table only t_test1 drop constraint foo cascade;
  84
  85     alter table t_test1 owner to foo;
  86 ];
  87
  88 $| = 1;
  89
  90 my $data   = $t->parse( $sql );
  91 my $schema = $t->schema;
  92
  93 isa_ok( $schema, 'SQL::Translator::Object::Schema', 'Schema object' );
  94 my @tables = $schema->get_tables;
  95 is( scalar @tables, 5, 'Five tables' );
  96
  97 my $t1 = shift @tables;
  98 is( $t1->name, 't_test1', 'Table t_test1 exists' );
  99
 100 is( $t1->comments, 'comment on t_test1', 'Table comment exists' );
 101
 102 my @t1_fields = $t1->get_fields;
 103 is( scalar @t1_fields, 13, '13 fields in t_test1' );
 104
 105 my $f1 = shift @t1_fields;
 106 is( $f1->name, 'f_serial', 'First field is "f_serial"' );
 107 is( $f1->data_type, 'integer', 'Field is an integer' );
 108 is( $f1->is_nullable, 0, 'Field cannot be null' );
 109 is( $f1->size, 11, 'Size is "11"' );
 110 is( $f1->default_value, '0', 'Default value is "0"' );
 111 is( $f1->is_primary_key, 1, 'Field is PK' );
 112 is( $f1->comments, 'this is the primary key', 'Comment' );
 113 is( $f1->is_auto_increment, 1, 'Field is auto increment' );
 114
 115 my $f2 = shift @t1_fields;
 116 is( $f2->name, 'f_varchar', 'Second field is "f_varchar"' );
 117 is( $f2->data_type, 'varchar', 'Field is a varchar' );
 118 is( $f2->is_nullable, 1, 'Field can be null' );
 119 is( $f2->size, 255, 'Size is "255"' );
 120 is( $f2->default_value, undef, 'Default value is undefined' );
 121 is( $f2->is_primary_key, 0, 'Field is not PK' );
 122 is( $f2->is_auto_increment, 0, 'Field is not auto increment' );
 123
 124 my $f3 = shift @t1_fields;
 125 is( $f3->name, 'f_double', 'Third field is "f_double"' );
 126 is( $f3->data_type, 'float', 'Field is a float' );
 127 is( $f3->is_nullable, 1, 'Field can be null' );
 128 is( $f3->size, 20, 'Size is "20"' );
 129 is( $f3->default_value, undef, 'Default value is undefined' );
 130 is( $f3->is_primary_key, 0, 'Field is not PK' );
 131
 132 my $f4 = shift @t1_fields;
 133 is( $f4->name, 'f_bigint', 'Fourth field is "f_bigint"' );
 134 is( $f4->data_type, 'integer', 'Field is an integer' );
 135 is( $f4->is_nullable, 0, 'Field cannot be null' );
 136 is( $f4->size, 20, 'Size is "20"' );
 137 is( $f4->default_value, undef, 'Default value is undefined' );
 138 is( $f4->is_primary_key, 0, 'Field is not PK' );
 139
 140 my $f5 = shift @t1_fields;
 141 is( $f5->name, 'f_char', 'Fifth field is "f_char"' );
 142 is( $f5->data_type, 'char', 'Field is char' );
 143 is( $f5->is_nullable, 1, 'Field can be null' );
 144 is( $f5->size, 10, 'Size is "10"' );
 145 is( $f5->default_value, undef, 'Default value is undefined' );
 146 is( $f5->is_primary_key, 0, 'Field is not PK' );
 147
 148 my $f6 = shift @t1_fields;
 149 is( $f6->name, 'f_bool', 'Sixth field is "f_bool"' );
 150 is( $f6->data_type, 'boolean', 'Field is a boolean' );
 151 is( $f6->is_nullable, 1, 'Field can be null' );
 152 is( $f6->size, 0, 'Size is "0"' );
 153 is( $f6->default_value, undef, 'Default value is undefined' );
 154 is( $f6->is_primary_key, 0, 'Field is not PK' );
 155
 156 my $f7 = shift @t1_fields;
 157 is( $f7->name, 'f_bin', 'Seventh field is "f_bin"' );
 158 is( $f7->data_type, 'bytea', 'Field is bytea' );
 159 is( $f7->is_nullable, 1, 'Field can be null' );
 160 is( $f7->size, 0, 'Size is "0"' );
 161 is( $f7->default_value, undef, 'Default value is undefined' );
 162 is( $f7->is_primary_key, 0, 'Field is not PK' );
 163
 164 my $f8 = shift @t1_fields;
 165 is( $f8->name, 'f_tz', 'Eighth field is "f_tz"' );
 166 is( $f8->data_type, 'timestamp', 'Field is a timestamp' );
 167 is( $f8->is_nullable, 1, 'Field can be null' );
 168 is( $f8->size, 0, 'Size is "0"' );
 169 is( $f8->default_value, undef, 'Default value is undefined' );
 170 is( $f8->is_primary_key, 0, 'Field is not PK' );
 171
 172 my $f9 = shift @t1_fields;
 173 is( $f9->name, 'f_text', 'Ninth field is "f_text"' );
 174 is( $f9->data_type, 'text', 'Field is text' );
 175 is( $f9->is_nullable, 1, 'Field can be null' );
 176 is( $f9->size, 64000, 'Size is "64,000"' );
 177 is( $f9->default_value, undef, 'Default value is undefined' );
 178 is( $f9->is_primary_key, 0, 'Field is not PK' );
 179
 180 my $f10 = shift @t1_fields;
 181 is( $f10->name, 'f_fk1', 'Tenth field is "f_fk1"' );
 182 is( $f10->data_type, 'integer', 'Field is an integer' );
 183 is( $f10->is_nullable, 0, 'Field cannot be null' );
 184 is( $f10->size, 10, 'Size is "10"' );
 185 is( $f10->default_value, undef, 'Default value is undefined' );
 186 is( $f10->is_primary_key, 0, 'Field is not PK' );
 187 is( $f10->is_foreign_key, 1, 'Field is a FK' );
 188 my $fk_ref1 = $f10->foreign_key_reference;
 189 isa_ok( $fk_ref1, 'SQL::Translator::Object::ForeignKey', 'FK' );
 190 is( $fk_ref1->reference_table, 't_test2', 'FK is to "t_test2" table' );
 191
 192 my $f11 = shift @t1_fields;
 193 is( $f11->name, 'f_timestamp', 'Eleventh field is "f_timestamp"' );
 194 is( $f11->data_type, 'timestamp with time zone', 'Field is a timestamp with time zone' );
 195 is( $f11->is_nullable, 1, 'Field can be null' );
 196 is( $f11->size, 0, 'Size is "0"' );
 197 is( $f11->default_value, undef, 'Default value is "undef"' );
 198 is( $f11->is_primary_key, 0, 'Field is not PK' );
 199 is( $f11->is_foreign_key, 0, 'Field is not FK' );
 200 # my $fk_ref2 = $f11->foreign_key_reference;
 201 # isa_ok( $fk_ref2, 'SQL::Translator::Object::ForeignKey', 'FK' );
 202 # is( $fk_ref2->reference_table, 't_test2', 'FK is to "t_test2" table' );
 203
 204 my @t1_constraints = $t1->get_constraints;
 205 is( scalar @t1_constraints, 8, '8 constraints on t_test1' );
 206
 207 my $c1 = $t1_constraints[0];
 208 is( $c1->type, NOT_NULL, 'First constraint is NOT NULL' );
 209 is( join(',', $c1->fields), 'f_serial', 'Constraint is on field "f_serial"' );
 210
 211 my $c2 = $t1_constraints[4];
 212 is( $c2->type, FOREIGN_KEY, 'Second constraint is foreign key' );
 213 is( join(',', $c2->fields), 'f_fk1', 'Constraint is on field "f_fk1"' );
 214 is( $c2->reference_table, 't_test2', 'Constraint is to table "t_test2"' );
 215 is( join(',', $c2->reference_fields), 'f_id', 'Constraint is to field "f_id"' );
 216
 217 my $c3 = $t1_constraints[5];
 218 is( $c3->type, UNIQUE, 'Third constraint is unique' );
 219 is( join(',', $c3->fields), 'f_varchar', 'Constraint is on field "f_varchar"' );
 220
 221 my $c4 = $t1_constraints[6];
 222 is( $c4->type, FOREIGN_KEY, 'Fourth constraint is foreign key' );
 223 is( join(',', $c4->fields), 'f_fk2', 'Constraint is on field "f_fk2"' );
 224 is( $c4->reference_table, 't_test2', 'Constraint is to table "t_test2"' );
 225 is( join(',', $c4->reference_fields), 'f_id', 'Constraint is to field "f_id"' );
 226 is( $c4->on_delete, 'cascade', 'On delete: cascade' );
 227 is( $c4->on_update, 'no_action', 'On delete: no action' );
 228
 229 my $t2 = shift @tables;
 230 is( $t2->name, 't_test2', 'Table t_test2 exists' );
 231
 232 my @t2_fields = $t2->get_fields;
 233 is( scalar @t2_fields, 3, '3 fields in t_test2' );
 234
 235 my $t2_f1 = shift @t2_fields;
 236 is( $t2_f1->name, 'f_id', 'First field is "f_id"' );
 237 is( $t2_f1->data_type, 'integer', 'Field is an integer' );
 238 is( $t2_f1->is_nullable, 0, 'Field cannot be null' );
 239 is( $t2_f1->size, 10, 'Size is "10"' );
 240 is( $t2_f1->default_value, undef, 'Default value is undefined' );
 241 is( $t2_f1->is_primary_key, 1, 'Field is PK' );
 242
 243 my $t2_f2 = shift @t2_fields;
 244 is( $t2_f2->name, 'f_varchar', 'Second field is "f_varchar"' );
 245 is( $t2_f2->data_type, 'varchar', 'Field is an varchar' );
 246 is( $t2_f2->is_nullable, 1, 'Field can be null' );
 247 is( $t2_f2->size, 25, 'Size is "25"' );
 248 is( $t2_f2->default_value, undef, 'Default value is undefined' );
 249 is( $t2_f2->is_primary_key, 0, 'Field is not PK' );
 250
 251 my $t2_f3 = shift @t2_fields;
 252 is( $t2_f3->name, 'f_int', 'Third field is "f_int"' );
 253 is( $t2_f3->data_type, 'integer', 'Field is an integer' );
 254 is( $t2_f3->is_nullable, 1, 'Field can be null' );
 255 is( $t2_f3->size, 5, 'Size is "5"' );
 256 is( $t2_f3->default_value, undef, 'Default value is undefined' );
 257 is( $t2_f3->is_primary_key, 0, 'Field is not PK' );
 258
 259 my @t2_constraints = $t2->get_constraints;
 260 is( scalar @t2_constraints, 3, "Three constraints on table" );
 261
 262 my $t2_c1 = shift @t2_constraints;
 263 is( $t2_c1->type, NOT_NULL, "Constraint is NOT NULL" );
 264
 265 my $t2_c2 = shift @t2_constraints;
 266 is( $t2_c2->type, PRIMARY_KEY, "Constraint is a PK" );
 267
 268 my $t2_c3 = shift @t2_constraints;
 269 is( $t2_c3->type, CHECK_C, "Constraint is a 'CHECK'" );
 270
 271 # test temporary tables
 272 is( exists $schema->get_table('products_1')->extra()->{'temporary'}, "", "Table is NOT temporary");
 273 is( $schema->get_table('products_2')->extra('temporary'), 1,"Table is TEMP");
 274 is( $schema->get_table('products_3')->extra('temporary'), 1,"Table is TEMPORARY");
 275
 276 done_testing;