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
|
# frozen_string_literal: true
require "cases/helper"
require "models/post"
require "models/comment"
require "models/project"
require "models/developer"
require "models/computer"
require "models/company_in_module"
class AssociationsExtensionsTest < ActiveRecord::TestCase
fixtures :projects, :developers, :developers_projects, :comments, :posts
def test_extension_on_has_many
assert_equal comments(:more_greetings), posts(:welcome).comments.find_most_recent
end
def test_extension_on_habtm
assert_equal projects(:action_controller), developers(:david).projects.find_most_recent
end
def test_named_extension_on_habtm
assert_equal projects(:action_controller), developers(:david).projects_extended_by_name.find_most_recent
end
def test_named_two_extensions_on_habtm
assert_equal projects(:action_controller), developers(:david).projects_extended_by_name_twice.find_most_recent
assert_equal projects(:active_record), developers(:david).projects_extended_by_name_twice.find_least_recent
end
def test_named_extension_and_block_on_habtm
assert_equal projects(:action_controller), developers(:david).projects_extended_by_name_and_block.find_most_recent
assert_equal projects(:active_record), developers(:david).projects_extended_by_name_and_block.find_least_recent
end
def test_extension_with_scopes
assert_equal comments(:greetings), posts(:welcome).comments.offset(1).find_most_recent
assert_equal comments(:greetings), posts(:welcome).comments.not_again.find_most_recent
end
def test_extension_with_dirty_target
comment = posts(:welcome).comments.build(body: "New comment")
assert_equal comment, posts(:welcome).comments.with_content("New comment")
end
def test_marshalling_extensions
david = developers(:david)
assert_equal projects(:action_controller), david.projects.find_most_recent
marshalled = Marshal.dump(david)
# Marshaling an association shouldn't make it unusable by wiping its reflection.
assert_not_nil david.association(:projects).reflection
david_too = Marshal.load(marshalled)
assert_equal projects(:action_controller), david_too.projects.find_most_recent
end
def test_marshalling_named_extensions
david = developers(:david)
assert_equal projects(:action_controller), david.projects_extended_by_name.find_most_recent
marshalled = Marshal.dump(david)
david = Marshal.load(marshalled)
assert_equal projects(:action_controller), david.projects_extended_by_name.find_most_recent
end
def test_extension_name
extend!(Developer)
extend!(MyApplication::Business::Developer)
assert Object.const_get "DeveloperAssociationNameAssociationExtension"
assert MyApplication::Business.const_get "DeveloperAssociationNameAssociationExtension"
end
def test_proxy_association_after_scoped
post = posts(:welcome)
assert_equal post.association(:comments), post.comments.the_association
assert_equal post.association(:comments), post.comments.where("1=1").the_association
end
def test_association_with_default_scope
assert_raises OopsError do
posts(:welcome).comments.destroy_all
end
end
private
def extend!(model)
ActiveRecord::Associations::Builder::HasMany.define_extensions(model, :association_name) {}
end
end
|
