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
|
require_relative "spec_helper"
describe "Sequel::Plugins::InsertReturningSelect" do
before do
@db = Sequel.mock(:fetch=>{:id=>1, :x=>2}, :autoid=>1)
@db.extend_datasets do
def supports_returning?(_) true end
def insert_select(*v) with_sql_first("#{insert_sql(*v)}#{" RETURNING #{opts[:returning].map{|x| literal(x)}.join(', ')}" if opts[:returning]}") end
end
@Album = Class.new(Sequel::Model(@db[:albums].select(:id, :x)))
@Album.columns :id, :x
@db.sqls
end
it "should work if loaded into a model without a dataset that also uses the lazy_attributes or dataset associations plugins" do
c = Sequel::Model(@db)
c.plugin :insert_returning_select
c.columns :id, :x
c.plugin :lazy_attributes
c.plugin :dataset_associations
c.set_dataset @db[:albums].select(:id, :x)
c.plugin :lazy_attributes, :x
c.many_to_one :c, :key=>:id, :class=>c
@db.sqls
c.dataset.sql.must_equal 'SELECT id FROM albums'
c.create(:x=>2)
@db.sqls.must_equal ['INSERT INTO albums (x) VALUES (2) RETURNING id']
c.load(:id=>2).x
@db.sqls.must_equal ['SELECT albums.x FROM albums WHERE (id = 2) LIMIT 1']
c.dataset.cs.sql.must_equal "SELECT id FROM albums WHERE (albums.id IN (SELECT albums.id FROM albums))"
end
it "should add a returning clause when inserting using selected columns" do
@Album.plugin :insert_returning_select
@Album.create(:x=>2).must_equal @Album.load(:id=>1, :x=>2)
@db.sqls.must_equal ['INSERT INTO albums (x) VALUES (2) RETURNING id, x']
end
it "should not add a returning clause if no columns are selected" do
@Album.plugin :insert_returning_select
@Album.dataset = @Album.dataset.select_all
@db.sqls.clear
@Album.create(:x=>2).must_equal @Album.load(:id=>1, :x=>2)
@db.sqls.must_equal ['INSERT INTO albums (x) VALUES (2)']
end
it "should not add a returning clause if selection does not consist of just columns" do
@Album.dataset = @Album.dataset.select_append(Sequel.as(1, :b))
@Album.plugin :insert_returning_select
@db.sqls.clear
@Album.create(:x=>2).must_equal @Album.load(:id=>1, :x=>2)
@db.sqls.must_equal ['INSERT INTO albums (x) VALUES (2)', 'SELECT id, x, 1 AS b FROM albums WHERE (id = 1) LIMIT 1']
end
it "should not add a returning clause if database doesn't support it" do
@db.extend_datasets{def supports_returning?(_) false end}
@Album.plugin :insert_returning_select
@Album.create(:x=>2).must_equal @Album.load(:id=>1, :x=>2)
@db.sqls.must_equal ['INSERT INTO albums (x) VALUES (2)', 'SELECT id, x FROM albums WHERE (id = 1) LIMIT 1']
end
it "should work correctly with subclasses" do
c = Class.new(Sequel::Model)
c.plugin :insert_returning_select
b = Class.new(c)
b.columns :id, :x
b.dataset = @db[:albums].select(:id, :x)
@db.sqls.clear
b.create(:x=>2).must_equal b.load(:id=>1, :x=>2)
@db.sqls.must_equal ['INSERT INTO albums (x) VALUES (2) RETURNING id, x']
end
it "should freeze instance_insert_dataset when freezing model class" do
@Album.plugin :insert_returning_select
@Album.freeze
@Album.instance_insert_dataset.frozen?.must_equal true
end
end
|
