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
|
# 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") { Course.retrieve_connection }
@colleges = create_fixtures("colleges") { College.retrieve_connection }
@entrants = create_fixtures("entrants")
end
def test_connected
assert_not_nil Entrant.connection
assert_not_nil Course.connection
end
def test_proper_connection
assert_not_equal(Entrant.connection, Course.connection)
assert_equal(Entrant.connection, Entrant.retrieve_connection)
assert_equal(Course.connection, Course.retrieve_connection)
assert_equal(ActiveRecord::Base.connection, Entrant.connection)
end
def test_swapping_the_connection
old_spec_name, Course.connection_specification_name = Course.connection_specification_name, "primary"
assert_equal(Entrant.connection, Course.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_dependency_reload
assert Course.connection
ActiveSupport::Dependencies.clear
Object.send(:remove_const, :Course)
require_dependency "models/course"
assert Course.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_arel_table_engines
assert_not_equal Entrant.arel_engine, Bird.arel_engine
assert_not_equal Entrant.arel_engine, Course.arel_engine
end
def test_connection
assert_equal Entrant.arel_engine.connection.object_id, Bird.arel_engine.connection.object_id
assert_not_equal Entrant.arel_engine.connection.object_id, Course.arel_engine.connection.object_id
end
unless in_memory_db?
def test_count_on_custom_connection
ActiveRecord::Base.remove_connection
assert_equal 1, College.count
ensure
ActiveRecord::Base.establish_connection :arunit
end
def test_associations_should_work_when_model_has_no_connection
begin
ActiveRecord::Base.remove_connection
assert_nothing_raised do
College.first.courses.first
end
ensure
ActiveRecord::Base.establish_connection :arunit
end
end
end
end
|
