ydbi 0.5.9 → 0.6.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/.envrc +3 -0
- data/.github/workflows/ruby.yml +65 -0
- data/.gitignore +19 -0
- data/ChangeLog +9 -0
- data/Gemfile +5 -0
- data/Rakefile +14 -0
- data/TODO +44 -0
- data/bench/bench.rb +79 -0
- data/build/rake_task_lib.rb +186 -0
- data/devenv.lock +228 -0
- data/devenv.nix +55 -0
- data/devenv.yaml +8 -0
- data/doc/DBD_SPEC.rdoc +88 -0
- data/doc/DBI_SPEC.rdoc +157 -0
- data/doc/homepage/contact.html +62 -0
- data/doc/homepage/development.html +124 -0
- data/doc/homepage/index.html +83 -0
- data/doc/homepage/ruby-dbi.css +91 -0
- data/lib/dbd/Mysql.rb +137 -0
- data/lib/dbd/ODBC.rb +89 -0
- data/lib/dbd/Pg.rb +189 -0
- data/lib/dbd/SQLite.rb +97 -0
- data/lib/dbd/SQLite3.rb +124 -0
- data/lib/dbd/mysql/database.rb +405 -0
- data/lib/dbd/mysql/driver.rb +125 -0
- data/lib/dbd/mysql/statement.rb +188 -0
- data/lib/dbd/odbc/database.rb +128 -0
- data/lib/dbd/odbc/driver.rb +38 -0
- data/lib/dbd/odbc/statement.rb +137 -0
- data/lib/dbd/pg/database.rb +508 -0
- data/lib/dbd/pg/exec.rb +47 -0
- data/lib/dbd/pg/statement.rb +160 -0
- data/lib/dbd/pg/tuples.rb +121 -0
- data/lib/dbd/pg/type.rb +209 -0
- data/lib/dbd/sqlite/database.rb +151 -0
- data/lib/dbd/sqlite/statement.rb +125 -0
- data/lib/dbd/sqlite3/database.rb +201 -0
- data/lib/dbd/sqlite3/statement.rb +78 -0
- data/lib/dbi/version.rb +1 -1
- data/prototypes/types2.rb +237 -0
- data/setup.rb +1585 -0
- data/test/DBD_TESTS +50 -0
- data/test/TESTING +16 -0
- data/test/dbd/general/test_database.rb +206 -0
- data/test/dbd/general/test_statement.rb +325 -0
- data/test/dbd/general/test_types.rb +295 -0
- data/test/dbd/mysql/base.rb +26 -0
- data/test/dbd/mysql/down.sql +19 -0
- data/test/dbd/mysql/test_blob.rb +18 -0
- data/test/dbd/mysql/test_new_methods.rb +7 -0
- data/test/dbd/mysql/test_patches.rb +111 -0
- data/test/dbd/mysql/up.sql +28 -0
- data/test/dbd/odbc/base.rb +30 -0
- data/test/dbd/odbc/down.sql +19 -0
- data/test/dbd/odbc/test_new_methods.rb +12 -0
- data/test/dbd/odbc/test_ping.rb +10 -0
- data/test/dbd/odbc/test_statement.rb +44 -0
- data/test/dbd/odbc/test_transactions.rb +58 -0
- data/test/dbd/odbc/up.sql +33 -0
- data/test/dbd/postgresql/base.rb +31 -0
- data/test/dbd/postgresql/down.sql +31 -0
- data/test/dbd/postgresql/test_arrays.rb +179 -0
- data/test/dbd/postgresql/test_async.rb +121 -0
- data/test/dbd/postgresql/test_blob.rb +37 -0
- data/test/dbd/postgresql/test_bytea.rb +88 -0
- data/test/dbd/postgresql/test_ping.rb +10 -0
- data/test/dbd/postgresql/test_timestamp.rb +77 -0
- data/test/dbd/postgresql/test_transactions.rb +58 -0
- data/test/dbd/postgresql/testdbipg.rb +307 -0
- data/test/dbd/postgresql/up.sql +60 -0
- data/test/dbd/sqlite/base.rb +32 -0
- data/test/dbd/sqlite/test_database.rb +30 -0
- data/test/dbd/sqlite/test_driver.rb +68 -0
- data/test/dbd/sqlite/test_statement.rb +112 -0
- data/test/dbd/sqlite/up.sql +25 -0
- data/test/dbd/sqlite3/base.rb +32 -0
- data/test/dbd/sqlite3/test_database.rb +77 -0
- data/test/dbd/sqlite3/test_driver.rb +67 -0
- data/test/dbd/sqlite3/test_statement.rb +88 -0
- data/test/dbd/sqlite3/up.sql +33 -0
- data/test/dbi/tc_dbi.rb +1 -1
- data/test/ts_dbd.rb +136 -0
- data/ydbi.gemspec +25 -0
- metadata +150 -13
@@ -0,0 +1,33 @@
|
|
1
|
+
create table names (
|
2
|
+
name varchar(255),
|
3
|
+
age integer
|
4
|
+
);
|
5
|
+
---
|
6
|
+
insert into names (name, age) values ('Joe', 19);
|
7
|
+
---
|
8
|
+
insert into names (name, age) values ('Jim', 30);
|
9
|
+
---
|
10
|
+
insert into names (name, age) values ('Bob', 21);
|
11
|
+
---
|
12
|
+
create table precision_test (text_field varchar(20) primary key not null, integer_field integer, decimal_field decimal(2,1), numeric_field numeric(30,6));
|
13
|
+
---
|
14
|
+
CREATE TABLE blob_test (name VARCHAR(30), data OID);
|
15
|
+
---
|
16
|
+
create view view_names as select * from names;
|
17
|
+
---
|
18
|
+
create or replace function test_insert (varchar(255), integer)
|
19
|
+
returns integer
|
20
|
+
language sql
|
21
|
+
as 'insert into names (name, age) values ($1, $2); select age from names where name = $1';
|
22
|
+
---
|
23
|
+
create table boolean_test (num integer, mybool boolean);
|
24
|
+
---
|
25
|
+
create table time_test (mytime time);
|
26
|
+
---
|
27
|
+
create table timestamp_test (mytimestamp timestamp);
|
28
|
+
---
|
29
|
+
create table bit_test (mybit bit);
|
30
|
+
---
|
31
|
+
create table field_types_test (foo integer not null primary key default 1);
|
32
|
+
---
|
33
|
+
create table db_specific_types_test (foo integer);
|
@@ -0,0 +1,31 @@
|
|
1
|
+
require 'fileutils'
|
2
|
+
|
3
|
+
DBDConfig.set_testbase(:postgresql, Class.new(Test::Unit::TestCase) do
|
4
|
+
|
5
|
+
def dbtype
|
6
|
+
"postgresql"
|
7
|
+
end
|
8
|
+
|
9
|
+
def test_base
|
10
|
+
assert_equal(@dbh.driver_name, "Pg")
|
11
|
+
assert_kind_of(DBI::DBD::Pg::Database, @dbh.instance_variable_get(:@handle))
|
12
|
+
end
|
13
|
+
|
14
|
+
def set_base_dbh
|
15
|
+
config = DBDConfig.get_config['postgresql']
|
16
|
+
@dbh = DBI.connect("dbi:Pg:#{config['dbname']}", config['username'], config['password'])
|
17
|
+
end
|
18
|
+
|
19
|
+
def setup
|
20
|
+
set_base_dbh
|
21
|
+
DBDConfig.inject_sql(@dbh, dbtype, "dbd/postgresql/up.sql")
|
22
|
+
end
|
23
|
+
|
24
|
+
def teardown
|
25
|
+
@sth.finish if @sth && !@sth.finished?
|
26
|
+
config = DBDConfig.get_config['postgresql']
|
27
|
+
DBDConfig.inject_sql(@dbh, dbtype, "dbd/postgresql/down.sql")
|
28
|
+
@dbh.disconnect
|
29
|
+
end
|
30
|
+
end
|
31
|
+
)
|
@@ -0,0 +1,31 @@
|
|
1
|
+
drop view view_names;
|
2
|
+
---
|
3
|
+
drop table names;
|
4
|
+
---
|
5
|
+
drop table blob_test;
|
6
|
+
---
|
7
|
+
drop table boolean_test;
|
8
|
+
---
|
9
|
+
drop table time_test;
|
10
|
+
---
|
11
|
+
drop table timestamp_test;
|
12
|
+
---
|
13
|
+
drop table bit_test;
|
14
|
+
---
|
15
|
+
drop table field_types_test;
|
16
|
+
---
|
17
|
+
drop table db_specific_types_test;
|
18
|
+
---
|
19
|
+
drop table array_test;
|
20
|
+
---
|
21
|
+
drop table bytea_test;
|
22
|
+
---
|
23
|
+
drop table precision_test;
|
24
|
+
---
|
25
|
+
drop table enum_type_test;
|
26
|
+
---
|
27
|
+
drop type enum_test;
|
28
|
+
---
|
29
|
+
drop schema schema1 cascade;
|
30
|
+
---
|
31
|
+
drop schema schema2 cascade;
|
@@ -0,0 +1,179 @@
|
|
1
|
+
class TestPostgresArray < DBDConfig.testbase(:postgresql)
|
2
|
+
def test_array_type
|
3
|
+
assert_nothing_raised do
|
4
|
+
cols = @dbh.columns("array_test")
|
5
|
+
assert_equal(
|
6
|
+
[
|
7
|
+
{
|
8
|
+
:name =>"foo",
|
9
|
+
:default =>nil,
|
10
|
+
:primary =>nil,
|
11
|
+
:scale =>nil,
|
12
|
+
:sql_type =>DBI::SQL_INTEGER,
|
13
|
+
:nullable =>true,
|
14
|
+
:indexed =>false,
|
15
|
+
:precision =>-1,
|
16
|
+
:type_name =>"integer",
|
17
|
+
:unique =>nil,
|
18
|
+
:array_of_type => true
|
19
|
+
},
|
20
|
+
{
|
21
|
+
:name =>"bar",
|
22
|
+
:default =>nil,
|
23
|
+
:primary =>nil,
|
24
|
+
:scale =>nil,
|
25
|
+
:sql_type =>DBI::SQL_INTEGER,
|
26
|
+
:nullable =>true,
|
27
|
+
:indexed =>false,
|
28
|
+
:precision =>-1,
|
29
|
+
:type_name =>"integer",
|
30
|
+
:unique =>nil,
|
31
|
+
:array_of_type => true
|
32
|
+
},
|
33
|
+
{
|
34
|
+
:name =>"baz",
|
35
|
+
:default =>nil,
|
36
|
+
:primary =>nil,
|
37
|
+
:scale =>nil,
|
38
|
+
:sql_type =>DBI::SQL_INTEGER,
|
39
|
+
:nullable =>true,
|
40
|
+
:indexed =>false,
|
41
|
+
:precision =>-1,
|
42
|
+
:type_name =>"integer",
|
43
|
+
:unique =>nil,
|
44
|
+
:array_of_type => true
|
45
|
+
},
|
46
|
+
{
|
47
|
+
:array_of_type=>true,
|
48
|
+
:unique=>nil,
|
49
|
+
:precision=>-1,
|
50
|
+
:name=>"quux",
|
51
|
+
:default=>nil,
|
52
|
+
:indexed=>false,
|
53
|
+
:scale=>nil,
|
54
|
+
:primary=>nil,
|
55
|
+
:sql_type=>12,
|
56
|
+
:nullable=>true,
|
57
|
+
:type_name=>"character varying"
|
58
|
+
}
|
59
|
+
], cols.collect { |x| x.reject { |key, value| key == :dbi_type } }
|
60
|
+
)
|
61
|
+
|
62
|
+
assert_equal(([DBI::DBD::Pg::Type::Array] * 4), cols.collect { |x| x["dbi_type"].class })
|
63
|
+
assert_equal((([DBI::Type::Integer] * 3) + [DBI::Type::Varchar]), cols.collect { |x| x["dbi_type"].base_type })
|
64
|
+
end
|
65
|
+
end
|
66
|
+
|
67
|
+
def test_array_parser
|
68
|
+
# string representation
|
69
|
+
assert_nothing_raised do
|
70
|
+
sth = @dbh.prepare('insert into array_test (foo) values (?)')
|
71
|
+
sth.execute('{1,2,3}')
|
72
|
+
sth.finish
|
73
|
+
end
|
74
|
+
|
75
|
+
assert_nothing_raised do
|
76
|
+
sth = @dbh.prepare('insert into array_test (foo) values (?)')
|
77
|
+
sth.execute([1,2,3])
|
78
|
+
sth.finish
|
79
|
+
end
|
80
|
+
|
81
|
+
assert_nothing_raised do
|
82
|
+
sth = @dbh.prepare('insert into array_test (quux) values (?)')
|
83
|
+
sth.execute(["Hello\\ World", "Again\\"])
|
84
|
+
sth.finish
|
85
|
+
end
|
86
|
+
|
87
|
+
assert_nothing_raised do
|
88
|
+
sth = @dbh.prepare('select foo from array_test where foo is not null')
|
89
|
+
sth.execute
|
90
|
+
assert_equal(
|
91
|
+
[
|
92
|
+
[[1,2,3]],
|
93
|
+
[[1,2,3]],
|
94
|
+
], sth.fetch_all
|
95
|
+
)
|
96
|
+
sth.finish
|
97
|
+
|
98
|
+
sth = @dbh.prepare('select quux from array_test where quux is not null')
|
99
|
+
sth.execute
|
100
|
+
|
101
|
+
assert_equal(
|
102
|
+
[
|
103
|
+
[["Hello\\ World", "Again\\"]]
|
104
|
+
], sth.fetch_all
|
105
|
+
)
|
106
|
+
|
107
|
+
sth.finish
|
108
|
+
end
|
109
|
+
end
|
110
|
+
|
111
|
+
def test_array_boundaries
|
112
|
+
# bar has a max extents of 3
|
113
|
+
sth = @dbh.prepare('insert into array_test (bar) values (?)')
|
114
|
+
|
115
|
+
assert_nothing_raised do
|
116
|
+
sth.execute('{1,2,3}')
|
117
|
+
end
|
118
|
+
|
119
|
+
# XXX postgresql does not enforce extents on single-dimension arrays
|
120
|
+
assert_nothing_raised do
|
121
|
+
sth.execute('{1,2,3,4}')
|
122
|
+
end
|
123
|
+
|
124
|
+
sth.finish
|
125
|
+
sth = @dbh.prepare('insert into array_test(baz) values (?)')
|
126
|
+
|
127
|
+
assert_nothing_raised do
|
128
|
+
sth.execute('{{1,2,3}, {1,2,3}}')
|
129
|
+
end
|
130
|
+
|
131
|
+
assert_nothing_raised do
|
132
|
+
# XXX for the record, I have no fucking idea why this works, what
|
133
|
+
# it's technically represented as and what backwards array
|
134
|
+
# implementation would allow it to work.
|
135
|
+
#
|
136
|
+
# I'm hoping it'll break on some future version of postgresql so I
|
137
|
+
# can fix it.
|
138
|
+
sth.execute('{1,2,3}')
|
139
|
+
end
|
140
|
+
|
141
|
+
assert_raises(DBI::ProgrammingError) do
|
142
|
+
sth.execute('{{1,2,3,4}, {1,2,3}}')
|
143
|
+
end
|
144
|
+
|
145
|
+
assert_raises(DBI::ProgrammingError) do
|
146
|
+
sth.execute('{{1,2,3}, {1,2,3,4}}')
|
147
|
+
end
|
148
|
+
|
149
|
+
sth.finish
|
150
|
+
end
|
151
|
+
|
152
|
+
def test_array_type_parser
|
153
|
+
pc = DBI::DBD::Pg::Type::Array
|
154
|
+
|
155
|
+
assert_nothing_raised do
|
156
|
+
po = pc.new(DBI::Type::Integer)
|
157
|
+
assert_equal([1,2,3], po.parse("{1,2,3}"))
|
158
|
+
assert_equal([[1,2,3],[4,5,6]], po.parse("{{1,2,3},{4,5,6}}"))
|
159
|
+
end
|
160
|
+
|
161
|
+
assert_nothing_raised do
|
162
|
+
po = pc.new(DBI::Type::Varchar)
|
163
|
+
assert_equal(["one", "two", "three"], po.parse("{\"one\",\"two\",\"three\"}"))
|
164
|
+
assert_equal([["one"], ["two\\"]], po.parse("{{\"one\"},{\"two\\\\\"}}"))
|
165
|
+
assert_equal([["one", "two\\"], ["three\\", "four"]], po.parse("{{\"one\",\"two\\\\\"},{\"three\\\\\",\"four\"}}"))
|
166
|
+
end
|
167
|
+
end
|
168
|
+
|
169
|
+
def test_array_generator
|
170
|
+
pg = DBI::DBD::Pg
|
171
|
+
|
172
|
+
assert_nothing_raised do
|
173
|
+
assert_equal("{1,2,3}", pg.generate_array([1,2,3]))
|
174
|
+
assert_equal("{{1,2,3},{1,2,3}}", pg.generate_array([[1,2,3],[1,2,3]]))
|
175
|
+
assert_equal("{\"one\",\"two\"}", pg.generate_array(["one", "two"]))
|
176
|
+
assert_equal("{\"hello\\\\ world\",\"again\"}", pg.generate_array(["hello\\ world", "again"]))
|
177
|
+
end
|
178
|
+
end
|
179
|
+
end
|
@@ -0,0 +1,121 @@
|
|
1
|
+
class TestPostgresAsync < DBDConfig.testbase(:postgresql)
|
2
|
+
def get_dbh
|
3
|
+
config = DBDConfig.get_config['postgresql']
|
4
|
+
DBI.connect("dbi:Pg:#{config['dbname']}", config['username'], config['password'])
|
5
|
+
end
|
6
|
+
|
7
|
+
def get_db_info
|
8
|
+
config = DBDConfig.get_config['postgresql']
|
9
|
+
dsn = "dbi:Pg:database=#{config['dbname']}"
|
10
|
+
user = config['username']
|
11
|
+
pass = config['password']
|
12
|
+
|
13
|
+
return [dsn, user, pass]
|
14
|
+
end
|
15
|
+
|
16
|
+
def test_async_default
|
17
|
+
dsn, user, pass = get_db_info
|
18
|
+
|
19
|
+
DBI.connect(dsn, user, pass) do |dbh|
|
20
|
+
assert_equal false, dbh['pg_async']
|
21
|
+
assert_equal false, dbh['NonBlocking']
|
22
|
+
dbh.prepare('SELECT 1') do |sth| # Statement inherits
|
23
|
+
assert_equal false, sth['pg_async']
|
24
|
+
assert_equal false, sth['NonBlocking']
|
25
|
+
end
|
26
|
+
end
|
27
|
+
end
|
28
|
+
|
29
|
+
def test_async_dsn_enable
|
30
|
+
dsn, user, pass = get_db_info
|
31
|
+
|
32
|
+
for enable in ['true', 'TRUE', 'tRuE']
|
33
|
+
DBI.connect(dsn + ";pg_async=#{enable}", user, pass) do |dbh|
|
34
|
+
assert_equal true, dbh['pg_async']
|
35
|
+
assert_equal true, dbh['NonBlocking']
|
36
|
+
dbh.prepare('SELECT 1') do |sth| # Statement inherits
|
37
|
+
assert_equal true, sth['pg_async']
|
38
|
+
assert_equal true, sth['NonBlocking']
|
39
|
+
end
|
40
|
+
end
|
41
|
+
end
|
42
|
+
end
|
43
|
+
|
44
|
+
def test_async_attr_enable
|
45
|
+
dsn, user, pass = get_db_info
|
46
|
+
|
47
|
+
for enable in ['true', 'TRUE', 'tRuE']
|
48
|
+
DBI.connect(dsn, user, pass, { 'pg_async' => enable } ) do |dbh|
|
49
|
+
assert_equal true, dbh['pg_async']
|
50
|
+
assert_equal true, dbh['NonBlocking']
|
51
|
+
dbh.prepare('SELECT 1') do |sth| # Statement inherits
|
52
|
+
assert_equal true, sth['pg_async']
|
53
|
+
assert_equal true, sth['NonBlocking']
|
54
|
+
end
|
55
|
+
end
|
56
|
+
end
|
57
|
+
end
|
58
|
+
|
59
|
+
def test_async_dsn_disable
|
60
|
+
dsn, user, pass = get_db_info
|
61
|
+
|
62
|
+
for disable in ['false', 'FALSE', 'fAlSe']
|
63
|
+
DBI.connect(dsn + ";pg_async=#{disable}", user, pass) do |dbh|
|
64
|
+
assert_equal false, dbh['pg_async']
|
65
|
+
assert_equal false, dbh['NonBlocking']
|
66
|
+
dbh.prepare('SELECT 1') do |sth| # Statement inherits
|
67
|
+
assert_equal false, sth['pg_async']
|
68
|
+
assert_equal false, sth['NonBlocking']
|
69
|
+
end
|
70
|
+
end
|
71
|
+
end
|
72
|
+
end
|
73
|
+
|
74
|
+
def test_async_attr_disable
|
75
|
+
dsn, user, pass = get_db_info
|
76
|
+
|
77
|
+
for disable in ['false', 'FALSE', 'fAlSe']
|
78
|
+
DBI.connect(dsn, user, pass, { 'pg_async' => disable }) do |dbh|
|
79
|
+
assert_equal false, dbh['pg_async']
|
80
|
+
assert_equal false, dbh['NonBlocking']
|
81
|
+
dbh.prepare('SELECT 1') do |sth| # Statement inherits
|
82
|
+
assert_equal false, sth['pg_async']
|
83
|
+
assert_equal false, sth['NonBlocking']
|
84
|
+
end
|
85
|
+
end
|
86
|
+
end
|
87
|
+
end
|
88
|
+
|
89
|
+
def test_manual_enable
|
90
|
+
dsn, user, pass = get_db_info
|
91
|
+
|
92
|
+
DBI.connect(dsn, user, pass) do |dbh|
|
93
|
+
dbh['pg_async'] = true
|
94
|
+
assert_equal true, dbh['pg_async']
|
95
|
+
assert_equal true, dbh['NonBlocking']
|
96
|
+
dbh.prepare('SELECT 1') do |sth| # Statement inherits
|
97
|
+
assert_equal true, sth['pg_async']
|
98
|
+
assert_equal true, sth['NonBlocking']
|
99
|
+
end
|
100
|
+
end
|
101
|
+
end
|
102
|
+
|
103
|
+
def test_async_commands
|
104
|
+
dsn, user, pass = get_db_info
|
105
|
+
|
106
|
+
DBI.connect(dsn + ";pg_async=true", user, pass) do |dbh|
|
107
|
+
assert_equal true, dbh['pg_async']
|
108
|
+
assert_equal true, dbh['NonBlocking']
|
109
|
+
ret = dbh.select_all('SELECT 1')
|
110
|
+
assert_equal [[1]], ret
|
111
|
+
|
112
|
+
ret = dbh.select_all(%q{SELECT 1 WHERE 'foo' = ?}, 'bar')
|
113
|
+
assert_equal [], ret
|
114
|
+
|
115
|
+
dbh.prepare(%q{SELECT 1 WHERE 'foo' = ?}) do |sth|
|
116
|
+
sth.execute('bar')
|
117
|
+
assert_equal [], sth.fetch_all
|
118
|
+
end
|
119
|
+
end
|
120
|
+
end
|
121
|
+
end
|
@@ -0,0 +1,37 @@
|
|
1
|
+
class TestPostgresBlob < DBDConfig.testbase(:postgresql)
|
2
|
+
DATA = "this is my new binary object"
|
3
|
+
|
4
|
+
def test_insert
|
5
|
+
assert @dbh
|
6
|
+
assert @dbh.ping
|
7
|
+
|
8
|
+
# test with DBI::Binary
|
9
|
+
assert_equal 1, @dbh.do("INSERT INTO blob_test (name, data) VALUES (?,?)", "test", DBI::Binary.new(DATA))
|
10
|
+
|
11
|
+
# test with blob_create directly
|
12
|
+
if false
|
13
|
+
blob = @dbh.func(:blob_create, PGconn::INV_WRITE)
|
14
|
+
assert blob
|
15
|
+
assert @dbh.func(:blob_write, blob, DATA)
|
16
|
+
assert_equal 1, @dbh.do("INSERT INTO blob_test (name, data) VALUES (?,?)", "test (2)", blob)
|
17
|
+
end
|
18
|
+
# test with blob_import directly
|
19
|
+
File.open('/tmp/pg_dbi_import_test', 'w') { |f| f << DATA }
|
20
|
+
blob = @dbh.func(:blob_import, '/tmp/pg_dbi_import_test')
|
21
|
+
assert blob
|
22
|
+
assert_equal 1, @dbh.do("INSERT INTO blob_test (name, data) VALUES (?,?)", "test (2)", blob)
|
23
|
+
|
24
|
+
index = 0
|
25
|
+
@dbh.select_all("SELECT name, data FROM blob_test") do |name, data|
|
26
|
+
index += 1
|
27
|
+
assert_equal DATA, @dbh.func(:blob_read, data, DATA.length)
|
28
|
+
@dbh.func(:blob_export, data, '/tmp/pg_dbi_read_test')
|
29
|
+
assert_equal DATA, File.read('/tmp/pg_dbi_read_test')
|
30
|
+
end
|
31
|
+
|
32
|
+
assert_equal 2, index
|
33
|
+
|
34
|
+
File.unlink("/tmp/pg_dbi_read_test")
|
35
|
+
File.unlink("/tmp/pg_dbi_import_test")
|
36
|
+
end
|
37
|
+
end
|
@@ -0,0 +1,88 @@
|
|
1
|
+
require 'dbd/Pg'
|
2
|
+
|
3
|
+
begin
|
4
|
+
require 'rubygems'
|
5
|
+
gem 'pg'
|
6
|
+
rescue Exception => e
|
7
|
+
end
|
8
|
+
|
9
|
+
require 'pg'
|
10
|
+
|
11
|
+
LEN = 50
|
12
|
+
|
13
|
+
class TestPostgresByteA < DBDConfig.testbase(:postgresql)
|
14
|
+
# FIXME the 'pg' module is broken and doesn't encode/decode properly.
|
15
|
+
# this test should prove that the 'pg' module works so we can back out our
|
16
|
+
# hacks.
|
17
|
+
def skip_underlying_driver
|
18
|
+
str = generate_random_string
|
19
|
+
|
20
|
+
encoded = PGconn.escape_bytea(str.dup)
|
21
|
+
decoded = PGconn.unescape_bytea(encoded)
|
22
|
+
|
23
|
+
assert_equal str, decoded
|
24
|
+
end
|
25
|
+
|
26
|
+
def test_encode_decode
|
27
|
+
bytea = DBI::DBD::Pg::Type::ByteA
|
28
|
+
|
29
|
+
# some specific cases that were failing intermittenly
|
30
|
+
# poor \\ handling
|
31
|
+
str = "\236\000\257\202G<\371\035TPEO\211\005*AH'H\3136\360\004\245\261\037\340u\003s\\772X\231\002\200\n\327\202\217\353\177r\317o\341\237\341"
|
32
|
+
|
33
|
+
omit "We omit all the encode/decode string because I do not want to fix the conversion between UTF-8 and ASCII"
|
34
|
+
encoded = bytea.escape_bytea(str)
|
35
|
+
decoded = bytea.parse(encoded)
|
36
|
+
|
37
|
+
assert_equal str, decoded
|
38
|
+
# the split hack not working
|
39
|
+
str = "\343\336e\260\337\373\314\026\323#\237i\035\0302\024\346X\274\016\324\371\206\036\230\374\206#rA\n\214\272\316\330\025\374\000\2663\244M\255x\360\002\266q\336\231"
|
40
|
+
|
41
|
+
encoded = bytea.escape_bytea(str)
|
42
|
+
decoded = bytea.parse(encoded)
|
43
|
+
|
44
|
+
assert_equal str, decoded
|
45
|
+
|
46
|
+
# delimiter at the end
|
47
|
+
str = "\343\336e\260\337\373\314\026\323#\237i\035\0302\024\346X\274\016\324\371\206\036\230\374\206#rA\n\214\272\316\330\025\374\000\2663\244M\255x\360\002\266q\336\231\\\\\\\\"
|
48
|
+
|
49
|
+
encoded = bytea.escape_bytea(str)
|
50
|
+
decoded = bytea.parse(encoded)
|
51
|
+
|
52
|
+
assert_equal str, decoded
|
53
|
+
|
54
|
+
# a huge test to weed out all the stragglers
|
55
|
+
5_000.times do
|
56
|
+
str = generate_random_string
|
57
|
+
|
58
|
+
encoded = bytea.escape_bytea(str)
|
59
|
+
decoded = bytea.parse(encoded)
|
60
|
+
|
61
|
+
assert_equal str, decoded
|
62
|
+
end
|
63
|
+
end
|
64
|
+
|
65
|
+
def test_insert_retrieve
|
66
|
+
assert_nothing_raised do
|
67
|
+
str = generate_random_string
|
68
|
+
sth = @dbh.prepare('insert into bytea_test (foo) values (?)')
|
69
|
+
sth.execute(DBI::DBD::Pg::Type::ByteA.new(str))
|
70
|
+
sth.finish
|
71
|
+
|
72
|
+
sth = @dbh.prepare('select * from bytea_test')
|
73
|
+
sth.execute
|
74
|
+
assert_equal([str], sth.fetch)
|
75
|
+
sth.finish
|
76
|
+
end
|
77
|
+
end
|
78
|
+
|
79
|
+
def generate_random_string
|
80
|
+
# random string test
|
81
|
+
str = " " * LEN
|
82
|
+
for i in 0...LEN
|
83
|
+
str[i] = (rand * 256).to_i.chr
|
84
|
+
end
|
85
|
+
|
86
|
+
return str
|
87
|
+
end
|
88
|
+
end
|
@@ -0,0 +1,10 @@
|
|
1
|
+
class TestPostgresPing < DBDConfig.testbase(:postgresql)
|
2
|
+
def test_ping
|
3
|
+
config = DBDConfig.get_config['postgresql']
|
4
|
+
dbh = DBI.connect("dbi:Pg:#{config['dbname']}", config['username'], config['password'])
|
5
|
+
assert dbh
|
6
|
+
assert dbh.ping
|
7
|
+
dbh.disconnect
|
8
|
+
assert_raises(DBI::InterfaceError) { dbh.ping }
|
9
|
+
end
|
10
|
+
end
|
@@ -0,0 +1,77 @@
|
|
1
|
+
class TestPostgresTimestamp < DBDConfig.testbase(:postgresql)
|
2
|
+
def get_tstamp
|
3
|
+
DateTime.parse("2008-03-08 10:39:01.012300")
|
4
|
+
end
|
5
|
+
|
6
|
+
# XXX DateTime is not allowed to modify the sec_fraction component, so this test is moot.
|
7
|
+
# perhaps we need to fix DateTime.
|
8
|
+
def skip_test_timestamp_altered_fraction
|
9
|
+
ts = nil
|
10
|
+
|
11
|
+
assert_nothing_raised do
|
12
|
+
@sth = @dbh.prepare("insert into timestamp_test (mytimestamp) values (?)")
|
13
|
+
ts = DateTime.parse(Time.now.to_s)
|
14
|
+
ts.sec_fraction = 22200000
|
15
|
+
@sth.execute(ts)
|
16
|
+
@sth.finish
|
17
|
+
end
|
18
|
+
|
19
|
+
assert_nothing_raised do
|
20
|
+
@sth = @dbh.prepare("select * from timestamp_test")
|
21
|
+
@sth.execute
|
22
|
+
row = @sth.fetch
|
23
|
+
@sth.finish
|
24
|
+
assert_equal ts.sec_fraction, row[0].sec_fraction
|
25
|
+
end
|
26
|
+
end
|
27
|
+
|
28
|
+
def test_current_timestamp
|
29
|
+
assert @dbh
|
30
|
+
# syntax db-specific (e.g., "from dual", "...timestamp()", etc.)
|
31
|
+
ts = @dbh.select_one("SELECT CURRENT_TIMESTAMP")[0]
|
32
|
+
assert_kind_of DateTime, ts
|
33
|
+
assert_not_nil ts.sec_fraction
|
34
|
+
end
|
35
|
+
|
36
|
+
# Just like the 'general' test, but checking for fractional seconds
|
37
|
+
def test_timestamp_fractional
|
38
|
+
assert @dbh
|
39
|
+
@sth = nil
|
40
|
+
t = get_tstamp
|
41
|
+
assert_nothing_raised do
|
42
|
+
@sth = @dbh.prepare("insert into timestamp_test (mytimestamp) values (?)")
|
43
|
+
@sth.execute(t)
|
44
|
+
@sth.finish
|
45
|
+
end
|
46
|
+
|
47
|
+
assert_nothing_raised do
|
48
|
+
@sth = @dbh.prepare("select * from timestamp_test")
|
49
|
+
@sth.execute
|
50
|
+
row = @sth.fetch
|
51
|
+
assert_kind_of DateTime, row[0]
|
52
|
+
assert_equal t.year, row[0].year
|
53
|
+
assert_equal t.month, row[0].month
|
54
|
+
assert_equal t.day, row[0].day
|
55
|
+
assert_equal t.hour, row[0].hour
|
56
|
+
assert_equal t.min, row[0].min
|
57
|
+
assert_equal t.sec, row[0].sec
|
58
|
+
assert_not_nil row[0].sec_fraction
|
59
|
+
assert_equal t.sec_fraction, row[0].sec_fraction
|
60
|
+
@sth.finish
|
61
|
+
end
|
62
|
+
end
|
63
|
+
|
64
|
+
# Is our DBI::Timestamp equivalent to its canonical string literal
|
65
|
+
# form cast appropriately?
|
66
|
+
def test_timestamp_from_cast
|
67
|
+
assert @dbh
|
68
|
+
sql_ts = "SELECT CAST('2008-03-08 10:39:01.012300' AS TIMESTAMP)"
|
69
|
+
|
70
|
+
row = @dbh.select_one(sql_ts)
|
71
|
+
assert_not_nil row
|
72
|
+
assert_equal 1, row.size
|
73
|
+
|
74
|
+
assert_kind_of DateTime, row[0]
|
75
|
+
assert_equal row[0], get_tstamp
|
76
|
+
end
|
77
|
+
end
|
@@ -0,0 +1,58 @@
|
|
1
|
+
class TestPostgresTransaction < DBDConfig.testbase(:postgresql)
|
2
|
+
def test_rollback
|
3
|
+
dbh = get_dbh
|
4
|
+
dbh["AutoCommit"] = false
|
5
|
+
@sth = dbh.prepare('insert into names (name, age) values (?, ?)')
|
6
|
+
@sth.execute("Foo", 51)
|
7
|
+
dbh.rollback
|
8
|
+
assert_equal 1, @sth.rows
|
9
|
+
@sth.finish
|
10
|
+
|
11
|
+
|
12
|
+
@sth = dbh.prepare('select name, age from names where name=?')
|
13
|
+
@sth.execute("Foo")
|
14
|
+
assert !@sth.fetch
|
15
|
+
@sth.finish
|
16
|
+
dbh.disconnect
|
17
|
+
end
|
18
|
+
|
19
|
+
def test_commit
|
20
|
+
dbh = get_dbh
|
21
|
+
dbh["AutoCommit"] = false
|
22
|
+
@sth = dbh.prepare('insert into names (name, age) values (?, ?)')
|
23
|
+
@sth.execute("Foo", 51)
|
24
|
+
dbh.commit
|
25
|
+
assert_equal 1, @sth.rows
|
26
|
+
@sth.finish
|
27
|
+
|
28
|
+
@sth = dbh.prepare('select name, age from names where name=?')
|
29
|
+
@sth.execute("Foo")
|
30
|
+
row = @sth.fetch
|
31
|
+
assert row
|
32
|
+
assert_equal "Foo", row[0]
|
33
|
+
assert_equal 51, row[1]
|
34
|
+
@sth.finish
|
35
|
+
dbh.disconnect
|
36
|
+
end
|
37
|
+
|
38
|
+
def test_select_transaction
|
39
|
+
# per bug #10466
|
40
|
+
dbh = get_dbh
|
41
|
+
dbh["AutoCommit"] = false
|
42
|
+
@sth = dbh.prepare('select * from test_insert(?, ?)');
|
43
|
+
@sth.execute("Foo", 51)
|
44
|
+
dbh.rollback
|
45
|
+
@sth.finish
|
46
|
+
|
47
|
+
@sth = dbh.prepare('select name, age from names where name=?')
|
48
|
+
@sth.execute("Foo")
|
49
|
+
assert !@sth.fetch
|
50
|
+
@sth.finish
|
51
|
+
dbh.disconnect
|
52
|
+
end
|
53
|
+
|
54
|
+
def get_dbh
|
55
|
+
config = DBDConfig.get_config
|
56
|
+
DBI.connect("dbi:Pg:#{config['postgresql']['dbname']}", config['postgresql']['username'], config['postgresql']['password'])
|
57
|
+
end
|
58
|
+
end
|