mirror of
https://github.com/valitydev/thrift.git
synced 2024-11-07 18:58:51 +00:00
5d6bd5a606
Client: rb Patch: Nathan Beyer The spec_helper.rb defines an Object#tee method, which is functionally equivalent to Object#tap. Object#tap was added to Ruby 1.9 and to 1.8.7. git-svn-id: https://svn.apache.org/repos/asf/thrift/trunk@1392509 13f79535-47bb-0310-9956-ffa450edef68
72 lines
2.5 KiB
Ruby
72 lines
2.5 KiB
Ruby
#
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
#
|
|
|
|
require 'spec_helper'
|
|
|
|
describe 'Thrift::HTTPClientTransport' do
|
|
|
|
describe Thrift::HTTPClientTransport do
|
|
before(:each) do
|
|
@client = Thrift::HTTPClientTransport.new("http://my.domain.com/path/to/service?param=value")
|
|
end
|
|
|
|
it "should always be open" do
|
|
@client.should be_open
|
|
@client.close
|
|
@client.should be_open
|
|
end
|
|
|
|
it "should post via HTTP and return the results" do
|
|
@client.write "a test"
|
|
@client.write " frame"
|
|
Net::HTTP.should_receive(:new).with("my.domain.com", 80).and_return do
|
|
mock("Net::HTTP").tap do |http|
|
|
http.should_receive(:use_ssl=).with(false)
|
|
http.should_receive(:post).with("/path/to/service?param=value", "a test frame", {"Content-Type"=>"application/x-thrift"}).and_return do
|
|
mock("Net::HTTPOK").tap do |response|
|
|
response.should_receive(:body).and_return "data"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
@client.flush
|
|
@client.read(10).should == "data"
|
|
end
|
|
|
|
it "should send custom headers if defined" do
|
|
@client.write "test"
|
|
custom_headers = {"Cookie" => "Foo"}
|
|
headers = {"Content-Type"=>"application/x-thrift"}.merge(custom_headers)
|
|
|
|
@client.add_headers(custom_headers)
|
|
Net::HTTP.should_receive(:new).with("my.domain.com", 80).and_return do
|
|
mock("Net::HTTP").tap do |http|
|
|
http.should_receive(:use_ssl=).with(false)
|
|
http.should_receive(:post).with("/path/to/service?param=value", "test", headers).and_return do
|
|
mock("Net::HTTPOK").tap do |response|
|
|
response.should_receive(:body).and_return "data"
|
|
end
|
|
end
|
|
end
|
|
end
|
|
@client.flush
|
|
end
|
|
end
|
|
end
|