FK ON clause introspection for MSSQL
[dbsrgits/DBIx-Class-Schema-Loader.git] / t / 10_07mssql_common.t
index df6ba37..6382fc6 100644 (file)
@@ -7,6 +7,7 @@ use Try::Tiny;
 use File::Path 'rmtree';
 use DBIx::Class::Schema::Loader 'make_schema_at';
 use namespace::clean;
+use Scope::Guard ();
 
 # use this if you keep a copy of DBD::Sybase linked to FreeTDS somewhere else
 BEGIN {
@@ -22,8 +23,8 @@ use dbixcsl_test_dir '$tdir';
 
 use constant EXTRA_DUMP_DIR => "$tdir/mssql_extra_dump";
 
-# for cleanup in END for extra tests
-my ($schema, $schemas_created, $databases_created);
+# for extra tests cleanup
+my $schema;
 
 my ($dsns, $common_version);
 
@@ -93,6 +94,7 @@ my $tester = dbixcsl_common_tests->new(
     preserve_case_mode_is_exclusive => 1,
     quote_char => [ qw/[ ]/ ],
     basic_date_datatype => ($common_version >= 10) ? 'DATE' : 'SMALLDATETIME',
+    default_on_clause => 'NO ACTION',
     data_types => {
         # http://msdn.microsoft.com/en-us/library/ms187752.aspx
 
@@ -199,6 +201,20 @@ my $tester = dbixcsl_common_tests->new(
                     [Five_Id] INT REFERENCES [MSSQL_Loader_Test5] ([Id])
                 )
             },
+            # 8 through 12 are used for the multi-schema tests and 13 through 16 are used for multi-db tests
+            q{
+                create table mssql_loader_test17 (
+                    id int identity primary key
+                )
+            },
+            q{
+                create table mssql_loader_test18 (
+                    id int identity primary key,
+                    seventeen_id int,
+                    foreign key (seventeen_id) references mssql_loader_test17(id)
+                        on delete set default on update set null
+                )
+            },
         ],
         pre_drop_ddl => [
             'CREATE TABLE mssql_loader_test3 (id INT IDENTITY NOT NULL PRIMARY KEY)',
@@ -209,8 +225,10 @@ my $tester = dbixcsl_common_tests->new(
             'mssql_loader_test3',
             'MSSQL_Loader_Test6',
             'MSSQL_Loader_Test5',
+            'mssql_loader_test17',
+            'mssql_loader_test18',
         ],
-        count  => 10 + 28 * 2 + 24,
+        count  => 14 + 30 * 2 + 26 * 2, # extra + multi-schema + mutli-db
         run    => sub {
             my ($monikers, $classes, $self);
             ($schema, $monikers, $classes, $self) = @_;
@@ -285,15 +303,27 @@ my $tester = dbixcsl_common_tests->new(
             } qr/Can't find source/,
                 'no source registered for bad view';
 
+            # test on delete/update fk clause introspection
+            ok ((my $rel_info = $schema->source('MssqlLoaderTest18')->relationship_info('seventeen')),
+                'got rel info');
+
+            is $rel_info->{attrs}{on_delete}, 'SET DEFAULT',
+                'ON DELETE clause introspected correctly';
+
+            is $rel_info->{attrs}{on_update}, 'SET NULL',
+                'ON UPDATE clause introspected correctly';
+
+            is $rel_info->{attrs}{is_deferrable}, 1,
+                'is_deferrable defaults to 1';
+
             SKIP: {
                 my $dbh = $schema->storage->dbh;
 
                 try {
-                    $dbh->do('CREATE SCHEMA "dbicsl-test"');
+                    $dbh->do('CREATE SCHEMA [dbicsl-test]');
                 }
                 catch {
-                    $schemas_created = 0;
-                    skip "no CREATE SCHEMA privileges", 28 * 2;
+                    skip "no CREATE SCHEMA privileges", 30 * 2;
                 };
 
                 $dbh->do(<<"EOF");
@@ -306,12 +336,22 @@ EOF
                     CREATE TABLE [dbicsl-test].mssql_loader_test9 (
                         id INT IDENTITY PRIMARY KEY,
                         value VARCHAR(100),
-                        eight_id INTEGER NOT NULL UNIQUE,
+                        eight_id INTEGER NOT NULL,
+                        CONSTRAINT loader_test9_uniq UNIQUE (eight_id),
                         FOREIGN KEY (eight_id) REFERENCES [dbicsl-test].mssql_loader_test8 (id)
                     )
 EOF
                 $dbh->do('CREATE SCHEMA [dbicsl.test]');
                 $dbh->do(<<"EOF");
+                    CREATE TABLE [dbicsl.test].mssql_loader_test9 (
+                        pk INT IDENTITY PRIMARY KEY,
+                        value VARCHAR(100),
+                        eight_id INTEGER NOT NULL,
+                        CONSTRAINT loader_test9_uniq UNIQUE (eight_id),
+                        FOREIGN KEY (eight_id) REFERENCES [dbicsl-test].mssql_loader_test8 (id)
+                    )
+EOF
+                $dbh->do(<<"EOF");
                     CREATE TABLE [dbicsl.test].mssql_loader_test10 (
                         id INT IDENTITY PRIMARY KEY,
                         value VARCHAR(100),
@@ -336,7 +376,7 @@ EOF
                     )
 EOF
 
-                $schemas_created = 1;
+                my $guard = Scope::Guard->new(\&cleanup_schemas);
 
                 foreach my $db_schema (['dbicsl-test', 'dbicsl.test'], '%') {
                     lives_and {
@@ -390,7 +430,7 @@ EOF
                         ok $row = $rs->create({ value => 'foo' });
                     } 'executed SQL on table in schema name with dash';
 
-                    $rel_info = try { $rsrc->relationship_info('mssql_loader_test9') };
+                    $rel_info = try { $rsrc->relationship_info('dbicsl_dash_test_mssql_loader_test9') };
 
                     is_deeply $rel_info->{cond}, {
                         'foreign.eight_id' => 'self.id'
@@ -403,7 +443,7 @@ EOF
                         'relationship in schema name with dash';
 
                     lives_and {
-                        ok $rsrc = $test_schema->source('MssqlLoaderTest9');
+                        ok $rsrc = $test_schema->source('DbicslDashTestMssqlLoaderTest9');
                     } 'got source for table in schema name with dash';
 
                     %uniqs = try { $rsrc->unique_constraints };
@@ -411,6 +451,11 @@ EOF
                     is keys %uniqs, 2,
                         'got unique and primary constraint in schema name with dash';
 
+                    delete $uniqs{primary};
+
+                    is_deeply ((values %uniqs)[0], ['eight_id'],
+                        'correct unique constraint in schema name with dash');
+
                     lives_and {
                         ok $rsrc = $test_schema->source('MssqlLoaderTest10');
                     } 'got source for table in schema name with dot';
@@ -453,6 +498,11 @@ EOF
                     is keys %uniqs, 2,
                         'got unique and primary constraint in schema name with dot';
 
+                    delete $uniqs{primary};
+
+                    is_deeply ((values %uniqs)[0], ['ten_id'],
+                        'correct unique constraint in schema name with dot');
+
                     lives_and {
                         ok $test_schema->source('MssqlLoaderTest10')
                             ->has_relationship('mssql_loader_test8');
@@ -476,6 +526,12 @@ EOF
             }
 
             SKIP: {
+                # for ADO
+                my $warn_handler = $SIG{__WARN__} || sub { warn @_ };
+                local $SIG{__WARN__} = sub {
+                    $warn_handler->(@_) unless $_[0] =~ /Changed database context/;
+                };
+
                 my $dbh = $schema->storage->dbh;
 
                 try {
@@ -483,7 +539,8 @@ EOF
                     $dbh->do('CREATE DATABASE dbicsl_test1');
                 }
                 catch {
-                    skip "no CREATE DATABASE privileges", 24;
+                    diag "no CREATE DATABASE privileges: '$_'";
+                    skip "no CREATE DATABASE privileges", 26 * 2;
                 };
 
                 $dbh->do('CREATE DATABASE dbicsl_test2');
@@ -500,13 +557,22 @@ EOF
                     CREATE TABLE mssql_loader_test14 (
                         id INT IDENTITY PRIMARY KEY,
                         value VARCHAR(100),
-                        thirteen_id INTEGER UNIQUE REFERENCES mssql_loader_test13 (id)
+                        thirteen_id INTEGER REFERENCES mssql_loader_test13 (id),
+                        CONSTRAINT loader_test14_uniq UNIQUE (thirteen_id)
                     )
 EOF
 
-                $dbh->do('USE master');
                 $dbh->do('USE dbicsl_test2');
 
+                $dbh->do(<<'EOF');
+                    CREATE TABLE mssql_loader_test14 (
+                        pk INT IDENTITY PRIMARY KEY,
+                        value VARCHAR(100),
+                        thirteen_id INTEGER,
+                        CONSTRAINT loader_test14_uniq UNIQUE (thirteen_id)
+                    )
+EOF
+
                 $dbh->do(<<"EOF");
                     CREATE TABLE mssql_loader_test15 (
                         id INT IDENTITY PRIMARY KEY,
@@ -521,121 +587,133 @@ EOF
                     )
 EOF
 
-                $databases_created = 1;
+                my $guard = Scope::Guard->new(\&cleanup_databases);
+
+                foreach my $db_schema ({ dbicsl_test1 => '%', dbicsl_test2 => '%' }, { '%' => '%' }) {
+                    lives_and {
+                        my @warns;
+                        local $SIG{__WARN__} = sub {
+                            push @warns, $_[0] unless $_[0] =~ /\bcollides\b/;
+                        };
+     
+                        make_schema_at(
+                            'MSSQLMultiDatabase',
+                            {
+                                naming => 'current',
+                                db_schema => $db_schema,
+                                dump_directory => EXTRA_DUMP_DIR,
+                                quiet => 1,
+                            },
+                            $connect_info,
+                        );
+
+                        diag join "\n", @warns if @warns;
+
+                        is @warns, 0;
+                    } "dumped schema for databases 'dbicsl_test1' and 'dbicsl_test2' with no warnings";
 
-                lives_and {
-                    my @warns;
-                    local $SIG{__WARN__} = sub {
-                        push @warns, $_[0] unless $_[0] =~ /\bcollides\b/;
-                    };
-                    make_schema_at(
-                        'MSSQLMultiDatabase',
-                        {
-                            naming => 'current',
-                            db_schema => { '%' => '%' },
-                            dump_directory => EXTRA_DUMP_DIR,
-                            quiet => 1,
-                        },
-                        $connect_info,
-                    );
+                    my $test_schema;
+
+                    lives_and {
+                        ok $test_schema = MSSQLMultiDatabase->connect(@$connect_info);
+                    } 'connected test schema';
 
-                    diag join "\n", @warns if @warns;
+                    my ($rsrc, $rs, $row, $rel_info, %uniqs);
 
-                    is @warns, 0;
-                } 'dumped schema for all databases with no warnings';
+                    lives_and {
+                        ok $rsrc = $test_schema->source('MssqlLoaderTest13');
+                    } 'got source for table in database one';
 
-                my $test_schema;
+                    is try { $rsrc->column_info('id')->{is_auto_increment} }, 1,
+                        'column in database one';
 
-                lives_and {
-                    ok $test_schema = MSSQLMultiDatabase->connect(@$connect_info);
-                } 'connected test schema';
+                    is try { $rsrc->column_info('value')->{data_type} }, 'varchar',
+                        'column in database one';
 
-                my ($rsrc, $rs, $row, $rel_info, %uniqs);
+                    is try { $rsrc->column_info('value')->{size} }, 100,
+                        'column in database one';
 
-                lives_and {
-                    ok $rsrc = $test_schema->source('MssqlLoaderTest13');
-                } 'got source for table in database one';
+                    lives_and {
+                        ok $rs = $test_schema->resultset('MssqlLoaderTest13');
+                    } 'got resultset for table in database one';
 
-                is try { $rsrc->column_info('id')->{is_auto_increment} }, 1,
-                    'column in database one';
+                    lives_and {
+                        ok $row = $rs->create({ value => 'foo' });
+                    } 'executed SQL on table in database one';
 
-                is try { $rsrc->column_info('value')->{data_type} }, 'varchar',
-                    'column in database one';
+                    $rel_info = try { $rsrc->relationship_info('mssql_loader_test14') };
 
-                is try { $rsrc->column_info('value')->{size} }, 100,
-                    'column in database one';
+                    is_deeply $rel_info->{cond}, {
+                        'foreign.thirteen_id' => 'self.id'
+                    }, 'relationship in database one';
 
-                lives_and {
-                    ok $rs = $test_schema->resultset('MssqlLoaderTest13');
-                } 'got resultset for table in database one';
+                    is $rel_info->{attrs}{accessor}, 'single',
+                        'relationship in database one';
 
-                lives_and {
-                    ok $row = $rs->create({ value => 'foo' });
-                } 'executed SQL on table in database one';
+                    is $rel_info->{attrs}{join_type}, 'LEFT',
+                        'relationship in database one';
 
-                $rel_info = try { $rsrc->relationship_info('mssql_loader_test14') };
+                    lives_and {
+                        ok $rsrc = $test_schema->source('DbicslTest1MssqlLoaderTest14');
+                    } 'got source for table in database one';
 
-                is_deeply $rel_info->{cond}, {
-                    'foreign.thirteen_id' => 'self.id'
-                }, 'relationship in database one';
+                    %uniqs = try { $rsrc->unique_constraints };
 
-                is $rel_info->{attrs}{accessor}, 'single',
-                    'relationship in database one';
+                    is keys %uniqs, 2,
+                        'got unique and primary constraint in database one';
 
-                is $rel_info->{attrs}{join_type}, 'LEFT',
-                    'relationship in database one';
+                    delete $uniqs{primary};
 
-                lives_and {
-                    ok $rsrc = $test_schema->source('MssqlLoaderTest14');
-                } 'got source for table in database one';
+                    is_deeply ((values %uniqs)[0], ['thirteen_id'],
+                        'correct unique constraint in database one');
 
-                %uniqs = try { $rsrc->unique_constraints };
+                    lives_and {
+                        ok $rsrc = $test_schema->source('MssqlLoaderTest15');
+                    } 'got source for table in database two';
 
-                is keys %uniqs, 2,
-                    'got unique and primary constraint in database one';
+                    is try { $rsrc->column_info('id')->{is_auto_increment} }, 1,
+                        'column in database two introspected correctly';
 
-                lives_and {
-                    ok $rsrc = $test_schema->source('MssqlLoaderTest15');
-                } 'got source for table in database two';
+                    is try { $rsrc->column_info('value')->{data_type} }, 'varchar',
+                        'column in database two introspected correctly';
 
-                is try { $rsrc->column_info('id')->{is_auto_increment} }, 1,
-                    'column in database two introspected correctly';
+                    is try { $rsrc->column_info('value')->{size} }, 100,
+                        'column in database two introspected correctly';
 
-                is try { $rsrc->column_info('value')->{data_type} }, 'varchar',
-                    'column in database two introspected correctly';
+                    lives_and {
+                        ok $rs = $test_schema->resultset('MssqlLoaderTest15');
+                    } 'got resultset for table in database two';
 
-                is try { $rsrc->column_info('value')->{size} }, 100,
-                    'column in database two introspected correctly';
+                    lives_and {
+                        ok $row = $rs->create({ value => 'foo' });
+                    } 'executed SQL on table in database two';
 
-                lives_and {
-                    ok $rs = $test_schema->resultset('MssqlLoaderTest15');
-                } 'got resultset for table in database two';
+                    $rel_info = try { $rsrc->relationship_info('mssql_loader_test16') };
 
-                lives_and {
-                    ok $row = $rs->create({ value => 'foo' });
-                } 'executed SQL on table in database two';
+                    is_deeply $rel_info->{cond}, {
+                        'foreign.fifteen_id' => 'self.id'
+                    }, 'relationship in database two';
 
-                $rel_info = try { $rsrc->relationship_info('mssql_loader_test16') };
+                    is $rel_info->{attrs}{accessor}, 'single',
+                        'relationship in database two';
 
-                is_deeply $rel_info->{cond}, {
-                    'foreign.fifteen_id' => 'self.id'
-                }, 'relationship in database two';
+                    is $rel_info->{attrs}{join_type}, 'LEFT',
+                        'relationship in database two';
 
-                is $rel_info->{attrs}{accessor}, 'single',
-                    'relationship in database two';
+                    lives_and {
+                        ok $rsrc = $test_schema->source('MssqlLoaderTest16');
+                    } 'got source for table in database two';
 
-                is $rel_info->{attrs}{join_type}, 'LEFT',
-                    'relationship in database two';
+                    %uniqs = try { $rsrc->unique_constraints };
 
-                lives_and {
-                    ok $rsrc = $test_schema->source('MssqlLoaderTest16');
-                } 'got source for table in database two';
+                    is keys %uniqs, 2,
+                        'got unique and primary constraint in database two';
 
-                %uniqs = try { $rsrc->unique_constraints };
+                    delete $uniqs{primary};
 
-                is keys %uniqs, 2,
-                    'got unique and primary constraint in database two';
+                    is_deeply ((values %uniqs)[0], ['fifteen_id'],
+                        'correct unique constraint in database two');
+                }
             }
         },
     },
@@ -643,76 +721,86 @@ EOF
 
 $tester->run_tests();
 
-END {
-    if (not $ENV{SCHEMA_LOADER_TESTS_NOCLEANUP}) {
-        if ($schema) {
-            # switch back to default database
-            $schema->storage->disconnect;
-            my $dbh = $schema->storage->dbh;
+sub cleanup_schemas {
+    return if $ENV{SCHEMA_LOADER_TESTS_NOCLEANUP};
+
+    # switch back to default database
+    $schema->storage->disconnect;
+    my $dbh = $schema->storage->dbh;
+
+    foreach my $table ('[dbicsl-test].mssql_loader_test12',
+                       '[dbicsl.test].mssql_loader_test11',
+                       '[dbicsl.test].mssql_loader_test10',
+                       '[dbicsl.test].mssql_loader_test9',
+                       '[dbicsl-test].mssql_loader_test9',
+                       '[dbicsl-test].mssql_loader_test8') {
+        try {
+            $dbh->do("DROP TABLE $table");
+        }
+        catch {
+            diag "Error dropping table: $_";
+        };
+    }
 
-            if ($schemas_created) {
-                foreach my $table ('[dbicsl-test].mssql_loader_test12',
-                                   '[dbicsl.test].mssql_loader_test11',
-                                   '[dbicsl.test].mssql_loader_test10',
-                                   '[dbicsl-test].mssql_loader_test9',
-                                   '[dbicsl-test].mssql_loader_test8') {
-                    try {
-                        $dbh->do("DROP TABLE $table");
-                    }
-                    catch {
-                        diag "Error dropping table: $_";
-                    };
-                }
+    foreach my $db_schema (qw/dbicsl-test dbicsl.test/) {
+        try {
+            $dbh->do(qq{DROP SCHEMA [$db_schema]});
+        }
+        catch {
+            diag "Error dropping test schema $db_schema: $_";
+        };
+    }
 
-                foreach my $db_schema (qw/dbicsl-test dbicsl.test/) {
-                    try {
-                        $dbh->do(qq{DROP SCHEMA [$db_schema]});
-                    }
-                    catch {
-                        diag "Error dropping test schema $db_schema: $_";
-                    };
-                }
-            }
+    rmtree EXTRA_DUMP_DIR;
+}
 
-            if ($databases_created) {
-                $dbh->do('USE dbicsl_test1');
+sub cleanup_databases {
+    return if $ENV{SCHEMA_LOADER_TESTS_NOCLEANUP};
 
-                foreach my $table ('mssql_loader_test14',
-                                   'mssql_loader_test13') {
-                    try {
-                        $dbh->do("DROP TABLE $table");
-                    }
-                    catch {
-                        diag "Error dropping table: $_";
-                    };
-                }
+    # for ADO
+    my $warn_handler = $SIG{__WARN__} || sub { warn @_ };
+    local $SIG{__WARN__} = sub {
+        $warn_handler->(@_) unless $_[0] =~ /Changed database context/;
+    };
 
-                $dbh->do('USE dbicsl_test2');
+    my $dbh = $schema->storage->dbh;
 
-                foreach my $table ('mssql_loader_test16',
-                                   'mssql_loader_test15') {
-                    try {
-                        $dbh->do("DROP TABLE $table");
-                    }
-                    catch {
-                        diag "Error dropping table: $_";
-                    };
-                }
+    $dbh->do('USE dbicsl_test1');
 
-                $dbh->do('USE master');
+    foreach my $table ('mssql_loader_test14',
+                       'mssql_loader_test13') {
+        try {
+            $dbh->do("DROP TABLE $table");
+        }
+        catch {
+            diag "Error dropping table: $_";
+        };
+    }
 
-                foreach my $database (qw/dbicsl_test1 dbicsl_test2/) {
-                    try {
-                        $dbh->do(qq{DROP DATABASE $database});
-                    }
-                    catch {
-                        diag "Error dropping test database '$database': $_";
-                    };
-                }
-            }
+    $dbh->do('USE dbicsl_test2');
+
+    foreach my $table ('mssql_loader_test16',
+                       'mssql_loader_test15',
+                       'mssql_loader_test14') {
+        try {
+            $dbh->do("DROP TABLE $table");
+        }
+        catch {
+            diag "Error dropping table: $_";
+        };
+    }
+
+    $dbh->do('USE master');
 
-            rmtree EXTRA_DUMP_DIR;
+    foreach my $database (qw/dbicsl_test1 dbicsl_test2/) {
+        try {
+            $dbh->do(qq{DROP DATABASE $database});
         }
+        catch {
+            diag "Error dropping test database '$database': $_";
+        };
     }
+
+    rmtree EXTRA_DUMP_DIR;
 }
 # vim:et sts=4 sw=4 tw=0: