substantially reduced ping count, dynamic cursors support for mssql through odbc
[dbsrgits/DBIx-Class.git] / t / 746mssql.t
1 use strict;
2 use warnings;
3
4 use Test::More;
5 use Test::Exception;
6 use lib qw(t/lib);
7 use DBICTest;
8 use DBIC::SqlMakerTest;
9
10 my ($dsn, $user, $pass) = @ENV{map { "DBICTEST_MSSQL_ODBC_${_}" } qw/DSN USER PASS/};
11
12 plan skip_all => 'Set $ENV{DBICTEST_MSSQL_ODBC_DSN}, _USER and _PASS to run this test'
13   unless ($dsn && $user);
14
15 plan tests => 28;
16
17 my $schema = DBICTest::Schema->connect($dsn, $user, $pass);
18
19 {
20   no warnings 'redefine';
21   my $connect_count = 0;
22   my $orig_connect = \&DBI::connect;
23   local *DBI::connect = sub { $connect_count++; goto &$orig_connect };
24
25   $schema->storage->ensure_connected;
26
27   is( $connect_count, 1, 'only one connection made');
28 }
29
30 isa_ok( $schema->storage, 'DBIx::Class::Storage::DBI::ODBC::Microsoft_SQL_Server' );
31
32 $schema->storage->dbh_do (sub {
33     my ($storage, $dbh) = @_;
34     eval { $dbh->do("DROP TABLE artist") };
35     $dbh->do(<<'SQL');
36
37 CREATE TABLE artist (
38    artistid INT IDENTITY NOT NULL,
39    name VARCHAR(100),
40    rank INT NOT NULL DEFAULT '13',
41    charfield CHAR(10) NULL,
42    primary key(artistid)
43 )
44
45 SQL
46
47 });
48
49 my %seen_id;
50
51 my @opts = (
52   { on_connect_call => 'use_dynamic_cursors' },
53   {},
54 );
55 my $new;
56
57 # test Auto-PK with different options
58 for my $opts (@opts) {
59   $schema = DBICTest::Schema->clone;
60   $schema->connection($dsn, $user, $pass, $opts);
61
62   $schema->resultset('Artist')->search({ name => 'foo' })->delete;
63
64   $new = $schema->resultset('Artist')->create({ name => 'foo' });
65   ok($new->artistid > 0, "Auto-PK worked");
66 }
67
68 $seen_id{$new->artistid}++;
69
70 # test LIMIT support
71 for (1..6) {
72     $new = $schema->resultset('Artist')->create({ name => 'Artist ' . $_ });
73     is ( $seen_id{$new->artistid}, undef, "id for Artist $_ is unique" );
74     $seen_id{$new->artistid}++;
75 }
76
77 my $it = $schema->resultset('Artist')->search( {}, {
78     rows => 3,
79     order_by => 'artistid',
80 });
81
82 is( $it->count, 3, "LIMIT count ok" );
83 is( $it->next->name, "foo", "iterator->next ok" );
84 $it->next;
85 is( $it->next->name, "Artist 2", "iterator->next ok" );
86 is( $it->next, undef, "next past end of resultset ok" );
87
88 $schema->storage->dbh_do (sub {
89     my ($storage, $dbh) = @_;
90     eval { $dbh->do("DROP TABLE Owners") };
91     eval { $dbh->do("DROP TABLE Books") };
92     $dbh->do(<<'SQL');
93
94
95 CREATE TABLE Books (
96    id INT IDENTITY (1, 1) NOT NULL,
97    source VARCHAR(100),
98    owner INT,
99    title VARCHAR(10),
100    price INT NULL
101 )
102
103 CREATE TABLE Owners (
104    id INT IDENTITY (1, 1) NOT NULL,
105    name VARCHAR(100),
106 )
107
108 SQL
109
110 });
111
112 lives_ok ( sub {
113   $schema->populate ('Owners', [
114     [qw/id  name  /],
115     [qw/1   wiggle/],
116     [qw/2   woggle/],
117     [qw/3   boggle/],
118     [qw/4   fREW/],
119     [qw/5   fRIOUX/],
120     [qw/6   fROOH/],
121     [qw/7   fRUE/],
122     [qw/8   fISMBoC/],
123     [qw/9   station/],
124     [qw/10   mirror/],
125     [qw/11   dimly/],
126     [qw/12   face_to_face/],
127     [qw/13   icarus/],
128     [qw/14   dream/],
129     [qw/15   dyrstyggyr/],
130   ]);
131 }, 'populate with PKs supplied ok' );
132
133 lives_ok ( sub {
134   $schema->populate ('BooksInLibrary', [
135     [qw/source  owner title   /],
136     [qw/Library 1     secrets0/],
137     [qw/Library 1     secrets1/],
138     [qw/Eatery  1     secrets2/],
139     [qw/Library 2     secrets3/],
140     [qw/Library 3     secrets4/],
141     [qw/Eatery  3     secrets5/],
142     [qw/Library 4     secrets6/],
143     [qw/Library 5     secrets7/],
144     [qw/Eatery  5     secrets8/],
145     [qw/Library 6     secrets9/],
146     [qw/Library 7     secrets10/],
147     [qw/Eatery  7     secrets11/],
148     [qw/Library 8     secrets12/],
149   ]);
150 }, 'populate without PKs supplied ok' );
151
152 #
153 # try a prefetch on tables with identically named columns
154 #
155
156 # set quote char - make sure things work while quoted
157 $schema->storage->_sql_maker->{quote_char} = [qw/[ ]/];
158 $schema->storage->_sql_maker->{name_sep} = '.';
159
160 {
161   # try a ->has_many direction
162   my $owners = $schema->resultset ('Owners')->search ({
163       'books.id' => { '!=', undef }
164     }, {
165       prefetch => 'books',
166       order_by => 'name',
167       rows     => 3,  # 8 results total
168     });
169
170   is ($owners->page(1)->all, 3, 'has_many prefetch returns correct number of rows');
171   is ($owners->page(1)->count, 3, 'has-many prefetch returns correct count');
172
173   TODO: {
174     local $TODO = 'limit past end of resultset problem';
175     is ($owners->page(3)->all, 2, 'has_many prefetch returns correct number of rows');
176     is ($owners->page(3)->count, 2, 'has-many prefetch returns correct count');
177     is ($owners->page(3)->count_rs->next, 2, 'has-many prefetch returns correct count_rs');
178
179     # make sure count does not become overly complex FIXME
180     is_same_sql_bind (
181       $owners->page(3)->count_rs->as_query,
182       '(
183         SELECT COUNT( * )
184           FROM (
185             SELECT TOP 3 [me].[id]
186               FROM [owners] [me]
187               LEFT JOIN [books] [books] ON [books].[owner] = [me].[id]
188             WHERE ( [books].[id] IS NOT NULL )
189             GROUP BY [me].[id]
190             ORDER BY [me].[id] DESC
191           ) [count_subq]
192       )',
193       [],
194     );
195   }
196
197   # try a ->belongs_to direction (no select collapse, group_by should work)
198   my $books = $schema->resultset ('BooksInLibrary')->search ({
199       'owner.name' => [qw/wiggle woggle/],
200     }, {
201       distinct => 1,
202       prefetch => 'owner',
203       rows     => 2,  # 3 results total
204       order_by => { -desc => 'owner' },
205       # there is no sane way to order by the right side of a grouped prefetch currently :(
206       #order_by => { -desc => 'owner.name' },
207     });
208
209
210   is ($books->page(1)->all, 2, 'Prefetched grouped search returns correct number of rows');
211   is ($books->page(1)->count, 2, 'Prefetched grouped search returns correct count');
212
213   TODO: {
214     local $TODO = 'limit past end of resultset problem';
215     is ($books->page(2)->all, 1, 'Prefetched grouped search returns correct number of rows');
216     is ($books->page(2)->count, 1, 'Prefetched grouped search returns correct count');
217     is ($books->page(2)->count_rs->next, 1, 'Prefetched grouped search returns correct count_rs');
218
219     # make sure count does not become overly complex FIXME
220     is_same_sql_bind (
221       $books->page(2)->count_rs->as_query,
222       '(
223         SELECT COUNT( * )
224           FROM (
225             SELECT TOP 2 [me].[id]
226               FROM [books] [me]
227               JOIN [owners] [owner] ON [owner].[id] = [me].[owner]
228             WHERE ( ( ( [owner].[name] = ? OR [owner].[name] = ? ) AND [source] = ? ) )
229             GROUP BY [me].[id], [me].[source], [me].[owner], [me].[title], [me].[price], [owner].[id], [owner].[name]
230             ORDER BY [me].[id] DESC
231           ) [count_subq]
232       )',
233       [
234         [ 'owner.name' => 'wiggle' ],
235         [ 'owner.name' => 'woggle' ],
236         [ 'source' => 'Library' ],
237       ],
238     );
239   }
240
241 }
242
243 # clean up our mess
244 END {
245     my $dbh = eval { $schema->storage->_dbh };
246     $dbh->do('DROP TABLE artist') if $dbh;
247 }
248 # vim:sw=2 sts=2