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
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
|
# frozen_string_literal: true
require "cases/helper"
require "models/entrant"
require "models/bird"
require "models/course"
class MultipleDbTest < ActiveRecord::TestCase
self.use_transactional_tests = false
def setup
@courses = create_fixtures("courses")
@colleges = create_fixtures("colleges")
@entrants = create_fixtures("entrants")
end
def test_connected
assert_not_nil Entrant.lease_connection
assert_not_nil Course.lease_connection
end
def test_proper_connection
assert_not_equal(Entrant.lease_connection, Course.lease_connection)
assert_equal(Entrant.lease_connection, Entrant.retrieve_connection)
assert_equal(Course.lease_connection, Course.retrieve_connection)
assert_equal(ActiveRecord::Base.lease_connection, Entrant.lease_connection)
end
def test_swapping_the_connection
old_spec_name, Course.connection_specification_name = Course.connection_specification_name, "ActiveRecord::Base"
assert_equal(Entrant.lease_connection, Course.lease_connection)
ensure
Course.connection_specification_name = old_spec_name
end
def test_find
c1 = Course.find(1)
assert_equal "Ruby Development", c1.name
c2 = Course.find(2)
assert_equal "Java Development", c2.name
e1 = Entrant.find(1)
assert_equal "Ruby Developer", e1.name
e2 = Entrant.find(2)
assert_equal "Ruby Guru", e2.name
e3 = Entrant.find(3)
assert_equal "Java Lover", e3.name
end
def test_associations
c1 = Course.find(1)
assert_equal 2, c1.entrants.count
e1 = Entrant.find(1)
assert_equal e1.course.id, c1.id
c2 = Course.find(2)
assert_equal 1, c2.entrants.count
e3 = Entrant.find(3)
assert_equal e3.course.id, c2.id
end
def test_course_connection_should_survive_reloads
assert Course.lease_connection
assert Object.send(:remove_const, :Course)
assert load("models/course.rb")
assert Course.lease_connection
end
def test_transactions_across_databases
c1 = Course.find(1)
e1 = Entrant.find(1)
begin
Course.transaction do
Entrant.transaction do
c1.name = "Typo"
e1.name = "Typo"
c1.save
e1.save
raise "No I messed up."
end
end
rescue
# Yup caught it
end
assert_equal "Typo", c1.name
assert_equal "Typo", e1.name
assert_equal "Ruby Development", Course.find(1).name
assert_equal "Ruby Developer", Entrant.find(1).name
end
def test_connection
assert_same Entrant.lease_connection, Bird.lease_connection
assert_not_same Entrant.lease_connection, Course.lease_connection
end
unless in_memory_db?
def test_count_on_custom_connection
assert_equal ARUnit2Model.lease_connection, College.lease_connection
assert_not_equal ActiveRecord::Base.lease_connection, College.lease_connection
assert_equal 1, College.count
end
def test_associations_should_work_when_model_has_no_connection
assert_nothing_raised do
College.first.courses.first
end
end
end
def test_exception_contains_connection_pool
error = assert_raises(ActiveRecord::StatementInvalid) do
Course.where(wrong_column: "wrong").first!
end
assert_equal Course.lease_connection.pool, error.connection_pool
end
def test_exception_contains_correct_pool
course_conn = Course.lease_connection
entrant_conn = Entrant.lease_connection
assert_not_equal course_conn, entrant_conn
course_error = assert_raises(ActiveRecord::StatementInvalid) do
course_conn.execute("SELECT * FROM entrants")
end
assert_equal course_conn.pool, course_error.connection_pool
entrant_error = assert_raises(ActiveRecord::StatementInvalid) do
entrant_conn.execute("SELECT * FROM courses")
end
assert_equal entrant_conn.pool, entrant_error.connection_pool
end
end
|
