c27de6d506
Fixes the error "Puma 5 is not compatible with Rack 3, please upgrade to Puma 6 or higher."
46 lines
933 B
Ruby
46 lines
933 B
Ruby
# frozen_string_literal: true
|
|
|
|
require 'puma'
|
|
require 'logger'
|
|
require 'puma/binder'
|
|
require 'puma/events'
|
|
|
|
class EchoServer
|
|
def call(env)
|
|
@socket = Faye::WebSocket.new(env, ['echo'])
|
|
|
|
@socket.onmessage = lambda do |event|
|
|
@socket.send(event.data)
|
|
end
|
|
|
|
@socket.rack_response
|
|
end
|
|
|
|
def send(message)
|
|
@socket.send(message)
|
|
end
|
|
|
|
def close(code, reason)
|
|
@socket.close(code, reason)
|
|
end
|
|
|
|
def log(*args); end
|
|
|
|
def listen(port)
|
|
# Instead of logging to the stdout/stderr, we'll log to a StringIO to prevent cluttering the test output
|
|
silent_stream = StringIO.new
|
|
logger = Puma::LogWriter.new(silent_stream, silent_stream)
|
|
events = Puma::Events.new
|
|
binder = Puma::Binder.new(logger)
|
|
binder.parse(["tcp://0.0.0.0:#{port}"], logger)
|
|
|
|
@server = Puma::Server.new(self, events)
|
|
@server.binder = binder
|
|
@server.run
|
|
end
|
|
|
|
def stop
|
|
@server&.stop(true)
|
|
end
|
|
end
|