1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
require_relative "spec_helper"
describe "transaction_connection_validator extension" do
database_error = Class.new(StandardError)
before do
@db = Sequel.mock
@m = Module.new do
def post_execute(conn, sql); end
def disconnect_connection(conn)
@sqls << 'disconnect'
end
def connect(server)
@sqls << 'connect'
super
end
private
define_method(:database_error_classes) do
[database_error]
end
def disconnect_error?(e, opts)
e.message.include? 'disconnect'
end
def log_connection_execute(conn, sql)
res = super
post_execute(conn, sql)
res
end
end
@db.extend @m
@db.extension(:transaction_connection_validator)
end
it "should not affect transactions that do not raise exceptions" do
@db.transaction{}
@db.sqls.must_equal ['BEGIN', 'COMMIT']
end
it "should retry transactions for disconnects during BEGIN" do
conns = []
@db.define_singleton_method(:post_execute) do |conn, sql|
conns << conn
raise database_error, "disconnect error" if @sqls == ['BEGIN']
end
@db.transaction{}
@db.sqls.must_equal ['BEGIN', 'ROLLBACK', 'disconnect', 'connect', 'BEGIN', 'COMMIT']
conns.uniq.size.must_equal 2
end
it "should handle DatabaseDisconnectErrors as disconnects" do
conns = []
@db.define_singleton_method(:post_execute) do |conn, sql|
conns << conn
raise Sequel::DatabaseDisconnectError if @sqls == ['BEGIN']
end
@db.transaction{}
@db.sqls.must_equal ['BEGIN', 'ROLLBACK', 'disconnect', 'connect', 'BEGIN', 'COMMIT']
conns.uniq.size.must_equal 2
end
it "should not retry if a connection has already been checked out before calling transaction" do
conns = []
@db.define_singleton_method(:post_execute) do |conn, sql|
conns << conn
raise Sequel::DatabaseDisconnectError if @sqls == ['BEGIN']
end
c = nil
proc do
@db.synchronize do |c1|
c = c1
@db.transaction{}
end
end.must_raise(Sequel::DatabaseDisconnectError)
@db.sqls.must_equal ['BEGIN', 'ROLLBACK', 'disconnect']
conns.uniq.must_equal [c]
end
it "should not retry transaction setup more than 5 times" do
conns = []
@db.define_singleton_method(:post_execute) do |conn, sql|
conns << conn
raise database_error, "disconnect error"
end
proc do
@db.transaction{}
end.must_raise(Sequel::DatabaseDisconnectError)
@db.sqls.must_equal(['BEGIN', 'ROLLBACK', 'disconnect', 'connect'] * 5)
conns.uniq.size.must_equal 5
end
it "should not retry on non-disconnect errors" do
conns = []
@db.define_singleton_method(:post_execute) do |conn, sql|
conns << conn
raise database_error, "normal error" if @sqls == ['BEGIN']
end
proc{@db.transaction{}}.must_raise(Sequel::DatabaseError)
@db.sqls.must_equal ['BEGIN', 'ROLLBACK']
conns.uniq.size.must_equal 1
end
end
|
