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
140
141
142
143
144
145
146
147
148
149
150
151
152
153
|
require_relative "spec_helper"
describe "Sequel::Plugins::UpdatePrimaryKey" do
before do
@c = Class.new(Sequel::Model(:a))
@c.plugin :update_primary_key
@c.columns :a, :b
def @c.set_dataset(*)
super
set_primary_key :a
end
@c.set_primary_key :a
@c.unrestrict_primary_key
@ds = @c.dataset
DB.reset
end
it "should handle regular updates" do
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3}], [{:a=>1, :b=>4}], [{:a=>1, :b=>4}], [{:a=>1, :b=>5}], [{:a=>1, :b=>5}], [{:a=>1, :b=>6}], [{:a=>1, :b=>6}]])
@c.first.update(:b=>4)
@c.all.must_equal [@c.load(:a=>1, :b=>4)]
DB.sqls.must_equal ["SELECT * FROM a LIMIT 1", "UPDATE a SET b = 4 WHERE (a = 1)", "SELECT * FROM a"]
@c.first.set(:b=>5).save
@c.all.must_equal [@c.load(:a=>1, :b=>5)]
DB.sqls.must_equal ["SELECT * FROM a LIMIT 1", "UPDATE a SET b = 5 WHERE (a = 1)", "SELECT * FROM a"]
@c.first.set(:b=>6).save(:columns=>:b)
@c.all.must_equal [@c.load(:a=>1, :b=>6)]
DB.sqls.must_equal ["SELECT * FROM a LIMIT 1", "UPDATE a SET b = 6 WHERE (a = 1)", "SELECT * FROM a"]
end
it "should handle updating the primary key field with another field" do
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3}], [{:a=>2, :b=>4}]])
@c.first.update(:a=>2, :b=>4)
@c.all.must_equal [@c.load(:a=>2, :b=>4)]
sqls = DB.sqls
["UPDATE a SET a = 2, b = 4 WHERE (a = 1)", "UPDATE a SET b = 4, a = 2 WHERE (a = 1)"].must_include(sqls.slice!(1))
sqls.must_equal ["SELECT * FROM a LIMIT 1", "SELECT * FROM a"]
end
it "should handle updating the primary key field with another field when using composite keys" do
@c = Class.new(Sequel::Model(:a))
@c.plugin :update_primary_key
@c.columns :a, :b, :c
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3, :c=>5}], [{:a=>2, :b=>4, :c=>6}]])
@c.set_primary_key [:a, :b]
@c.unrestrict_primary_key
DB.reset
@c.first.update(:a=>2, :b=>4, :c=>6)
@c.all.must_equal [@c.load(:a=>2, :b=>4, :c=>6)]
DB.sqls.must_equal ["SELECT * FROM a LIMIT 1", "UPDATE a SET a = 2, b = 4, c = 6 WHERE ((a = 1) AND (b = 3))", "SELECT * FROM a"]
end
it "should handle updating just the primary key field when saving changes" do
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3}], [{:a=>2, :b=>3}], [{:a=>2, :b=>3}], [{:a=>3, :b=>3}]])
@c.first.update(:a=>2)
@c.all.must_equal [@c.load(:a=>2, :b=>3)]
DB.sqls.must_equal ["SELECT * FROM a LIMIT 1", "UPDATE a SET a = 2 WHERE (a = 1)", "SELECT * FROM a"]
@c.first.set(:a=>3).save(:columns=>:a)
@c.all.must_equal [@c.load(:a=>3, :b=>3)]
DB.sqls.must_equal ["SELECT * FROM a LIMIT 1", "UPDATE a SET a = 3 WHERE (a = 2)", "SELECT * FROM a"]
end
it "should handle saving after modifying the primary key field with another field" do
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3}], [{:a=>2, :b=>4}]])
@c.first.set(:a=>2, :b=>4).save
@c.all.must_equal [@c.load(:a=>2, :b=>4)]
sqls = DB.sqls
["UPDATE a SET a = 2, b = 4 WHERE (a = 1)", "UPDATE a SET b = 4, a = 2 WHERE (a = 1)"].must_include(sqls.slice!(1))
sqls.must_equal ["SELECT * FROM a LIMIT 1", "SELECT * FROM a"]
end
it "should handle saving after modifying just the primary key field" do
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3}], [{:a=>2, :b=>3}]])
@c.first.set(:a=>2).save
@c.all.must_equal [@c.load(:a=>2, :b=>3)]
sqls = DB.sqls
["UPDATE a SET a = 2, b = 3 WHERE (a = 1)", "UPDATE a SET b = 3, a = 2 WHERE (a = 1)"].must_include(sqls.slice!(1))
sqls.must_equal ["SELECT * FROM a LIMIT 1", "SELECT * FROM a"]
end
it "should handle saving after updating the primary key" do
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3}], [{:a=>2, :b=>5}]])
@c.first.update(:a=>2).update(:b=>4).set(:b=>5).save
@c.all.must_equal [@c.load(:a=>2, :b=>5)]
DB.sqls.must_equal ["SELECT * FROM a LIMIT 1", "UPDATE a SET a = 2 WHERE (a = 1)", "UPDATE a SET b = 4 WHERE (a = 2)", "UPDATE a SET b = 5 WHERE (a = 2)", "SELECT * FROM a"]
end
it "should work correctly when using the prepared_statements plugin" do
@c.plugin :prepared_statements
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3}], [{:a=>2, :b=>4}], [{:a=>3}]])
o = @c.first
o.update(:a=>2, :b=>4)
@c.all.must_equal [@c.load(:a=>2, :b=>4)]
sqls = DB.sqls
["UPDATE a SET a = 2, b = 4 WHERE (a = 1)", "UPDATE a SET b = 4, a = 2 WHERE (a = 1)"].must_include(sqls.slice!(1))
sqls.must_equal ["SELECT * FROM a LIMIT 1", "SELECT * FROM a"]
@c.create(:a=>3)
DB.sqls.must_equal ["INSERT INTO a (a) VALUES (3)", "SELECT * FROM a WHERE a = 3"]
end
it "should work correctly when loading the prepared_statements plugin first" do
@c = Class.new(Sequel::Model(:a))
@c.plugin :prepared_statements
@c.plugin :update_primary_key
@c.columns :a, :b
def @c.set_dataset(*)
super
set_primary_key :a
end
@c.set_primary_key :a
@c.unrestrict_primary_key
@ds = @c.dataset
DB.reset
@c.dataset = @c.dataset.with_fetch([[{:a=>1, :b=>3}], [{:a=>2, :b=>4}], [{:a=>3}]])
o = @c.first
o.update(:a=>2, :b=>4)
@c.all.must_equal [@c.load(:a=>2, :b=>4)]
sqls = DB.sqls
["UPDATE a SET a = 2, b = 4 WHERE (a = 1)", "UPDATE a SET b = 4, a = 2 WHERE (a = 1)"].must_include(sqls.slice!(1))
sqls.must_equal ["SELECT * FROM a LIMIT 1", "SELECT * FROM a"]
@c.create(:a=>3)
DB.sqls.must_equal ["INSERT INTO a (a) VALUES (3)", "SELECT * FROM a WHERE a = 3"]
end
it "should clear the associations cache of non-many_to_one associations when changing the primary key" do
@c.one_to_many :cs, :class=>@c
@c.many_to_one :c, :class=>@c
o = @c.new(:a=>1)
o.associations[:cs] = @c.new
o.associations[:c] = o2 = @c.new
o.a = 2
o.associations.must_equal(:c=>o2)
end
it "should clear the associations cache of non-many_to_one associations when changing the primary key" do
@c.one_to_many :cs, :class=>@c
@c.many_to_one :c, :class=>@c
o = @c.new(:a=>1)
o.associations[:cs] = @c.new
o.associations[:c] = o2 = @c.new
o.a = 2
o.associations.must_equal(:c=>o2)
end
it "should handle frozen instances" do
o = @c.new
o.a = 1
o.freeze
o.pk_hash.must_equal(:a=>1)
end
end
|
