Current File : //opt/cpanel/ea-ruby27/src/passenger-release-6.0.23/test/ruby/utils/tee_input_spec.rb
# encoding: binary
require File.expand_path(File.dirname(__FILE__) + '/../spec_helper')
PhusionPassenger.require_passenger_lib 'utils/tee_input'
require 'stringio'
module PhusionPassenger
shared_examples "TeeInput#gets" do
context "if Content-Length is given" do
before :each do
init_input("hello\nworld\nlast line!", "CONTENT_LENGTH" => 21)
end
it "reads a line, including newline character" do
@input.gets.should == "hello\n"
@input.gets.should == "world\n"
end
it "reads until EOF if no newline is encountered before EOF" do
@input.gets
@input.gets
@input.gets.should == "last line"
end
it "returns nil if EOF is reached" do
@input.gets
@input.gets
@input.gets
@input.gets.should be_nil
end
end
context "if Transfer-Encoding is chunked" do
before :each do
init_input("hello\nworld\nlast line", "HTTP_TRANSFER_ENCODING" => "chunked")
end
it "reads a line, including newline character" do
@input.gets.should == "hello\n"
@input.gets.should == "world\n"
end
it "reads until EOF if no newline is encountered before EOF" do
@input.gets
@input.gets
@input.gets.should == "last line"
end
it "returns nil if EOF is reached" do
@input.gets
@input.gets
@input.gets
@input.gets.should be_nil
end
end
context "if neither Content-Length nor Transfer-Encoding chunked are given" do
before :each do
init_input("hello\nworld\nlast line")
end
it "returns nil" do
@input.gets.should be_nil
end
end
end
shared_examples "TeeInput#read" do
describe "with no arguments" do
context "if Content-Length is given" do
before :each do
init_input("hello!", "CONTENT_LENGTH" => 5)
end
it "slurps up to Content-Length bytes from the socket" do
@input.read.should == "hello"
end
it "returns the empty string if EOF is reached" do
@input.read
@input.read.should == ""
end
end
context "if Transfer-Encoding is chunked" do
before :each do
init_input("hello!", "HTTP_TRANSFER_ENCODING" => "chunked")
end
it "slurps the entire socket" do
@input.read.should == "hello!"
end
it "returns the empty string if EOF is reached" do
@input.read
@input.read.should == ""
end
end
context "if neither Content-Length nor Transfer-Encoding chunked are given" do
before :each do
init_input("hello!")
end
it "returns the empty string" do
@input.read.should == ""
end
end
end
describe "with a length argument" do
it "raises ArgumentError if len < 0" do
init_input("")
lambda { @input.read(-1) }.should raise_error(ArgumentError)
end
it "returns the empty string if len == 0" do
init_input("hello", "HTTP_TRANSFER_ENCODING" => "chunked")
@input.read(0).should == ""
end
context "if Content-Length is given" do
before :each do
init_input("hello!", "CONTENT_LENGTH" => 5)
end
it "reads exactly len bytes if available" do
@input.read(2).should == "he"
@input.read(2).should == "ll"
end
it "reads at most Content-Length bytes if Content-Length < len are available" do
@input.read(2).should == "he"
@input.read(4).should == "llo"
end
it "returns nil if Content-Length is reached" do
@input.read(5).should == "hello"
@input.read(1).should be_nil
end
end
context "if Transfer-Encoding is chunked" do
before :each do
init_input("hello", "HTTP_TRANSFER_ENCODING" => "chunked")
end
it "reads exactly len bytes if available" do
@input.read(2).should == "he"
@input.read(2).should == "ll"
end
it "reads until EOF if less than len bytes are available" do
@input.read(2).should == "he"
@input.read(4).should == "llo"
end
it "returns nil if EOF is reached" do
@input.read(5).should == "hello"
@input.read(1).should be_nil
end
end
context "if neither Content-Length nor Transfer-Encoding chunked are given" do
it "returns nil" do
init_input("hello")
@input.read(2).should be_nil
end
end
end
end
shared_examples "TeeInput#size" do
context "if Content-Length is given" do
it "returns the value in Content-Length" do
init_input("hello world", "CONTENT_LENGTH" => 10)
@input.size.should == 10
end
end
context "if Transfer-Encoding is chunked" do
it "returns the number of bytes that can be read from the socket until EOF" do
init_input("hello world", "HTTP_TRANSFER_ENCODING" => "chunked")
@input.size.should == 11
end
end
context "if neither Content-Length nor Transfer-Encoding chunked are given" do
it "returns 0" do
init_input("hello world")
@input.size.should == 0
end
end
end
describe Utils::TeeInput do
before :each do
@sock, @sock2 = UNIXSocket.pair
end
after :each do
[@sock, @sock2].each do |sock|
sock.close if sock && !sock.closed?
end
@input.close if @input
end
context "when unbuffered" do
def init_input(data, env = {})
@input = Utils::TeeInput.new(@sock2, env)
@sock.write(data)
@sock.close
end
describe("#gets") { include_examples "TeeInput#gets" }
describe("#read") { include_examples "TeeInput#read" }
describe("#size") { include_examples "TeeInput#size" }
end
context "when buffered" do
def init_input(data, env = {})
@input = Utils::TeeInput.new(@sock2, env)
@sock.write(data)
@sock.close
@input.read
@input.rewind
end
describe("#gets") { include_examples "TeeInput#gets" }
describe("#read") { include_examples "TeeInput#read" }
describe("#size") { include_examples "TeeInput#size" }
end
end
end # module PhusionPassenger
Mini Shell