summaryrefslogtreecommitdiff
path: root/spec/ruby/library/net/http/httpresponse/read_body_spec.rb
blob: 99d8e5fa88aae58ede697e64f3bf3edab5ba974b (plain)
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
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
require_relative '../../../../spec_helper'
require 'net/http'

describe "Net::HTTPResponse#read_body" do
  before :each do
    @res = Net::HTTPUnknownResponse.new("1.0", "???", "test response")
    @socket = Net::BufferedIO.new(StringIO.new("test body"))
  end

  describe "when passed no arguments" do
    it "returns the read body" do
      @res.reading_body(@socket, true) do
        @res.read_body.should == "test body"
      end
    end

    it "returns the previously read body if called a second time" do
      @res.reading_body(@socket, true) do
        @res.read_body.should equal(@res.read_body)
      end
    end
  end

  describe "when passed a buffer" do
    it "reads the body to the passed buffer" do
      @res.reading_body(@socket, true) do
        buffer = ""
        @res.read_body(buffer)
        buffer.should == "test body"
      end
    end

    it "returns the passed buffer" do
      @res.reading_body(@socket, true) do
        buffer = ""
        @res.read_body(buffer).should equal(buffer)
      end
    end

    it "raises an IOError if called a second time" do
      @res.reading_body(@socket, true) do
        @res.read_body("")
        -> { @res.read_body("") }.should raise_error(IOError)
      end
    end
  end

  describe "when passed a block" do
    it "reads the body and yields it to the passed block (in chunks)" do
      @res.reading_body(@socket, true) do
        yielded = false

        buffer = ""
        @res.read_body do |body|
          yielded = true
          buffer << body
        end

        yielded.should be_true
        buffer.should == "test body"
      end
    end

    it "returns the ReadAdapter" do
      @res.reading_body(@socket, true) do
        @res.read_body { nil }.should be_kind_of(Net::ReadAdapter)
      end
    end

    it "raises an IOError if called a second time" do
      @res.reading_body(@socket, true) do
        @res.read_body {}
        -> { @res.read_body {} }.should raise_error(IOError)
      end
    end
  end

  describe "when passed buffer and block" do
    it "raises an ArgumentError" do
      @res.reading_body(@socket, true) do
        -> { @res.read_body("") {} }.should raise_error(ArgumentError)
      end
    end
  end
end