mirror of
https://github.com/discourse/discourse.git
synced 2024-11-24 00:51:03 +08:00
70 lines
1.9 KiB
Ruby
70 lines
1.9 KiB
Ruby
# encoding: utf-8
|
|
require 'spec_helper'
|
|
require_dependency 'sql_builder'
|
|
|
|
describe SqlBuilder do
|
|
|
|
describe "attached" do
|
|
before do
|
|
@builder = Post.sql_builder("select * from posts /*where*/ /*limit*/")
|
|
end
|
|
|
|
it "should find a post by id" do
|
|
p = Fabricate(:post)
|
|
@builder.where('id = :id and topic_id = :topic_id', id: p.id, topic_id: p.topic_id)
|
|
p2 = @builder.exec.first
|
|
p2.id.should == p.id
|
|
p2.should == p
|
|
end
|
|
end
|
|
|
|
describe "map" do
|
|
class SqlBuilder::TestClass
|
|
attr_accessor :int, :string, :date, :text
|
|
end
|
|
|
|
it "correctly maps to a klass" do
|
|
rows = SqlBuilder.new("SELECT 1 AS int, 'string' AS string, CAST(NOW() at time zone 'utc' AS timestamp without time zone) AS date, 'text'::text AS text")
|
|
.map_exec(SqlBuilder::TestClass)
|
|
|
|
rows.count.should == 1
|
|
row = rows[0]
|
|
row.int.should == 1
|
|
row.string.should == "string"
|
|
row.text.should == "text"
|
|
row.date.should be_within(10.seconds).of(DateTime.now)
|
|
end
|
|
end
|
|
|
|
describe "detached" do
|
|
before do
|
|
@builder = SqlBuilder.new("select * from (select :a A union all select :b) as X /*where*/ /*order_by*/ /*limit*/ /*offset*/")
|
|
end
|
|
|
|
it "should allow for 1 param exec" do
|
|
@builder.exec(a: 1, b: 2).values[0][0].should == '1'
|
|
end
|
|
|
|
it "should allow for a single where" do
|
|
@builder.where(":a = 1")
|
|
@builder.exec(a: 1, b: 2).values[0][0].should == '1'
|
|
end
|
|
|
|
it "should allow where chaining" do
|
|
@builder.where(":a = 1")
|
|
@builder.where("2 = 1")
|
|
@builder.exec(a: 1, b: 2).to_a.length.should == 0
|
|
end
|
|
|
|
it "should allow order by" do
|
|
@builder.order_by("A desc").limit(1)
|
|
.exec(a:1, b:2).values[0][0].should == "2"
|
|
end
|
|
it "should allow offset" do
|
|
@builder.order_by("A desc").offset(1)
|
|
.exec(a:1, b:2).values[0][0].should == "1"
|
|
end
|
|
end
|
|
|
|
end
|