t/14postgres-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);
   9
  10 BEGIN {
  11     maybe_plan(undef, 'SQL::Translator::Parser::PostgreSQL');
  12     SQL::Translator::Parser::PostgreSQL->import('parse');
  13 }
  14
  15 my $t   = SQL::Translator->new( trace => 0 );
  16 my $sql = q{
  17     -- comment on t_test1
  18     create table t_test1 (
  19         -- this is the primary key
  20         f_serial serial NOT NULL default '0' primary key,
  21         f_varchar character varying (255),
  22         f_double double precision,
  23         f_bigint bigint not null,
  24         f_char character(10) default 'FOO'::character(10),
  25         f_bool boolean,
  26         f_bin bytea,
  27         f_tz timestamp default '1970-01-01 00:00:00'::TIMESTAMP,
  28         f_text text default $foo$ foo
  29 $bar$
  30 baz $foo$,
  31         f_fk1 integer not null references t_test2 (f_id),
  32         f_dropped text,
  33         f_timestamp timestamp(0) with time zone,
  34         f_timestamp2 timestamp without time zone,
  35         f_json json,
  36         f_hstore hstore,
  37         f_numarray numeric(7,2) [ ],
  38         f_uuid uuid,
  39         f_time time(0) with time zone,
  40         f_time2 time without time zone,
  41         f_text2 text default $$$$,
  42         f_text3 text default $$$ $$
  43     );
  44
  45     create table t_test2 (
  46         f_id integer NOT NULL,
  47         f_varchar varchar(25),
  48         f_int smallint,
  49         f_smallint smallint default (0)::smallint,
  50         primary key (f_id),
  51         check (f_int between 1 and 5)
  52     );
  53
  54     CREATE TABLE products_1 (
  55         product_no integer,
  56         name text,
  57         price numeric
  58     );
  59
  60     CREATE TEMP TABLE products_2 (
  61         product_no integer,
  62         name text,
  63         price numeric
  64     );
  65
  66     CREATE TEMPORARY TABLE products_3 (
  67         product_no integer,
  68         name text,
  69         price numeric
  70     );
  71
  72   CREATE TRIGGER test_trigger
  73     BEFORE INSERT OR UPDATE OR DELETE
  74     ON products_1
  75     FOR EACH ROW
  76     EXECUTE PROCEDURE foo();
  77
  78     CREATE INDEX test_index1 ON t_test1 (f_varchar);
  79     CREATE INDEX test_index2 ON t_test1 USING hash (f_char, f_bool);
  80     CREATE INDEX test_index3 ON t_test1 USING hash (f_bigint, f_tz) WHERE f_bigint = '1' AND f_tz IS NULL;
  81
  82     alter table t_test1 add f_fk2 integer;
  83
  84     alter table only t_test1 add constraint c_u1 unique (f_varchar);
  85
  86     alter table t_test1 add constraint "c_fk2" foreign key (f_fk2)
  87     references t_test2 (f_id) match simple
  88     on update no action on delete cascade deferrable;
  89
  90
  91     alter table t_test1 drop column f_dropped restrict;
  92
  93     alter table t_test1 alter column f_fk2 set default 'FOO';
  94
  95     alter table t_test1 alter column f_char drop default;
  96
  97     -- The following are allowed by the grammar
  98     -- but won\'t do anything... - ky
  99
 100     alter table t_text1 alter column f_char set not null;
 101
 102     alter table t_text1 alter column f_char drop not null;
 103
 104     alter table t_test1 alter f_char set statistics 10;
 105
 106     alter table t_test1 alter f_text set storage extended;
 107
 108     alter table t_test1 rename column f_text to foo;
 109
 110     alter table t_test1 rename to foo;
 111
 112     alter table only t_test1 drop constraint foo cascade;
 113
 114     alter table t_test1 owner to foo;
 115
 116     commit;
 117 };
 118
 119 $| = 1;
 120
 121 my $data   = parse( $t, $sql );
 122 my $schema = $t->schema;
 123
 124 isa_ok( $schema, 'SQL::Translator::Schema', 'Schema object' );
 125 my @tables = $schema->get_tables;
 126 is( scalar @tables, 5, 'Five tables' );
 127
 128 my $t1 = shift @tables;
 129 is( $t1->name, 't_test1', 'Table t_test1 exists' );
 130
 131 is( $t1->comments, 'comment on t_test1', 'Table comment exists' );
 132
 133 my @t1_fields = $t1->get_fields;
 134 is( scalar @t1_fields, 21, '21 fields in t_test1' );
 135
 136 my $f1 = shift @t1_fields;
 137 is( $f1->name, 'f_serial', 'First field is "f_serial"' );
 138 is( $f1->data_type, 'integer', 'Field is an integer' );
 139 is( $f1->is_nullable, 0, 'Field cannot be null' );
 140 is( $f1->size, 11, 'Size is "11"' );
 141 is( $f1->default_value, '0', 'Default value is "0"' );
 142 is( $f1->is_primary_key, 1, 'Field is PK' );
 143 is( $f1->comments, 'this is the primary key', 'Comment' );
 144 is( $f1->is_auto_increment, 1, 'Field is auto increment' );
 145
 146 my $f2 = shift @t1_fields;
 147 is( $f2->name, 'f_varchar', 'Second field is "f_varchar"' );
 148 is( $f2->data_type, 'varchar', 'Field is a varchar' );
 149 is( $f2->is_nullable, 1, 'Field can be null' );
 150 is( $f2->size, 255, 'Size is "255"' );
 151 is( $f2->default_value, undef, 'Default value is undefined' );
 152 is( $f2->is_primary_key, 0, 'Field is not PK' );
 153 is( $f2->is_auto_increment, 0, 'Field is not auto increment' );
 154
 155 my $f3 = shift @t1_fields;
 156 is( $f3->name, 'f_double', 'Third field is "f_double"' );
 157 is( $f3->data_type, 'float', 'Field is a float' );
 158 is( $f3->is_nullable, 1, 'Field can be null' );
 159 is( $f3->size, 20, 'Size is "20"' );
 160 is( $f3->default_value, undef, 'Default value is undefined' );
 161 is( $f3->is_primary_key, 0, 'Field is not PK' );
 162
 163 my $f4 = shift @t1_fields;
 164 is( $f4->name, 'f_bigint', 'Fourth field is "f_bigint"' );
 165 is( $f4->data_type, 'integer', 'Field is an integer' );
 166 is( $f4->is_nullable, 0, 'Field cannot be null' );
 167 is( $f4->size, 20, 'Size is "20"' );
 168 is( $f4->default_value, undef, 'Default value is undefined' );
 169 is( $f4->is_primary_key, 0, 'Field is not PK' );
 170
 171 my $f5 = shift @t1_fields;
 172 is( $f5->name, 'f_char', 'Fifth field is "f_char"' );
 173 is( $f5->data_type, 'char', 'Field is char' );
 174 is( $f5->is_nullable, 1, 'Field can be null' );
 175 is( $f5->size, 10, 'Size is "10"' );
 176 is( $f5->default_value, undef, 'Default value is undefined' );
 177 is( $f5->is_primary_key, 0, 'Field is not PK' );
 178
 179 my $f6 = shift @t1_fields;
 180 is( $f6->name, 'f_bool', 'Sixth field is "f_bool"' );
 181 is( $f6->data_type, 'boolean', 'Field is a boolean' );
 182 is( $f6->is_nullable, 1, 'Field can be null' );
 183 is( $f6->size, 0, 'Size is "0"' );
 184 is( $f6->default_value, undef, 'Default value is undefined' );
 185 is( $f6->is_primary_key, 0, 'Field is not PK' );
 186
 187 my $f7 = shift @t1_fields;
 188 is( $f7->name, 'f_bin', 'Seventh field is "f_bin"' );
 189 is( $f7->data_type, 'bytea', 'Field is bytea' );
 190 is( $f7->is_nullable, 1, 'Field can be null' );
 191 is( $f7->size, 0, 'Size is "0"' );
 192 is( $f7->default_value, undef, 'Default value is undefined' );
 193 is( $f7->is_primary_key, 0, 'Field is not PK' );
 194
 195 my $f8 = shift @t1_fields;
 196 is( $f8->name, 'f_tz', 'Eighth field is "f_tz"' );
 197 is( $f8->data_type, 'timestamp', 'Field is a timestamp' );
 198 is( $f8->is_nullable, 1, 'Field can be null' );
 199 is( $f8->size, 0, 'Size is "0"' );
 200 is( $f8->default_value, '1970-01-01 00:00:00', 'Default value is 1970-01-01 00:00:00' );
 201 is( $f8->is_primary_key, 0, 'Field is not PK' );
 202
 203 my $f9 = shift @t1_fields;
 204 is( $f9->name, 'f_text', 'Ninth field is "f_text"' );
 205 is( $f9->data_type, 'text', 'Field is text' );
 206 is( $f9->is_nullable, 1, 'Field can be null' );
 207 is( $f9->size, 64000, 'Size is "64,000"' );
 208 is( $f9->default_value, " foo\n\$bar\$\nbaz ", 'Dollar-quoted default value is " foo\n$bar$\nbaz "' );
 209 is( $f9->is_primary_key, 0, 'Field is not PK' );
 210
 211 my $f10 = shift @t1_fields;
 212 is( $f10->name, 'f_fk1', 'Tenth field is "f_fk1"' );
 213 is( $f10->data_type, 'integer', 'Field is an integer' );
 214 is( $f10->is_nullable, 0, 'Field cannot be null' );
 215 is( $f10->size, 10, 'Size is "10"' );
 216 is( $f10->default_value, undef, 'Default value is undefined' );
 217 is( $f10->is_primary_key, 0, 'Field is not PK' );
 218 is( $f10->is_foreign_key, 1, 'Field is a FK' );
 219 my $fk_ref1 = $f10->foreign_key_reference;
 220 isa_ok( $fk_ref1, 'SQL::Translator::Schema::Constraint', 'FK' );
 221 is( $fk_ref1->reference_table, 't_test2', 'FK is to "t_test2" table' );
 222
 223 my $f11 = shift @t1_fields;
 224 is( $f11->name, 'f_timestamp', 'Eleventh field is "f_timestamp"' );
 225 is( $f11->data_type, 'timestamp with time zone', 'Field is a timestamp with time zone' );
 226 is( $f11->is_nullable, 1, 'Field can be null' );
 227 is( $f11->size, 0, 'Size is "0"' );
 228 is( $f11->default_value, undef, 'Default value is "undef"' );
 229 is( $f11->is_primary_key, 0, 'Field is not PK' );
 230 is( $f11->is_foreign_key, 0, 'Field is not FK' );
 231
 232 my $f12 = shift @t1_fields;
 233 is( $f12->name, 'f_timestamp2', '12th field is "f_timestamp2"' );
 234 is( $f12->data_type, 'timestamp without time zone', 'Field is a timestamp without time zone' );
 235 is( $f12->is_nullable, 1, 'Field can be null' );
 236 is( $f12->size, 0, 'Size is "0"' );
 237 is( $f12->default_value, undef, 'Default value is "undef"' );
 238 is( $f12->is_primary_key, 0, 'Field is not PK' );
 239 is( $f12->is_foreign_key, 0, 'Field is not FK' );
 240
 241 my $f13 = shift @t1_fields;
 242 is( $f13->name, 'f_json', '13th field is "f_json"' );
 243 is( $f13->data_type, 'json', 'Field is Json' );
 244 is( $f13->is_nullable, 1, 'Field can be null' );
 245 is( $f13->size, 0, 'Size is "0"' );
 246 is( $f13->default_value, undef, 'Default value is "undef"' );
 247 is( $f13->is_primary_key, 0, 'Field is not PK' );
 248 is( $f13->is_foreign_key, 0, 'Field is not FK' );
 249
 250 my $f14 = shift @t1_fields;
 251 is( $f14->name, 'f_hstore', '14th field is "f_hstore"' );
 252 is( $f14->data_type, 'hstore', 'Field is hstore' );
 253 is( $f14->is_nullable, 1, 'Field can be null' );
 254 is( $f14->size, 0, 'Size is "0"' );
 255 is( $f14->default_value, undef, 'Default value is "undef"' );
 256 is( $f14->is_primary_key, 0, 'Field is not PK' );
 257 is( $f14->is_foreign_key, 0, 'Field is not FK' );
 258
 259 my $f15 = shift @t1_fields;
 260 is( $f15->name, 'f_numarray', '15th field is "f_numarray"' );
 261 is( $f15->data_type, 'numeric[]', 'Field is numeric[]' );
 262 is( $f15->is_nullable, 1, 'Field can be null' );
 263 is_deeply( [$f15->size], [7,2] , 'Size is "7,2"' );
 264 is( $f15->default_value, undef, 'Default value is "undef"' );
 265 is( $f15->is_primary_key, 0, 'Field is not PK' );
 266 is( $f15->is_foreign_key, 0, 'Field is not FK' );
 267
 268 my $f16 = shift @t1_fields;
 269 is( $f16->name, 'f_uuid', '16th field is "f_uuid"' );
 270 is( $f16->data_type, 'uuid', 'Field is a UUID' );
 271 is( $f16->is_nullable, 1, 'Field can be null' );
 272 is( $f16->size, 0, 'Size is "0"' );
 273 is( $f16->default_value, undef, 'Default value is "undef"' );
 274 is( $f16->is_primary_key, 0, 'Field is not PK' );
 275 is( $f16->is_foreign_key, 0, 'Field is not FK' );
 276
 277 my $f17 = shift @t1_fields;
 278 is( $f17->name, 'f_time', '17th field is "f_time"' );
 279 is( $f17->data_type, 'time with time zone', 'Field is a time with time zone' );
 280 is( $f17->is_nullable, 1, 'Field can be null' );
 281 is( $f17->size, 0, 'Size is "0"' );
 282 is( $f17->default_value, undef, 'Default value is "undef"' );
 283 is( $f17->is_primary_key, 0, 'Field is not PK' );
 284 is( $f17->is_foreign_key, 0, 'Field is not FK' );
 285
 286 my $f18 = shift @t1_fields;
 287 is( $f18->name, 'f_time2', '18th field is "f_time2"' );
 288 is( $f18->data_type, 'time without time zone', 'Field is a time without time zone' );
 289 is( $f18->is_nullable, 1, 'Field can be null' );
 290 is( $f18->size, 0, 'Size is "0"' );
 291 is( $f18->default_value, undef, 'Default value is "undef"' );
 292 is( $f18->is_primary_key, 0, 'Field is not PK' );
 293 is( $f18->is_foreign_key, 0, 'Field is not FK' );
 294
 295 my $f19 = shift @t1_fields;
 296 is( $f19->name, 'f_text2', '19th field is "f_text2"' );
 297 is( $f19->data_type, 'text', 'Field is text' );
 298 is( $f19->is_nullable, 1, 'Field can be null' );
 299 is( $f19->size, 64000, 'Size is "64,000"' );
 300 is( $f19->default_value, '', 'Dollar-quoted default value is empty' );
 301 is( $f19->is_primary_key, 0, 'Field is not PK' );
 302
 303 my $f20 = shift @t1_fields;
 304 is( $f20->name, 'f_text3', '20th field is "f_text3"' );
 305 is( $f20->data_type, 'text', 'Field is text' );
 306 is( $f20->is_nullable, 1, 'Field can be null' );
 307 is( $f20->size, 64000, 'Size is "64,000"' );
 308 is( $f20->default_value, '$ ', 'Dollar-quoted default value is "$ "' );
 309 is( $f20->is_primary_key, 0, 'Field is not PK' );
 310
 311 # my $fk_ref2 = $f11->foreign_key_reference;
 312 # isa_ok( $fk_ref2, 'SQL::Translator::Schema::Constraint', 'FK' );
 313 # is( $fk_ref2->reference_table, 't_test2', 'FK is to "t_test2" table' );
 314
 315 my @t1_constraints = $t1->get_constraints;
 316 is( scalar @t1_constraints, 8, '8 constraints on t_test1' );
 317
 318 my $c1 = $t1_constraints[0];
 319 is( $c1->type, PRIMARY_KEY, 'First constraint is PK' );
 320 is( join(',', $c1->fields), 'f_serial', 'Constraint is on field "f_serial"' );
 321
 322 my $c2 = $t1_constraints[4];
 323 is( $c2->type, FOREIGN_KEY, 'Second constraint is foreign key' );
 324 is( join(',', $c2->fields), 'f_fk1', 'Constraint is on field "f_fk1"' );
 325 is( $c2->reference_table, 't_test2', 'Constraint is to table "t_test2"' );
 326 is( join(',', $c2->reference_fields), 'f_id', 'Constraint is to field "f_id"' );
 327
 328 my $c3 = $t1_constraints[5];
 329 is( $c3->type, UNIQUE, 'Third constraint is unique' );
 330 is( join(',', $c3->fields), 'f_varchar', 'Constraint is on field "f_varchar"' );
 331
 332 my $c4 = $t1_constraints[6];
 333 is( $c4->type, FOREIGN_KEY, 'Fourth constraint is foreign key' );
 334 is( join(',', $c4->fields), 'f_fk2', 'Constraint is on field "f_fk2"' );
 335 is( $c4->reference_table, 't_test2', 'Constraint is to table "t_test2"' );
 336 is( join(',', $c4->reference_fields), 'f_id', 'Constraint is to field "f_id"' );
 337 is( $c4->on_delete, 'cascade', 'On delete: cascade' );
 338 is( $c4->on_update, 'no_action', 'On delete: no action' );
 339 is( $c4->match_type, 'simple', 'Match type: simple' );
 340 is( $c4->deferrable, 1, 'Deferrable detected' );
 341
 342 my $t2 = shift @tables;
 343 is( $t2->name, 't_test2', 'Table t_test2 exists' );
 344
 345 my @t2_fields = $t2->get_fields;
 346 is( scalar @t2_fields, 4, '4 fields in t_test2' );
 347
 348 my $t2_f1 = shift @t2_fields;
 349 is( $t2_f1->name, 'f_id', 'First field is "f_id"' );
 350 is( $t2_f1->data_type, 'integer', 'Field is an integer' );
 351 is( $t2_f1->is_nullable, 0, 'Field cannot be null' );
 352 is( $t2_f1->size, 10, 'Size is "10"' );
 353 is( $t2_f1->default_value, undef, 'Default value is undefined' );
 354 is( $t2_f1->is_primary_key, 1, 'Field is PK' );
 355
 356 my $t2_f2 = shift @t2_fields;
 357 is( $t2_f2->name, 'f_varchar', 'Second field is "f_varchar"' );
 358 is( $t2_f2->data_type, 'varchar', 'Field is an varchar' );
 359 is( $t2_f2->is_nullable, 1, 'Field can be null' );
 360 is( $t2_f2->size, 25, 'Size is "25"' );
 361 is( $t2_f2->default_value, undef, 'Default value is undefined' );
 362 is( $t2_f2->is_primary_key, 0, 'Field is not PK' );
 363
 364 my $t2_f3 = shift @t2_fields;
 365 is( $t2_f3->name, 'f_int', 'Third field is "f_int"' );
 366 is( $t2_f3->data_type, 'integer', 'Field is an integer' );
 367 is( $t2_f3->is_nullable, 1, 'Field can be null' );
 368 is( $t2_f3->size, 5, 'Size is "5"' );
 369 is( $t2_f3->default_value, undef, 'Default value is undefined' );
 370 is( $t2_f3->is_primary_key, 0, 'Field is not PK' );
 371
 372 my $t2_f4 = shift @t2_fields;
 373 is( $t2_f4->name, 'f_smallint', 'Fourth field is "f_smallint"' );
 374 is( $t2_f4->data_type, 'integer', 'Field is an integer' );
 375 is( $t2_f4->is_nullable, 1, 'Field can be null' );
 376 is( $t2_f4->size, 5, 'Size is "5"' );
 377 is( $t2_f4->default_value, 0, 'Default value is 0' );
 378 is( $t2_f4->is_primary_key, 0, 'Field is not PK' );
 379
 380
 381 my @t2_constraints = $t2->get_constraints;
 382 is( scalar @t2_constraints, 3, "Three constraints on table" );
 383
 384 my $t2_c1 = shift @t2_constraints;
 385 is( $t2_c1->type, NOT_NULL, "Constraint is NOT NULL" );
 386
 387 my $t2_c2 = shift @t2_constraints;
 388 is( $t2_c2->type, PRIMARY_KEY, "Constraint is a PK" );
 389
 390 my $t2_c3 = shift @t2_constraints;
 391 is( $t2_c3->type, CHECK_C, "Constraint is a 'CHECK'" );
 392
 393 # test temporary tables
 394 is( exists $schema->get_table('products_1')->extra()->{'temporary'}, "", "Table is NOT temporary");
 395 is( $schema->get_table('products_2')->extra('temporary'), 1,"Table is TEMP");
 396 is( $schema->get_table('products_3')->extra('temporary'), 1,"Table is TEMPORARY");
 397
 398 # test trigger
 399 my $trigger = $schema->get_trigger('test_trigger');
 400 is( $trigger->on_table, 'products_1', "Trigger is on correct table");
 401 is_deeply( scalar $trigger->database_events, [qw(insert update delete)], "Correct events for trigger");
 402
 403 is( $trigger->perform_action_when, 'before', "Correct time for trigger");
 404 is( $trigger->scope, 'row', "Correct scope for trigger");
 405 is( $trigger->action, 'EXECUTE PROCEDURE foo()', "Correct action for trigger");
 406
 407 # test index
 408 my @indices = $t1->get_indices;
 409 is(scalar @indices, 3, 'got three indexes');
 410
 411 my $t1_i1 = $indices[0];
 412 is( $t1_i1->name, 'test_index1', 'First index is "test_index1"' );
 413 is( join(',', $t1_i1->fields), 'f_varchar', 'Index is on field "f_varchar"' );
 414 is_deeply( [ $t1_i1->options ], [], 'Index is has no options' );
 415
 416 my $t1_i2 = $indices[1];
 417 is( $t1_i2->name, 'test_index2', 'Second index is "test_index2"' );
 418 is( join(',', $t1_i2->fields), 'f_char,f_bool', 'Index is on fields "f_char, f_bool"' );
 419 is_deeply( [ $t1_i2->options ], [ { using => 'hash' } ], 'Index is using hash method' );
 420
 421 my $t1_i3 = $indices[2];
 422 is( $t1_i3->name, 'test_index3', 'Third index is "test_index3"' );
 423 is( join(',', $t1_i3->fields), 'f_bigint,f_tz', 'Index is on fields "f_bigint, f_tz"' );
 424 is_deeply(
 425     [ $t1_i3->options ],
 426     [ { using => 'hash' }, { where => "f_bigint = '1' AND f_tz IS NULL" } ],
 427     'Index is using hash method and has predicate right'
 428 );
 429
 430 done_testing;