sub run_tests {
my $schema = shift;

my ($dsn, $user, $pass) = @ENV{map { "DBICTEST_MSSQL_${_}" } qw/DSN USER PASS/};

#warn "$dsn $user $pass";

plan skip_all, 'Set $ENV{DBICTEST_MSSQL_DSN}, _USER and _PASS to run this test'
  unless ($dsn);

plan tests => 4;

$schema->compose_connection( 'MSSQLTest' => $dsn, $user, $pass );

my $dbh = MSSQLTest->schema->storage->dbh;

$dbh->do("IF OBJECT_ID('artist', 'U') IS NOT NULL
    DROP TABLE artist");

$dbh->do("CREATE TABLE artist (artistid INT IDENTITY PRIMARY KEY, name VARCHAR(255));");

MSSQLTest::Artist->load_components('PK::Auto::MSSQL');

# Test PK
my $new = MSSQLTest::Artist->create( { name => 'foo' } );
ok($new->artistid, "Auto-PK worked");

# Test LIMIT
for (1..6) {
    MSSQLTest::Artist->create( { name => 'Artist ' . $_ } );
}

my $it = MSSQLTest::Artist->search( { },
    { rows     => 3,
      offset   => 2,
      order_by => 'artistid'
    }
);

is( $it->count, 3, "LIMIT count ok" );
ok( $it->next->name, "iterator->next ok" );
$it->next;
$it->next;
is( $it->next, undef, "next past end of resultset ok" );

}

1;