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
|
require_relative "spec_helper"
describe "Sequel::Plugins::AssociationProxies" do
before do
class ::Tag < Sequel::Model
end
class ::Item < Sequel::Model
plugin :association_proxies
many_to_many :tags, :extend=>Module.new{def size; count end}
end
@i = Item.load(:id=>1)
@t = @i.tags
Item.db.reset
end
after do
Object.send(:remove_const, :Tag)
Object.send(:remove_const, :Item)
end
it "should send method calls to the associated object array if sent an array method" do
@i.associations.has_key?(:tags).must_equal false
@t.select{|x| false}.must_equal []
@i.associations.has_key?(:tags).must_equal true
end
if RUBY_VERSION < '2.6'
deprecated "should issue deprecation warning when using filter on association proxy on ruby <2.6" do
@i.associations.has_key?(:tags).must_equal false
@t.filter{|x| false}.sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND 'f')"
@i.associations.has_key?(:tags).must_equal false
end
else
it "should treat filter on association proxy as array method on ruby 2.6+" do
@i.associations.has_key?(:tags).must_equal false
@t.filter{|x| false}.must_equal []
@i.associations.has_key?(:tags).must_equal true
end
end
it "should send method calls to the association dataset if sent a non-array method" do
@i.associations.has_key?(:tags).must_equal false
@t.where(:a=>1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
@i.associations.has_key?(:tags).must_equal false
end
it "should accept block to plugin to specify which methods to proxy to dataset" do
Item.plugin :association_proxies do |opts|
opts[:method] == :where || opts[:arguments].first.is_a?(Sequel::LiteralString) || opts[:block]
end
@i.associations.has_key?(:tags).must_equal false
@t.where(:a=>1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
@t.where(Sequel.lit('a = 1')).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
@t.where{{:a=>1}}.sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
@i.associations.has_key?(:tags).must_equal false
Item.plugin :association_proxies do |opts|
proxy_arg = opts[:proxy_argument]
proxy_block = opts[:proxy_block]
cached = opts[:instance].associations[opts[:reflection][:name]]
is_size = opts[:method] == :size
is_size && !cached && !proxy_arg[:reload] && !proxy_block
end
@t.size.must_equal 1
Item.db.sqls.must_equal ["SELECT count(*) AS count FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE (items_tags.item_id = 1) LIMIT 1"]
@i.tags{|ds| ds}.size.must_equal 1
Item.db.sqls.must_equal ["SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE (items_tags.item_id = 1)"]
@i.tags(:reload=>true).size.must_equal 1
Item.db.sqls.must_equal ["SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE (items_tags.item_id = 1)"]
@t.size.must_equal 1
Item.db.sqls.must_equal []
end
it "should reload the cached association if sent an array method and the reload flag was given" do
@t.select{|x| false}.must_equal []
Item.db.sqls.length.must_equal 1
@t.select{|x| false}.must_equal []
Item.db.sqls.length.must_equal 0
@i.tags(:reload=>true).select{|x| false}.must_equal []
Item.db.sqls.length.must_equal 1
@t.where(:a=>1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
Item.db.sqls.length.must_equal 0
end
it "should not return a proxy object for associations that do not return an array" do
Item.many_to_one :tag
proc{@i.tag.where(:a=>1)}.must_raise(NoMethodError)
Tag.one_to_one :item
proc{Tag.load(:id=>1, :item_id=>2).item.where(:a=>1)}.must_raise(NoMethodError)
end
it "should work correctly in subclasses" do
i = Class.new(Item).load(:id=>1)
i.associations.has_key?(:tags).must_equal false
i.tags.select{|x| false}.must_equal []
i.associations.has_key?(:tags).must_equal true
i.tags.where(:a=>1).sql.must_equal "SELECT tags.* FROM tags INNER JOIN items_tags ON (items_tags.tag_id = tags.id) WHERE ((items_tags.item_id = 1) AND (a = 1))"
end
if RUBY_VERSION >= '2.7'
it "should handle keywords when delegating" do
Tag.send(:define_method, :to_int){1}
s = String.new
@t.pack('i', buffer: s)
s.wont_be_empty
end
end
end
|
