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 | |
cc2c69c1 |
13 | plan tests => 21; |
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 | |
b9a2c3a5 |
96 | SQL |
97 | |
98 | }); |
99 | $schema->populate ('Owners', [ |
134a3bb9 |
100 | [qw/id name /], |
b9a2c3a5 |
101 | [qw/1 wiggle/], |
102 | [qw/2 woggle/], |
103 | [qw/3 boggle/], |
fc85215b |
104 | [qw/4 fREW/], |
105 | [qw/5 fRIOUX/], |
106 | [qw/6 fROOH/], |
107 | [qw/7 fRUE/], |
108 | [qw/8 fISMBoC/], |
109 | [qw/9 station/], |
110 | [qw/10 mirror/], |
111 | [qw/11 dimly/], |
112 | [qw/12 face_to_face/], |
113 | [qw/13 icarus/], |
114 | [qw/14 dream/], |
115 | [qw/15 dyrstyggyr/], |
b9a2c3a5 |
116 | ]); |
117 | |
118 | $schema->populate ('BooksInLibrary', [ |
119 | [qw/source owner title /], |
120 | [qw/Library 1 secrets1/], |
121 | [qw/Eatery 1 secrets2/], |
122 | [qw/Library 2 secrets3/], |
fc85215b |
123 | [qw/Library 3 secrets4/], |
124 | [qw/Eatery 3 secrets5/], |
125 | [qw/Library 4 secrets6/], |
126 | [qw/Library 5 secrets7/], |
127 | [qw/Eatery 5 secrets8/], |
128 | [qw/Library 6 secrets9/], |
129 | [qw/Library 7 secrets10/], |
130 | [qw/Eatery 7 secrets11/], |
131 | [qw/Library 8 secrets12/], |
b9a2c3a5 |
132 | ]); |
133 | |
134 | # |
135 | # try a distinct + prefetch on tables with identically named columns |
136 | # |
137 | |
138 | { |
139 | # try a ->has_many direction (due to a 'multi' accessor the select/group_by group is collapsed) |
fc85215b |
140 | my $owners = $schema->resultset ('Owners')->search ({ |
141 | 'books.id' => { '!=', undef } |
142 | }, { |
143 | prefetch => 'books', |
144 | distinct => 1, |
134a3bb9 |
145 | #order_by => 'name', |
146 | #page => 2, |
147 | #rows => 5, |
fc85215b |
148 | }); |
149 | |
b9a2c3a5 |
150 | my $owners2 = $schema->resultset ('Owners')->search ({ id => { -in => $owners->get_column ('me.id')->as_query }}); |
151 | for ($owners, $owners2) { |
134a3bb9 |
152 | is ($_->all, 8, 'Prefetched grouped search returns correct number of rows'); |
153 | is ($_->count, 8, 'Prefetched grouped search returns correct count'); |
b9a2c3a5 |
154 | } |
155 | |
156 | # try a ->belongs_to direction (no select collapse) |
fc85215b |
157 | my $books = $schema->resultset ('BooksInLibrary')->search ({ |
158 | 'owner.name' => 'wiggle' |
159 | }, { |
160 | prefetch => 'owner', |
161 | distinct => 1, |
134a3bb9 |
162 | #order_by => 'name', |
163 | #page => 2, |
164 | #rows => 5, |
fc85215b |
165 | }); |
166 | |
b9a2c3a5 |
167 | my $books2 = $schema->resultset ('BooksInLibrary')->search ({ id => { -in => $books->get_column ('me.id')->as_query }}); |
168 | for ($books, $books2) { |
169 | is ($_->all, 1, 'Prefetched grouped search returns correct number of rows'); |
170 | is ($_->count, 1, 'Prefetched grouped search returns correct count'); |
171 | } |
fc85215b |
172 | |
b9a2c3a5 |
173 | } |
c1cac633 |
174 | |
175 | # clean up our mess |
176 | END { |
c5f77f6c |
177 | my $dbh = eval { $schema->storage->_dbh }; |
c1cac633 |
178 | $dbh->do('DROP TABLE artist') if $dbh; |
179 | } |
fc85215b |
180 | # vim:sw=2 sts=2 |