multi-dsn support for common tests

[dbsrgits/DBIx-Class-Schema-Loader.git] / t / 16mssql_common.t

diff --git a/t/16mssql_common.t b/t/16mssql_common.t
index 4282aae..ccce4a4 100644 (file)
--- a/t/16mssql_common.t
+++ b/t/16mssql_common.t
@@ -1,5 +1,7 @@
 use strict;
 use warnings;
+use Test::More;
+use Test::Exception;
 
 # use this if you keep a copy of DBD::Sybase linked to FreeTDS somewhere else
 BEGIN {
@@ -10,25 +12,107 @@ BEGIN {
 
 use lib qw(t/lib);
 use dbixcsl_common_tests;
-use dbixcsl_mssql_extra_tests;
 
-my $dsn      = $ENV{DBICTEST_MSSQL_DSN} || '';
-my $user     = $ENV{DBICTEST_MSSQL_USER} || '';
-my $password = $ENV{DBICTEST_MSSQL_PASS} || '';
+my $dbd_sybase_dsn      = $ENV{DBICTEST_MSSQL_DSN} || '';
+my $dbd_sybase_user     = $ENV{DBICTEST_MSSQL_USER} || '';
+my $dbd_sybase_password = $ENV{DBICTEST_MSSQL_PASS} || '';
+
+my $odbc_dsn      = $ENV{DBICTEST_MSSQL_ODBC_DSN} || '';
+my $odbc_user     = $ENV{DBICTEST_MSSQL_ODBC_USER} || '';
+my $odbc_password = $ENV{DBICTEST_MSSQL_ODBC_PASS} || '';
 
 my $tester = dbixcsl_common_tests->new(
     vendor      => 'mssql',
     auto_inc_pk => 'INTEGER IDENTITY NOT NULL PRIMARY KEY',
     default_function     => 'getdate()',
     default_function_def => 'DATETIME DEFAULT getdate()',
-    dsn         => $dsn,
-    user        => $user,
-    password    => $password,
-    extra       => dbixcsl_mssql_extra_tests->extra,
+    connect_info => [ ($dbd_sybase_dsn ? {
+            dsn         => $dbd_sybase_dsn,
+            user        => $dbd_sybase_user,
+            password    => $dbd_sybase_password,
+        } : ()),
+        ($odbc_dsn ? {
+            dsn         => $odbc_dsn,
+            user        => $odbc_user,
+            password    => $odbc_password,
+        } : ()),
+    ],
+    extra => {
+        create => [
+            q{
+                CREATE TABLE [mssql_loader_test1.dot] (
+                    id INT IDENTITY NOT NULL PRIMARY KEY,
+                    dat VARCHAR(8)
+                )
+            },
+            q{
+                CREATE TABLE mssql_loader_test3 (
+                    id INT IDENTITY NOT NULL PRIMARY KEY
+                )
+            },
+            q{
+                CREATE VIEW mssql_loader_test4 AS
+                SELECT * FROM mssql_loader_test3
+            },
+        ],
+        pre_drop_ddl => [
+            'CREATE TABLE mssql_loader_test3 (id INT IDENTITY NOT NULL PRIMARY KEY)',
+            'DROP VIEW mssql_loader_test4',
+        ],
+        drop   => [
+            '[mssql_loader_test1.dot]',
+            'mssql_loader_test3'
+        ],
+        count  => 8,
+        run    => sub {
+            my ($schema, $monikers, $classes) = @_;
+
+# Test that the table above (with '.' in name) gets loaded correctly.
+            ok((my $rs = eval {
+                $schema->resultset($monikers->{'[mssql_loader_test1.dot]'}) }),
+                'got a resultset for table with dot in name');
+
+            ok((my $from = eval { $rs->result_source->from }),
+                'got an $rsrc->from for table with dot in name');
+
+            is ref($from), 'SCALAR', '->table with dot in name is a scalar ref';
+
+            is eval { $$from }, "[mssql_loader_test1.dot]",
+                '->table with dot in name has correct name';
+
+# Test that identity columns do not have 'identity' in the data_type, and do
+# have is_auto_increment.
+            my $identity_col_info = $schema->resultset($monikers->{loader_test10})
+                ->result_source->column_info('id10');
+
+            is $identity_col_info->{data_type}, 'int',
+                q{'INT IDENTITY' column has data_type => 'int'};
+
+            is $identity_col_info->{is_auto_increment}, 1,
+                q{'INT IDENTITY' column has is_auto_increment => 1};
+
+# Test that a bad view (where underlying table is gone) is ignored.
+            my $dbh = $schema->storage->dbh;
+            $dbh->do("DROP TABLE mssql_loader_test3");
+
+            my @warnings;
+            {
+                local $SIG{__WARN__} = sub { push @warnings, $_[0] };
+                $schema->rescan;
+            }
+            ok ((grep /^Bad table or view 'mssql_loader_test4'/, @warnings),
+                'bad view ignored');
+
+            throws_ok {
+                $schema->resultset($monikers->{mssql_loader_test4})
+            } qr/Can't find source/,
+                'no source registered for bad view';
+        },
+    },
 );
 
-if( !$dsn || !$user ) {
-    $tester->skip_tests('You need to set the DBICTEST_MSSQL_DSN, _USER, and _PASS environment variables');
+if(not ($dbd_sybase_dsn || $odbc_dsn)) {
+    $tester->skip_tests('You need to set the DBICTEST_MSSQL_DSN, _USER and _PASS and/or the DBICTEST_MSSQL_ODBC_DSN, _USER and _PASS environment variables');
 }
 else {
     $tester->run_tests();