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
|
require_relative "spec_helper"
describe "Dataset#select_remove" do
before do
@d = Sequel.mock.from(:test).extension(:select_remove)
@d.columns :a, :b, :c
end
it "should remove columns from the selected columns" do
@d.sql.must_equal 'SELECT * FROM test'
@d.select_remove(:a).sql.must_equal 'SELECT b, c FROM test'
@d.select_remove(:b).sql.must_equal 'SELECT a, c FROM test'
@d.select_remove(:c).sql.must_equal 'SELECT a, b FROM test'
end
it "should work correctly if there are already columns selected" do
d = @d.select(:a, :b, :c)
d.columns :a, :b, :c
d.select_remove(:c).sql.must_equal 'SELECT a, b FROM test'
end
it "should have no effect if the columns given are not currently selected" do
@d.select_remove(:d).sql.must_equal 'SELECT a, b, c FROM test'
end
it "should handle expressions where Sequel can't determine the alias by itself" do
d = @d.select(:a, Sequel.function(:b), Sequel.as(:c, :b))
d.columns :a, :"b()", :b
d.select_remove(:"b()").sql.must_equal 'SELECT a, c AS b FROM test'
end
it "should remove expressions if given exact expressions" do
d = @d.select(:a, Sequel.function(:b), Sequel.as(:c, :b))
d.columns :a, :"b()", :b
d.select_remove(Sequel.function(:b)).sql.must_equal 'SELECT a, c AS b FROM test'
d.select_remove(Sequel.as(:c, :b)).sql.must_equal 'SELECT a, b() FROM test'
end
end
|
