2024-03-18 22:26:00 +01:00
|
|
|
<?php
|
|
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
* Licensed to the Apache Software Foundation (ASF) under one
|
|
|
|
|
* or more contributor license agreements. See the NOTICE file
|
|
|
|
|
* distributed with this work for additional information
|
|
|
|
|
* regarding copyright ownership. The ASF licenses this file
|
|
|
|
|
* to you under the Apache License, Version 2.0 (the
|
|
|
|
|
* "License"); you may not use this file except in compliance
|
|
|
|
|
* with the License. You may obtain a copy of the License at
|
|
|
|
|
*
|
|
|
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
|
|
|
*
|
|
|
|
|
* Unless required by applicable law or agreed to in writing,
|
|
|
|
|
* software distributed under the License is distributed on an
|
|
|
|
|
* "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
|
|
|
* KIND, either express or implied. See the License for the
|
|
|
|
|
* specific language governing permissions and limitations
|
|
|
|
|
* under the License.
|
|
|
|
|
*/
|
|
|
|
|
|
|
|
|
|
namespace Test\Thrift\Unit\Lib\Server;
|
|
|
|
|
|
|
|
|
|
use phpmock\phpunit\PHPMock;
|
|
|
|
|
use PHPUnit\Framework\TestCase;
|
2026-03-21 20:45:04 +02:00
|
|
|
use Test\Thrift\Unit\Lib\ReflectionHelper;
|
2024-03-18 22:26:00 +01:00
|
|
|
use Thrift\Exception\TTransportException;
|
|
|
|
|
use Thrift\Server\TServerSocket;
|
|
|
|
|
use Thrift\Transport\TSocket;
|
|
|
|
|
|
|
|
|
|
class TServerSocketTest extends TestCase
|
|
|
|
|
{
|
|
|
|
|
use PHPMock;
|
2026-03-21 20:45:04 +02:00
|
|
|
use ReflectionHelper;
|
2024-03-18 22:26:00 +01:00
|
|
|
|
|
|
|
|
public function testSetAcceptTimeout(): void
|
|
|
|
|
{
|
|
|
|
|
$socket = new TServerSocket();
|
|
|
|
|
$socket->setAcceptTimeout(1000);
|
|
|
|
|
|
2026-03-21 20:45:04 +02:00
|
|
|
$this->assertEquals(1000, $this->getPropertyValue($socket, 'acceptTimeout_'));
|
2024-03-18 22:26:00 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public function testListenAndClose(): void
|
|
|
|
|
{
|
|
|
|
|
$socket = new TServerSocket('somehost', 999);
|
|
|
|
|
|
|
|
|
|
$listener = tmpfile();
|
|
|
|
|
$this->getFunctionMock('Thrift\Server', 'stream_socket_server')
|
|
|
|
|
->expects($this->once())
|
|
|
|
|
->with('tcp://somehost:999')
|
|
|
|
|
->willReturn($listener);
|
|
|
|
|
|
|
|
|
|
$socket->listen();
|
|
|
|
|
|
2026-03-21 20:45:04 +02:00
|
|
|
$this->assertIsResource($this->getPropertyValue($socket, 'listener_'));
|
2024-03-18 22:26:00 +01:00
|
|
|
|
|
|
|
|
$this->getFunctionMock('Thrift\Server', 'fclose')
|
|
|
|
|
->expects($this->once())
|
|
|
|
|
->with($this->equalTo($listener))
|
|
|
|
|
->willReturn(true);
|
|
|
|
|
|
|
|
|
|
$socket->close();
|
2026-03-21 20:45:04 +02:00
|
|
|
$this->assertNull($this->getPropertyValue($socket, 'listener_'));
|
2024-03-18 22:26:00 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public function testAccept()
|
|
|
|
|
{
|
|
|
|
|
$socket = new TServerSocket('somehost', 999);
|
|
|
|
|
$socket->setAcceptTimeout(1000);
|
|
|
|
|
|
|
|
|
|
$listener = tmpfile();
|
|
|
|
|
|
|
|
|
|
$this->getFunctionMock('Thrift\Server', 'stream_socket_server')
|
|
|
|
|
->expects($this->once())
|
|
|
|
|
->with('tcp://somehost:999')
|
|
|
|
|
->willReturn($listener);
|
|
|
|
|
|
|
|
|
|
$transportHandle = tmpfile();
|
|
|
|
|
$this->getFunctionMock('Thrift\Server', 'stream_socket_accept')
|
|
|
|
|
->expects($this->once())
|
|
|
|
|
->with(
|
|
|
|
|
$this->equalTo($listener),
|
|
|
|
|
1
|
|
|
|
|
)->willReturn($transportHandle);
|
|
|
|
|
|
|
|
|
|
$socket->listen();
|
|
|
|
|
$result = $socket->accept();
|
|
|
|
|
$this->assertInstanceOf(TSocket::class, $result);
|
|
|
|
|
|
2026-03-21 20:45:04 +02:00
|
|
|
$this->assertEquals($transportHandle, $this->getPropertyValue($result, 'handle_'));
|
2024-03-18 22:26:00 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
|
|
public function testAcceptFailed()
|
|
|
|
|
{
|
|
|
|
|
$socket = new TServerSocket('somehost', 999);
|
|
|
|
|
$socket->setAcceptTimeout(1000);
|
|
|
|
|
|
|
|
|
|
$listener = tmpfile();
|
|
|
|
|
|
|
|
|
|
$this->getFunctionMock('Thrift\Server', 'stream_socket_server')
|
|
|
|
|
->expects($this->once())
|
|
|
|
|
->with('tcp://somehost:999')
|
|
|
|
|
->willReturn($listener);
|
|
|
|
|
|
|
|
|
|
$this->getFunctionMock('Thrift\Server', 'stream_socket_accept')
|
|
|
|
|
->expects($this->once())
|
|
|
|
|
->with(
|
|
|
|
|
$this->equalTo($listener),
|
|
|
|
|
1
|
|
|
|
|
)->willReturn(null);
|
|
|
|
|
|
|
|
|
|
$this->expectException(TTransportException::class);
|
|
|
|
|
$this->expectExceptionMessage('accept() may not return NULL');
|
|
|
|
|
|
|
|
|
|
$socket->listen();
|
|
|
|
|
$socket->accept();
|
|
|
|
|
}
|
|
|
|
|
}
|
