72 lines
2.2 KiB
Ruby
72 lines
2.2 KiB
Ruby
require_relative '../sql_api.rb'
|
||
require_relative '../../../spec/rspec_configuration.rb'
|
||
|
||
module Cartodb; end
|
||
|
||
describe CartoDB::SQLApi do
|
||
|
||
before(:each) do
|
||
Cartodb.stubs(:config).returns(TEST_SQL_API_CONFIG)
|
||
end
|
||
|
||
describe '#fetch' do
|
||
|
||
let(:api) { CartoDB::SQLApi.new(username: 'maloshumos') }
|
||
|
||
it "returns an array of rows" do
|
||
stub_api_request 200, 'sql_api_success.json'
|
||
result = api.fetch("SELECT cartodb_id, description from public_table")
|
||
result.should eq [{"cartodb_id"=>1, "description"=>"a"}, {"cartodb_id"=>2, "description"=>"b"}, {"cartodb_id"=>3, "description"=>"c"}, {"cartodb_id"=>4, "description"=>"d"}]
|
||
end
|
||
|
||
it "raises PermissionError when the table is private" do
|
||
stub_api_request 400, 'sql_api_private.json'
|
||
expect { api.fetch("SELECT * FROM private_table") }.to raise_error(CartoDB::SQLApi::PermissionError)
|
||
end
|
||
|
||
it "raises SQLError when the query is flawed" do
|
||
stub_api_request 400, 'sql_api_error.json'
|
||
expect { api.fetch("wrong query") }.to raise_error(CartoDB::SQLApi::SQLError)
|
||
end
|
||
|
||
it "handles gzipped output" do
|
||
stub_api_request 200, 'sql_api_binary.bin'
|
||
result = api.fetch("SELECT description from public_table", 'csv')
|
||
result.should match /description,\r\n\"Pretend that you’ve opened this book/
|
||
result.should match /And if not, then the onion will make it all happen for you.*$/
|
||
end
|
||
|
||
end #fetch
|
||
|
||
|
||
def stub_api_request(code, response_file)
|
||
response = File.open(path_to(response_file)).read
|
||
Typhoeus.stub(/.*carto.com\/api\/v[12]/).and_return(
|
||
Typhoeus::Response.new(code: code, body: response)
|
||
)
|
||
end # stub_api_request
|
||
|
||
def path_to(filepath = '')
|
||
File.expand_path(
|
||
File.join(File.dirname(__FILE__), "../spec/fixtures/#{filepath}")
|
||
)
|
||
end #path_to
|
||
|
||
TEST_SQL_API_CONFIG = {:sql_api => {
|
||
"private" => {
|
||
"protocol" => 'http',
|
||
"domain" => 'carto.com',
|
||
"endpoint" => '/api/v1/sql',
|
||
"port" => 8080
|
||
},
|
||
"public" => {
|
||
"protocol" => 'http',
|
||
"domain" => 'carto.com',
|
||
"endpoint" => '/api/v2/sql',
|
||
"port" => 8080
|
||
}
|
||
}
|
||
}
|
||
|
||
end # CartoDB::SQLApi
|