|
| 1 | +# frozen_string_literal: true |
| 2 | +require 'spec_helper' |
| 3 | +require 'rack' |
| 4 | +require 'rack/lint' |
| 5 | +require 'rack/rewindable_input' |
| 6 | +if Gem::Version.new(Rack.release) < Gem::Version.new('3') |
| 7 | + require 'rack/handler/webrick' |
| 8 | +else |
| 9 | + require 'rackup/handler/webrick' |
| 10 | +end |
| 11 | + |
| 12 | +RSpec.describe Slack::Events::Request do |
| 13 | + let(:signing_secret) { 'test-signing-secret' } |
| 14 | + let(:timestamp) { Time.now.to_i.to_s } |
| 15 | + let(:body) { '{"type":"url_verification","challenge":"hello"}' } |
| 16 | + let(:signature) do |
| 17 | + "v0=#{OpenSSL::HMAC.hexdigest('SHA256', signing_secret, "v0:#{timestamp}:#{body}")}" |
| 18 | + end |
| 19 | + let(:consume_body) { false } |
| 20 | + let(:buffer_body) { false } |
| 21 | + let(:observed_inputs) { [] } |
| 22 | + let(:app) do |
| 23 | + Rack::Lint.new(lambda do |env| |
| 24 | + observed_inputs << env.fetch('rack.input') |
| 25 | + env['rack.input'] = Rack::RewindableInput.new(env['rack.input']) if buffer_body |
| 26 | + env['rack.input'].read if consume_body |
| 27 | + slack_request = described_class.new(Rack::Request.new(env), signing_secret: signing_secret) |
| 28 | + begin |
| 29 | + slack_request.verify! |
| 30 | + [200, { 'content-type' => 'application/json' }, [slack_request.body, slack_request.body]] |
| 31 | + rescue Slack::Events::Request::InvalidSignature |
| 32 | + [401, { 'content-type' => 'text/plain' }, ['invalid signature']] |
| 33 | + ensure |
| 34 | + env['rack.input'].close if buffer_body |
| 35 | + end |
| 36 | + end) |
| 37 | + end |
| 38 | + let(:server) { WEBrick::HTTPServer.new(DoNotListen: true, Logger: WEBrick::Log.new(File::NULL), AccessLog: []) } |
| 39 | + |
| 40 | + def http_request |
| 41 | + raw_request = [ |
| 42 | + 'POST /events HTTP/1.1', |
| 43 | + 'Host: localhost', |
| 44 | + 'Content-Type: application/json', |
| 45 | + "Content-Length: #{body.bytesize}", |
| 46 | + "X-Slack-Request-Timestamp: #{timestamp}", |
| 47 | + "X-Slack-Signature: #{signature}", |
| 48 | + '', body |
| 49 | + ].join("\r\n") |
| 50 | + WEBrick::HTTPRequest.new(server.config).tap { |request| request.parse(StringIO.new(raw_request)) } |
| 51 | + end |
| 52 | + |
| 53 | + after do |
| 54 | + server.shutdown |
| 55 | + end |
| 56 | + |
| 57 | + def dispatch |
| 58 | + response = WEBrick::HTTPResponse.new(server.config) |
| 59 | + handler = Gem::Version.new(Rack.release) < Gem::Version.new('3') ? Rack::Handler::WEBrick : Rackup::Handler::WEBrick |
| 60 | + handler.new(server, app).service(http_request, response) |
| 61 | + response |
| 62 | + end |
| 63 | + |
| 64 | + it 'verifies a signed WEBrick request and caches its complete body' do |
| 65 | + http_response = dispatch |
| 66 | + expect(http_response.status).to eq 200 |
| 67 | + expect(http_response.body).to eq body * 2 |
| 68 | + expect(observed_inputs.first.respond_to?(:rewind)).to eq(Gem::Version.new(Rack.release) < Gem::Version.new('3')) |
| 69 | + end |
| 70 | + |
| 71 | + context 'with an invalid signature' do |
| 72 | + let(:signature) { 'v0=invalid' } |
| 73 | + |
| 74 | + it 'rejects the request' do |
| 75 | + http_response = dispatch |
| 76 | + expect(http_response.status).to eq 401 |
| 77 | + end |
| 78 | + end |
| 79 | + |
| 80 | + context 'with input already read by another consumer' do |
| 81 | + let(:consume_body) { true } |
| 82 | + |
| 83 | + it 'only verifies the request if the server input can be rewound' do |
| 84 | + http_response = dispatch |
| 85 | + expected_status = Gem::Version.new(Rack.release) < Gem::Version.new('3') ? 200 : 401 |
| 86 | + expect(http_response.status).to eq expected_status |
| 87 | + end |
| 88 | + |
| 89 | + context 'with input buffering before the first consumer' do |
| 90 | + let(:buffer_body) { true } |
| 91 | + |
| 92 | + it 'verifies the request and preserves the full body' do |
| 93 | + http_response = dispatch |
| 94 | + expect(http_response.status).to eq 200 |
| 95 | + expect(http_response.body).to eq body * 2 |
| 96 | + end |
| 97 | + end |
| 98 | + end |
| 99 | +end |
0 commit comments