2017-11-24 12:31:23 +08:00
|
|
|
require 'rails_helper'
|
|
|
|
|
|
|
|
|
|
describe Hijack do
|
2017-11-27 14:43:24 +08:00
|
|
|
class Hijack::Tester < ApplicationController
|
2017-11-24 12:31:23 +08:00
|
|
|
attr_reader :io
|
|
|
|
|
|
|
|
|
|
include Hijack
|
2017-11-27 14:43:24 +08:00
|
|
|
|
2017-11-28 13:47:20 +08:00
|
|
|
def initialize(env = {})
|
2017-11-24 12:31:23 +08:00
|
|
|
@io = StringIO.new
|
2017-11-28 13:47:20 +08:00
|
|
|
|
|
|
|
|
env.merge!(
|
2017-11-28 06:28:40 +08:00
|
|
|
"rack.hijack" => lambda { @io },
|
|
|
|
|
"rack.input" => StringIO.new
|
2017-11-24 12:31:23 +08:00
|
|
|
)
|
2017-11-28 13:47:20 +08:00
|
|
|
|
|
|
|
|
self.request = ActionController::TestRequest.new(env, nil, nil)
|
|
|
|
|
|
2017-11-27 14:43:24 +08:00
|
|
|
# we need this for the 418
|
|
|
|
|
self.response = ActionDispatch::Response.new
|
2017-11-24 12:31:23 +08:00
|
|
|
end
|
|
|
|
|
|
2017-11-27 14:43:24 +08:00
|
|
|
def hijack_test(&blk)
|
|
|
|
|
hijack(&blk)
|
2017-11-24 12:31:23 +08:00
|
|
|
end
|
2017-11-27 14:43:24 +08:00
|
|
|
|
2017-11-24 12:31:23 +08:00
|
|
|
end
|
|
|
|
|
|
|
|
|
|
let :tester do
|
|
|
|
|
Hijack::Tester.new
|
|
|
|
|
end
|
|
|
|
|
|
2017-11-28 13:47:20 +08:00
|
|
|
context "Request Tracker integration" do
|
|
|
|
|
let :logger do
|
|
|
|
|
lambda do |env, data|
|
|
|
|
|
@calls += 1
|
|
|
|
|
@status = data[:status]
|
|
|
|
|
@total = data[:timing][:total_duration]
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
before do
|
|
|
|
|
Middleware::RequestTracker.register_detailed_request_logger logger
|
|
|
|
|
@calls = 0
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
after do
|
|
|
|
|
Middleware::RequestTracker.unregister_detailed_request_logger logger
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
it "can properly track execution" do
|
|
|
|
|
app = lambda do |env|
|
|
|
|
|
tester = Hijack::Tester.new(env)
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
render body: "hello", status: 201
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
env = {}
|
|
|
|
|
middleware = Middleware::RequestTracker.new(app)
|
|
|
|
|
|
|
|
|
|
middleware.call(env)
|
|
|
|
|
|
|
|
|
|
expect(@calls).to eq(1)
|
|
|
|
|
expect(@status).to eq(201)
|
|
|
|
|
expect(@status).to be > 0
|
|
|
|
|
end
|
|
|
|
|
end
|
|
|
|
|
|
2017-11-28 06:28:40 +08:00
|
|
|
it "dupes the request params and env" do
|
|
|
|
|
orig_req = tester.request
|
|
|
|
|
copy_req = nil
|
|
|
|
|
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
copy_req = request
|
|
|
|
|
render body: "hello world", status: 200
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
expect(copy_req.object_id).not_to eq(orig_req.object_id)
|
|
|
|
|
end
|
|
|
|
|
|
2017-11-27 14:43:24 +08:00
|
|
|
it "handles expires_in" do
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
expires_in 1.year
|
|
|
|
|
render body: "hello world", status: 402
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
expect(tester.io.string).to include("max-age=31556952")
|
|
|
|
|
end
|
|
|
|
|
|
2017-11-24 12:31:23 +08:00
|
|
|
it "renders non 200 status if asked for" do
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
render body: "hello world", status: 402
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
expect(tester.io.string).to include("402")
|
|
|
|
|
expect(tester.io.string).to include("world")
|
|
|
|
|
end
|
|
|
|
|
|
2017-11-28 07:59:53 +08:00
|
|
|
it "handles send_file correctly" do
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
send_file __FILE__, disposition: nil
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
expect(tester.io.string).to start_with("HTTP/1.1 200")
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
it "renders a redirect correctly" do
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
redirect_to 'http://awesome.com'
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
result = "HTTP/1.1 302 Found\r\nLocation: http://awesome.com\r\nContent-Type: text/html\r\nContent-Length: 84\r\nConnection: close\r\n\r\n<html><body>You are being <a href=\"http://awesome.com\">redirected</a>.</body></html>"
|
|
|
|
|
expect(tester.io.string).to eq(result)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
it "renders stuff correctly if is empty" do
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
render body: nil
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
result = "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\nContent-Length: 0\r\nConnection: close\r\n\r\n"
|
|
|
|
|
expect(tester.io.string).to eq(result)
|
|
|
|
|
end
|
|
|
|
|
|
2017-11-24 12:31:23 +08:00
|
|
|
it "renders stuff correctly if it works" do
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
render plain: "hello world"
|
|
|
|
|
end
|
|
|
|
|
|
2017-11-27 14:43:24 +08:00
|
|
|
result = "HTTP/1.1 200 OK\r\nContent-Type: text/plain\r\nContent-Length: 11\r\nConnection: close\r\n\r\nhello world"
|
2017-11-24 12:31:23 +08:00
|
|
|
expect(tester.io.string).to eq(result)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
it "returns 500 by default" do
|
|
|
|
|
tester.hijack_test
|
|
|
|
|
|
2017-11-27 14:43:24 +08:00
|
|
|
expected = "HTTP/1.1 500 Internal Server Error\r\nContent-Type: text/html\r\nContent-Length: 0\r\nConnection: close\r\n\r\n"
|
2017-11-24 12:31:23 +08:00
|
|
|
expect(tester.io.string).to eq(expected)
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
it "does not run the block if io is closed" do
|
|
|
|
|
tester.io.close
|
|
|
|
|
|
|
|
|
|
ran = false
|
|
|
|
|
tester.hijack_test do
|
|
|
|
|
ran = true
|
|
|
|
|
end
|
|
|
|
|
|
|
|
|
|
expect(ran).to eq(false)
|
|
|
|
|
end
|
|
|
|
|
end
|
