diff options
| -rw-r--r-- | lib/zero/response.rb | 2 | ||||
| -rw-r--r-- | spec/unit/response/response_spec.rb | 7 |
2 files changed, 8 insertions, 1 deletions
diff --git a/lib/zero/response.rb b/lib/zero/response.rb index cfd1148..2886a0d 100644 --- a/lib/zero/response.rb +++ b/lib/zero/response.rb @@ -42,7 +42,7 @@ module Zero # Also creates one, if it does not exists # def content_length - header['Content-Length'] = 0 + header['Content-Length'] = body.join.length end end diff --git a/spec/unit/response/response_spec.rb b/spec/unit/response/response_spec.rb index 0bb0c4e..21d6e31 100644 --- a/spec/unit/response/response_spec.rb +++ b/spec/unit/response/response_spec.rb @@ -53,6 +53,13 @@ describe Zero::Response do subject.header['Content-Length'].should eq(0) end + + it "sets the content_length to the size of the message body" do + subject.body = ['foo', 'bar'] + subject.content_length + + subject.header['Content-Length'].should eq(6) + end end end
\ No newline at end of file |