MAss diff changes imported from Ash's local diff-refactor branch
[dbsrgits/SQL-Translator.git] / t / 02mysql-parser.t
index 1839795..a756b9f 100644 (file)
@@ -10,7 +10,7 @@ use SQL::Translator::Schema::Constants;
 use Test::SQL::Translator qw(maybe_plan);
 
 BEGIN {
-    maybe_plan(180, "SQL::Translator::Parser::MySQL");
+    maybe_plan(235, "SQL::Translator::Parser::MySQL");
     SQL::Translator::Parser::MySQL->import('parse');
 }
 
@@ -18,11 +18,12 @@ BEGIN {
     my $tr = SQL::Translator->new;
     my $data = q|create table sessions (
         id char(32) not null default '0' primary key,
-        a_session text
+        a_session text,
+        ssn varchar(12) unique key,
+        age int key
     );|;
 
     my $val = parse($tr, $data);
-
     my $schema = $tr->schema;
     is( $schema->is_valid, 1, 'Schema is valid' );
     my @tables = $schema->get_tables;
@@ -31,7 +32,7 @@ BEGIN {
     is( $table->name, 'sessions', 'Found "sessions" table' );
 
     my @fields = $table->get_fields;
-    is( scalar @fields, 2, 'Right number of fields (2)' );
+    is( scalar @fields, 4, 'Right number of fields (4)' );
     my $f1 = shift @fields;
     my $f2 = shift @fields;
     is( $f1->name, 'id', 'First field name is "id"' );
@@ -49,13 +50,16 @@ BEGIN {
     is( $f2->is_primary_key, 0, 'Field is not PK' );
 
     my @indices = $table->get_indices;
-    is( scalar @indices, 0, 'Right number of indices (0)' );
+    is( scalar @indices, 1, 'Right number of indices (1)' );
 
     my @constraints = $table->get_constraints;
-    is( scalar @constraints, 1, 'Right number of constraints (1)' );
+    is( scalar @constraints, 2, 'Right number of constraints (2)' );
     my $c = shift @constraints;
     is( $c->type, PRIMARY_KEY, 'Constraint is a PK' );
     is( join(',', $c->fields), 'id', 'Constraint is on "id"' );
+    my $c2 = shift @constraints;
+    is( $c2->type, UNIQUE, 'Constraint is UNIQUE' );
+    is( join(',', $c2->fields), 'ssn', 'Constraint is on "ssn"' );
 }
 
 {
@@ -217,7 +221,7 @@ BEGIN {
         q[
             CREATE TABLE orders (
               order_id                  integer NOT NULL auto_increment,
-              member_id                 varchar(255),
+              member_id                 varchar(255) comment 'fk to member',
               billing_address_id        int,
               shipping_address_id       int,
               credit_card_id            int,
@@ -234,7 +238,7 @@ BEGIN {
               FOREIGN KEY (status)              REFERENCES order_status(id) MATCH FULL ON DELETE CASCADE ON UPDATE CASCADE,
               FOREIGN KEY (billing_address_id)  REFERENCES address(address_id),
               FOREIGN KEY (shipping_address_id) REFERENCES address(address_id)
-            ) TYPE=INNODB;
+            ) TYPE=INNODB COMMENT = 'orders table comment';
 
             CREATE TABLE address (
               address_id                int NOT NULL auto_increment,
@@ -257,6 +261,7 @@ BEGIN {
 
     my $t1  = shift @tables;
     is( $t1->name, 'orders', 'Found "orders" table' );
+    is( $t1->comments, 'orders table comment', 'Table comment OK' );
 
     my @fields = $t1->get_fields;
     is( scalar @fields, 10, 'Right number of fields (10)' );
@@ -275,6 +280,7 @@ BEGIN {
     is( $f2->data_type, 'varchar', 'Type is "varchar"' );
     is( $f2->size, 255, 'Size is "255"' );
     is( $f2->is_nullable, 1, 'Field can be null' );
+    is( $f2->comments, 'fk to member', 'Field comment OK' );
     is( $f2->default_value, undef, 'Default value is undefined' );
 
     my $f3 = shift @fields;
@@ -389,3 +395,232 @@ BEGIN {
     my @t2_fields = $t2->get_fields;
     is( scalar @t2_fields, 8, 'Right number of fields (8)' );
 }
+
+# djh Tests for:
+#    USE database ;
+#    ALTER TABLE ADD FOREIGN KEY
+#    trailing comma on last create definition
+#    Ignoring INSERT statements
+#
+{
+    my $tr = SQL::Translator->new;
+    my $data = parse($tr, 
+        q[
+            USE database_name;
+
+            CREATE TABLE one (
+              id                     integer NOT NULL auto_increment,
+              two_id                 integer NOT NULL auto_increment,
+              some_data              text,
+              PRIMARY KEY (id),
+              INDEX (two_id),
+            ) TYPE=INNODB;
+
+            CREATE TABLE two (
+              id                     int NOT NULL auto_increment,
+              one_id                 int NOT NULL auto_increment,
+              some_data              text,
+              PRIMARY KEY (id),
+              INDEX (one_id),
+              FOREIGN KEY (one_id) REFERENCES one (id),
+            ) TYPE=INNODB;
+
+            ALTER TABLE one ADD FOREIGN KEY (two_id) REFERENCES two (id);
+
+            INSERT absolutely *#! any old $£ ? rubbish, even "quoted; semi-what""sits";
+        ]
+    ) or die $tr->error;
+
+    my $schema = $tr->schema;
+    is( $schema->is_valid, 1, 'Schema is valid' );
+    my $db_name = $schema->name;
+    is( $db_name, 'database_name', 'Database name extracted from USE' );
+    my @tables = $schema->get_tables;
+    is( scalar @tables, 2, 'Right number of tables (2)' );
+    my $table1 = shift @tables;
+    is( $table1->name, 'one', 'Found "one" table' );
+    my $table2 = shift @tables;
+    is( $table2->name, 'two', 'Found "two" table' );
+
+    my @constraints = $table1->get_constraints;
+    is(scalar @constraints, 2, 'Right number of constraints (2) on table one');
+
+    my $t1c1 = shift @constraints;
+    is( $t1c1->type, PRIMARY_KEY, 'Constraint is a PK' );
+    is( join(',', $t1c1->fields), 'id', 'Constraint is on "id"' );
+
+    my $t1c2 = shift @constraints;
+    is( $t1c2->type, FOREIGN_KEY, 'Constraint is a FK' );
+    is( join(',', $t1c2->fields), 'two_id', 'Constraint is on "two_id"' );
+    is( $t1c2->reference_table, 'two', 'To table "two"' );
+    is( join(',', $t1c2->reference_fields), 'id', 'To field "id"' );
+
+    @constraints = $table2->get_constraints;
+    is(scalar @constraints, 2, 'Right number of constraints (2) on table two');
+
+    my $t2c1 = shift @constraints;
+    is( $t2c1->type, PRIMARY_KEY, 'Constraint is a PK' );
+    is( join(',', $t2c1->fields), 'id', 'Constraint is on "id"' );
+
+    my $t2c2 = shift @constraints;
+    is( $t2c2->type, FOREIGN_KEY, 'Constraint is a FK' );
+    is( join(',', $t2c2->fields), 'one_id', 'Constraint is on "one_id"' );
+    is( $t2c2->reference_table, 'one', 'To table "one"' );
+    is( join(',', $t2c2->reference_fields), 'id', 'To field "id"' );
+}
+
+# cch Tests for:
+#    comments like: /*!40101 SET SQL_MODE=@OLD_SQL_MODE */;
+#    char fields with character set and collate qualifiers
+#    timestamp fields with on update qualifier
+#    charset table option
+#
+{
+    my $tr = SQL::Translator->new(parser_args => {mysql_parser_version => 50003});
+    my $data = parse($tr, 
+        q[
+               DELIMITER ;;
+            /*!40101 SET SQL_MODE=@OLD_SQL_MODE */;;
+                       /*!50003 CREATE */ /*!50017 DEFINER=`cmdomain`@`localhost` */
+                       /*!50003 TRIGGER `acl_entry_insert` BEFORE INSERT ON `acl_entry`
+                               FOR EACH ROW SET NEW.dateCreated = CONVERT_TZ(SYSDATE(),'SYSTEM','+0:00'),
+                               NEW.dateModified = CONVERT_TZ(SYSDATE(),'SYSTEM','+0:00') */;;
+
+                       DELIMITER ;
+            CREATE TABLE one (
+              `op` varchar(255) character set latin1 collate latin1_bin default NULL,
+              `last_modified` timestamp NOT NULL default CURRENT_TIMESTAMP on update CURRENT_TIMESTAMP,
+            ) TYPE=INNODB DEFAULT CHARSET=latin1;
+
+                       /*!50001 CREATE ALGORITHM=UNDEFINED */
+                       /*!50013 DEFINER=`cmdomain`@`localhost` SQL SECURITY DEFINER */
+                       /*! VIEW `vs_asset` AS 
+                               select `a`.`asset_id` AS `asset_id`,`a`.`fq_name` AS `fq_name`,
+                               `cfgmgmt_mig`.`ap_extract_folder`(`a`.`fq_name`) AS `folder_name`,
+                               `cfgmgmt_mig`.`ap_extract_asset`(`a`.`fq_name`) AS `asset_name`,
+                               `a`.`annotation` AS `annotation`,`a`.`asset_type` AS `asset_type`,
+                               `a`.`foreign_asset_id` AS `foreign_asset_id`,
+                               `a`.`foreign_asset_id2` AS `foreign_asset_id2`,`a`.`dateCreated` AS `date_created`,
+                               `a`.`dateModified` AS `date_modified`,`a`.`container_id` AS `container_id`,
+                               `a`.`creator_id` AS `creator_id`,`a`.`modifier_id` AS `modifier_id`,
+                               `m`.`user_id` AS `user_access` 
+                               from (`asset` `a` join `M_ACCESS_CONTROL` `m` on((`a`.`acl_id` = `m`.`acl_id`))) */;
+                       DELIMITER ;;
+                       /*!50003 CREATE*/ /*!50020 DEFINER=`cmdomain`@`localhost`*/ /*!50003 FUNCTION `ap_from_millitime_nullable`( millis_since_1970 BIGINT ) RETURNS timestamp
+                       DETERMINISTIC
+                               BEGIN
+                               DECLARE rval TIMESTAMP;
+                                   IF ( millis_since_1970 = 0 )
+                                   THEN
+                                       SET rval = NULL;
+                                   ELSE
+                                       SET rval = FROM_UNIXTIME( millis_since_1970 / 1000 );
+                                   END IF;
+                                   RETURN rval;
+                               END */;;
+                       /*!50003 CREATE*/ /*!50020 DEFINER=`cmdomain`@`localhost`*/ /*!50003 PROCEDURE `sp_update_security_acl`(IN t_acl_id INTEGER)
+                       BEGIN
+                               DECLARE hasMoreRows BOOL DEFAULT TRUE;
+                               DECLARE t_group_id INT;
+                               DECLARE t_user_id INT ;
+                               DECLARE t_user_name VARCHAR (512) ;
+                               DECLARE t_message VARCHAR (512) ;
+
+                               DROP TABLE IF EXISTS group_acl;
+                               DROP TABLE IF EXISTS user_group;
+                               DELETE FROM M_ACCESS_CONTROL WHERE acl_id = t_acl_id;
+
+                               CREATE TEMPORARY TABLE group_acl SELECT DISTINCT p.id group_id, d.acl_id acl_id
+                                       FROM  asset d, acl_entry e, alterpoint_principal p
+                                       WHERE d.acl_id = e.acl
+                                       AND p.id = e.principal AND d.acl_id = t_acl_id;
+
+                               CREATE TEMPORARY TABLE user_group  SELECT a.id user_id, a.name user_name, c.id group_id
+                                       FROM alterpoint_principal a, groups_for_user b, alterpoint_principal c
+                                       WHERE a.id = b.user_ref AND b.elt = c.id;
+
+                               INSERT INTO M_ACCESS_CONTROL SELECT DISTINCT group_acl.group_id, group_acl.acl_id, user_group.user_id, user_group.user_name
+                                       FROM group_acl, user_group
+                                       WHERE group_acl.group_id = user_group.group_id ;
+                       END */;;
+        ]
+    ) or die $tr->error;
+
+    my $schema = $tr->schema;
+    is( $schema->is_valid, 1, 'Schema is valid' );
+    my @tables = $schema->get_tables;
+    is( scalar @tables, 1, 'Right number of tables (1)' );
+    my $table1 = shift @tables;
+    is( $table1->name, 'one', 'Found "one" table' );
+
+    my @fields = $table1->get_fields;
+    is(scalar @fields, 2, 'Right number of fields (2) on table one');
+    my $tableTypeFound = 0;
+    my $charsetFound = 0;
+       for my $t1_option_ref ( $table1->options ) {
+               my($key, $value) = %{$t1_option_ref};
+               if ( $key eq 'TYPE' ) {
+                       is($value, 'INNODB', 'Table has right table type option' );
+                       $tableTypeFound = 1;
+               } elsif ( $key eq 'CHARACTER SET' ) {
+                       is($value, 'latin1', 'Table has right character set option' );
+                       $charsetFound = 1;
+               }
+       }
+       fail('Table did not have a type option') unless $tableTypeFound;
+       fail('Table did not have a character set option') unless $charsetFound;
+
+    my $t1f1 = shift @fields;
+    is( $t1f1->data_type, 'varchar', 'Field is a varchar' );
+    is( $t1f1->size, 255, 'Field is right size' );
+    is( $t1f1->extra('character set'), 'latin1', 'Field has right character set qualifier' );
+    is( $t1f1->extra('collate'), 'latin1_bin', 'Field has right collate qualifier' );
+    is( $t1f1->default_value, 'NULL', 'Field has right default value' );
+
+    my $t1f2 = shift @fields;
+    is( $t1f2->data_type, 'timestamp', 'Field is a timestamp' );
+    ok( !$t1f2->is_nullable, 'Field is not nullable' );
+    is( $t1f2->default_value, 'CURRENT_TIMESTAMP', 'Field has right default value' );
+    is( $t1f2->extra('on update'), 'CURRENT_TIMESTAMP', 'Field has right on update qualifier' );
+    
+    my @views = $schema->get_views;
+    is( scalar @views, 1, 'Right number of views (1)' );
+    my $view1 = shift @views;
+    is( $view1->name, 'vs_asset', 'Found "vs_asset" view' );
+       like($view1->sql, qr/ALGORITHM=UNDEFINED/, "Detected algorithm");
+       like($view1->sql, qr/vs_asset/, "Detected view vs_asset");
+       unlike($view1->sql, qr/cfgmgmt_mig/, "Did not detect cfgmgmt_mig");
+    
+    my @procs = $schema->get_procedures;
+    is( scalar @procs, 2, 'Right number of procedures (2)' );
+    my $proc1 = shift @procs;
+    is( $proc1->name, 'ap_from_millitime_nullable', 'Found "ap_from_millitime_nullable" procedure' );
+       like($proc1->sql, qr/CREATE FUNCTION ap_from_millitime_nullable/, "Detected procedure ap_from_millitime_nullable");
+    my $proc2 = shift @procs;
+    is( $proc2->name, 'sp_update_security_acl', 'Found "sp_update_security_acl" procedure' );
+       like($proc2->sql, qr/CREATE PROCEDURE sp_update_security_acl/, "Detected procedure sp_update_security_acl");
+}
+
+{
+    my $tr = SQL::Translator->new({parser_args => { no_default_sizes => 1 } });
+    my $data = q|create table sessions (
+        age int
+    );|;
+
+    my $val = parse($tr, $data);
+    my $schema = $tr->schema;
+    is( $schema->is_valid, 1, 'Schema is valid' );
+    my @tables = $schema->get_tables;
+    is( scalar @tables, 1, 'Right number of tables (1)' );
+    my $table  = shift @tables;
+    is( $table->name, 'sessions', 'Found "sessions" table' );
+
+    my @fields = $table->get_fields;
+    is( scalar @fields, 1, 'Right number of fields (1)' );
+    my $f1 = shift @fields;
+    is( $f1->name, 'age', 'First field name is "id"' );
+    is( $f1->data_type, 'int', 'Type is "int"' );
+    is( $f1->size, 0, 'No default size' );
+
+}
+