Commit | Line | Data |
c1cac633 |
1 | use strict; |
b9a2c3a5 |
2 | use warnings; |
c1cac633 |
3 | |
4 | use Test::More; |
5 | use lib qw(t/lib); |
6 | use DBICTest; |
7 | |
8 | my ($dsn, $user, $pass) = @ENV{map { "DBICTEST_MSSQL_ODBC_${_}" } qw/DSN USER PASS/}; |
9 | |
10 | plan skip_all => 'Set $ENV{DBICTEST_MSSQL_ODBC_DSN}, _USER and _PASS to run this test' |
11 | unless ($dsn && $user); |
12 | |
b9a2c3a5 |
13 | plan tests => 18; |
c1cac633 |
14 | |
15 | my $schema = DBICTest::Schema->connect($dsn, $user, $pass, {AutoCommit => 1}); |
16 | |
8c0104fe |
17 | { |
18 | no warnings 'redefine'; |
19 | my $connect_count = 0; |
20 | my $orig_connect = \&DBI::connect; |
21 | local *DBI::connect = sub { $connect_count++; goto &$orig_connect }; |
22 | |
23 | $schema->storage->ensure_connected; |
24 | |
25 | is( $connect_count, 1, 'only one connection made'); |
26 | } |
9b3e916d |
27 | |
c1cac633 |
28 | isa_ok( $schema->storage, 'DBIx::Class::Storage::DBI::ODBC::Microsoft_SQL_Server' ); |
29 | |
c5f77f6c |
30 | $schema->storage->dbh_do (sub { |
31 | my ($storage, $dbh) = @_; |
32 | eval { $dbh->do("DROP TABLE artist") }; |
33 | $dbh->do(<<'SQL'); |
c1cac633 |
34 | |
c1cac633 |
35 | CREATE TABLE artist ( |
36 | artistid INT IDENTITY NOT NULL, |
a0dd8679 |
37 | name VARCHAR(100), |
39da2a2b |
38 | rank INT NOT NULL DEFAULT '13', |
2eebd801 |
39 | charfield CHAR(10) NULL, |
c1cac633 |
40 | primary key(artistid) |
41 | ) |
42 | |
c5f77f6c |
43 | SQL |
44 | |
45 | }); |
46 | |
c1cac633 |
47 | my %seen_id; |
48 | |
2eebd801 |
49 | # fresh $schema so we start unconnected |
50 | $schema = DBICTest::Schema->connect($dsn, $user, $pass, {AutoCommit => 1}); |
51 | |
c1cac633 |
52 | # test primary key handling |
53 | my $new = $schema->resultset('Artist')->create({ name => 'foo' }); |
54 | ok($new->artistid > 0, "Auto-PK worked"); |
55 | |
56 | $seen_id{$new->artistid}++; |
57 | |
58 | # test LIMIT support |
59 | for (1..6) { |
60 | $new = $schema->resultset('Artist')->create({ name => 'Artist ' . $_ }); |
61 | is ( $seen_id{$new->artistid}, undef, "id for Artist $_ is unique" ); |
62 | $seen_id{$new->artistid}++; |
63 | } |
64 | |
65 | my $it = $schema->resultset('Artist')->search( {}, { |
66 | rows => 3, |
67 | order_by => 'artistid', |
68 | }); |
69 | |
70 | is( $it->count, 3, "LIMIT count ok" ); |
71 | is( $it->next->name, "foo", "iterator->next ok" ); |
72 | $it->next; |
73 | is( $it->next->name, "Artist 2", "iterator->next ok" ); |
74 | is( $it->next, undef, "next past end of resultset ok" ); |
75 | |
b9a2c3a5 |
76 | $schema->storage->dbh_do (sub { |
77 | my ($storage, $dbh) = @_; |
78 | eval { $dbh->do("DROP TABLE Owners") }; |
79 | eval { $dbh->do("DROP TABLE Books") }; |
80 | $dbh->do(<<'SQL'); |
81 | |
82 | |
83 | CREATE TABLE Books ( |
84 | id INT IDENTITY (1, 1) NOT NULL, |
85 | source VARCHAR(100), |
86 | owner INT, |
87 | title VARCHAR(10), |
88 | price INT NULL |
89 | ) |
90 | |
91 | CREATE TABLE Owners ( |
92 | id INT IDENTITY (1, 1) NOT NULL, |
93 | [name] VARCHAR(100), |
94 | ) |
95 | |
96 | SET IDENTITY_INSERT Owners ON |
97 | |
98 | SQL |
99 | |
100 | }); |
101 | $schema->populate ('Owners', [ |
102 | [qw/id [name] /], |
103 | [qw/1 wiggle/], |
104 | [qw/2 woggle/], |
105 | [qw/3 boggle/], |
106 | ]); |
107 | |
108 | $schema->populate ('BooksInLibrary', [ |
109 | [qw/source owner title /], |
110 | [qw/Library 1 secrets1/], |
111 | [qw/Eatery 1 secrets2/], |
112 | [qw/Library 2 secrets3/], |
113 | ]); |
114 | |
115 | # |
116 | # try a distinct + prefetch on tables with identically named columns |
117 | # |
118 | |
119 | { |
120 | # try a ->has_many direction (due to a 'multi' accessor the select/group_by group is collapsed) |
121 | my $owners = $schema->resultset ('Owners')->search ( |
122 | { 'books.id' => { '!=', undef }}, |
123 | { prefetch => 'books', distinct => 1 } |
124 | ); |
125 | my $owners2 = $schema->resultset ('Owners')->search ({ id => { -in => $owners->get_column ('me.id')->as_query }}); |
126 | for ($owners, $owners2) { |
127 | is ($_->all, 2, 'Prefetched grouped search returns correct number of rows'); |
128 | is ($_->count, 2, 'Prefetched grouped search returns correct count'); |
129 | } |
130 | |
131 | # try a ->belongs_to direction (no select collapse) |
132 | my $books = $schema->resultset ('BooksInLibrary')->search ( |
133 | { 'owner.name' => 'wiggle' }, |
134 | { prefetch => 'owner', distinct => 1 } |
135 | ); |
136 | my $books2 = $schema->resultset ('BooksInLibrary')->search ({ id => { -in => $books->get_column ('me.id')->as_query }}); |
137 | for ($books, $books2) { |
138 | is ($_->all, 1, 'Prefetched grouped search returns correct number of rows'); |
139 | is ($_->count, 1, 'Prefetched grouped search returns correct count'); |
140 | } |
141 | } |
c1cac633 |
142 | |
143 | # clean up our mess |
144 | END { |
c5f77f6c |
145 | my $dbh = eval { $schema->storage->_dbh }; |
c1cac633 |
146 | $dbh->do('DROP TABLE artist') if $dbh; |
147 | } |
148 | |