class Pretender::Rails::Middleware

Public Class Methods

new(app) click to toggle source
# File lib/pretender/rails/middleware.rb, line 4
def initialize(app)
  @app = app
end

Public Instance Methods

call(env) click to toggle source
# File lib/pretender/rails/middleware.rb, line 8
def call(env)
  status, headers, env = @app.call(env)
  if headers['Content-Type'].to_s.include?("text/html")
    body = Pretender.server.inject(env.body)
    headers['Content-Length'] = Rack::Utils.bytesize(body.to_s).to_s
    return [status, headers, [body]]
  end

  [status, headers, env]
end