Sha256: 06fde5efee17f7c504f041073b63fadbe5f5bb752a19df1cb9f602b4f8664414
Contents?: true
Size: 979 Bytes
Versions: 7
Compression:
Stored size: 979 Bytes
Contents
# encoding: utf-8
MockServer = Class.new
class ServerMock
include Celluloid::IO
finalizer :finalize
def initialize(host, port, mock_target = MockServer.new)
puts "*** Starting echo server on #{host}:#{port}"
@server = TCPServer.new host, port
@mock_target = mock_target
@clients = []
async.run
end
def finalize
Logger.debug "ServerMock finalizing"
@server.close if @server
@clients.each(&:close)
end
def run
after(0.5) { terminate }
loop { async.handle_connection @server.accept }
end
def handle_connection(socket)
@clients << socket
_, port, host = socket.peeraddr
puts "*** Received connection from #{host}:#{port}"
loop { receive_data socket.readpartial(4096) }
end
def receive_data(data)
Logger.debug "ServerMock receiving data: #{data}"
@mock_target.receive_data data, self
end
def send_data(data)
@clients.each { |client| client.write data.gsub("\n", "\r\n") }
end
end
Version data entries
7 entries across 7 versions & 1 rubygems