<?php
namespace React\Tests\Socket;
use Clue\React\Block;
use React\EventLoop\Factory;
use React\Socket\TcpServer;
use React\Stream\DuplexResourceStream;
class TcpServerTest extends TestCase
{
private $loop;
private $server;
private $port;
private function createLoop()
{
return Factory::create();
}
/**
* @covers React\Socket\TcpServer::__construct
* @covers React\Socket\TcpServer::getAddress
*/
public function setUp()
{
$this->loop = $this->createLoop();
$this->server = new TcpServer(0, $this->loop);
$this->port = parse_url($this->server->getAddress(), PHP_URL_PORT);
}
/**
* @covers React\Socket\TcpServer::handleConnection
*/
public function testConnection()
{
$client = stream_socket_client('tcp://localhost:'.$this->port);
$this->server->on('connection', $this->expectCallableOnce());
$this->tick();
}
/**
* @covers React\Socket\TcpServer::handleConnection
*/
public function testConnectionWithManyClients()
{
$client1 = stream_socket_client('tcp://localhost:'.$this->port);
$client2 = stream_socket_client('tcp://localhost:'.$this->port);
$client3 = stream_socket_client('tcp://localhost:'.$this->port);
$this->server->on('connection', $this->expectCallableExactly(3));
$this->tick();
$this->tick();
$this->tick();
}
public function testDataEventWillNotBeEmittedWhenClientSendsNoData()
{
$client = stream_socket_client('tcp://localhost:'.$this->port);
$mock = $this->expectCallableNever();
$this->server->on('connection', function ($conn) use ($mock) {
$conn->on('data', $mock);
});
$this->tick();
$this->tick();
}
public function testDataWillBeEmittedWithDataClientSends()
{
$client = stream_socket_client('tcp://localhost:'.$this->port);
fwrite($client, "foo\n");
$mock = $this->expectCallableOnceWith("foo\n");
$this->server->on('connection', function ($conn) use ($mock) {
$conn->on('data', $mock);
});
$this->tick();
$this->tick();
}
public function testDataWillBeEmittedEvenWhenClientShutsDownAfterSending()
{
$client = stream_socket_client('tcp://localhost:' . $this->port);
fwrite($client, "foo\n");
stream_socket_shutdown($client, STREAM_SHUT_WR);
$mock = $this->expectCallableOnceWith("foo\n");
$this->server->on('connection', function ($conn) use ($mock) {
$conn->on('data', $mock);
});
$this->tick();
$this->tick();
}
public function testLoopWillEndWhenServerIsClosed()
{
// explicitly unset server because we already call close()
$this->server->close();
$this->server = null;
$this->loop->run();
// if we reach this, then everything is good
$this->assertNull(null);
}
public function testCloseTwiceIsNoOp()
{
$this->server->close();
$this->server->close();
// if we reach this, then everything is good
$this->assertNull(null);
}
public function testGetAddressAfterCloseReturnsNull()
{
$this->server->close();
$this->assertNull($this->server->getAddress());
}
public function testLoopWillEndWhenServerIsClosedAfterSingleConnection()
{
$client = stream_socket_client('tcp://localhost:' . $this->port);
// explicitly unset server because we only accept a single connection
// and then already call close()
$server = $this->server;
$this->server = null;
$server->on('connection', function ($conn) use ($server) {
$conn->close();
$server->close();
});
$this->loop->run();
// if we reach this, then everything is good
$this->assertNull(null);
}
public function testDataWillBeEmittedInMultipleChunksWhenClientSendsExcessiveAmounts()
{
$client = stream_socket_client('tcp://localhost:' . $this->port);
$stream = new DuplexResourceStream($client, $this->loop);
$bytes = 1024 * 1024;
$stream->end(str_repeat('*', $bytes));
$mock = $this->expectCallableOnce();
// explicitly unset server because we only accept a single connection
// and then already call close()
$server = $this->server;
$this->server = null;
$received = 0;
$server->on('connection', function ($conn) use ($mock, &$received, $server) {
// count number of bytes received
$conn->on('data', function ($data) use (&$received) {
$received += strlen($data);
});
$conn->on('end', $mock);
// do not await any further connections in order to let the loop terminate
$server->close();
});
$this->loop->run();
$this->assertEquals($bytes, $received);
}
public function testConnectionDoesNotEndWhenClientDoesNotClose()
{
$client = stream_socket_client('tcp://localhost:'.$this->port);
$mock = $this->expectCallableNever();
$this->server->on('connection', function ($conn) use ($mock) {
$conn->on('end', $mock);
});
$this->tick();
$this->tick();
}
/**
* @covers React\Socket\Connection::end
*/
public function testConnectionDoesEndWhenClientCloses()
{
$client = stream_socket_client('tcp://localhost:'.$this->port);
fclose($client);
$mock = $this->expectCallableOnce();
$this->server->on('connection', function ($conn) use ($mock) {
$conn->on('end', $mock);
});
$this->tick();
$this->tick();
}
public function testCtorAddsResourceToLoop()
{
$loop = $this->getMockBuilder('React\EventLoop\LoopInterface')->getMock();
$loop->expects($this->once())->method('addReadStream');
$server = new TcpServer(0, $loop);
}
public function testResumeWithoutPauseIsNoOp()
{
$loop = $this->getMockBuilder('React\EventLoop\LoopInterface')->getMock();
$loop->expects($this->once())->method('addReadStream');
$server = new TcpServer(0, $loop);
$server->resume();
}
public function testPauseRemovesResourceFromLoop()
{
$loop = $this->getMockBuilder('React\EventLoop\LoopInterface')->getMock();
$loop->expects($this->once())->method('removeReadStream');
$server = new TcpServer(0, $loop);
$server->pause();
}
public function testPauseAfterPauseIsNoOp()
{
$loop = $this->getMockBuilder('React\EventLoop\LoopInterface')->getMock();
$loop->expects($this->once())->method('removeReadStream');
$server = new TcpServer(0, $loop);
$server->pause();
$server->pause();
}
public function testCloseRemovesResourceFromLoop()
{
$loop = $this->getMockBuilder('React\EventLoop\LoopInterface')->getMock();
$loop->expects($this->once())->method('removeReadStream');
$server = new TcpServer(0, $loop);
$server->close();
}
/**
* @expectedException RuntimeException
*/
public function testListenOnBusyPortThrows()
{
if (DIRECTORY_SEPARATOR === '\\') {
$this->markTestSkipped('Windows supports listening on same port multiple times');
}
$another = new TcpServer($this->port, $this->loop);
}
/**
* @covers React\Socket\TcpServer::close
*/
public function tearDown()
{
if ($this->server) {
$this->server->close();
}
}
private function tick()
{
Block\sleep(0, $this->loop);
}
}