128 lines
4.5 KiB
PHP
128 lines
4.5 KiB
PHP
<?php
|
|
namespace Ratchet\Tests\Application\WebSocket\Version;
|
|
use Ratchet\Component\WebSocket\Version\RFC6455;
|
|
use Ratchet\Component\WebSocket\Version\RFC6455\Frame;
|
|
use Guzzle\Http\Message\RequestFactory;
|
|
|
|
/**
|
|
* @covers Ratchet\Component\WebSocket\Version\RFC6455
|
|
*/
|
|
class RFC6455Test extends \PHPUnit_Framework_TestCase {
|
|
protected $_version;
|
|
|
|
public function setUp() {
|
|
$this->_version = new RFC6455();
|
|
}
|
|
|
|
/**
|
|
* Is this useful?
|
|
*/
|
|
public function testClassImplementsVersionInterface() {
|
|
$constraint = $this->isInstanceOf('\\Ratchet\\Component\\WebSocket\\Version\\VersionInterface');
|
|
$this->assertThat($this->_version, $constraint);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider HandshakeProvider
|
|
*/
|
|
public function testKeySigningForHandshake($key, $accept) {
|
|
$this->assertEquals($accept, $this->_version->sign($key));
|
|
}
|
|
|
|
public static function HandshakeProvider() {
|
|
return array(
|
|
array('x3JJHMbDL1EzLkh9GBhXDw==', 'HSmrc0sMlYUkAGmm5OPpG2HaGWk=')
|
|
, array('dGhlIHNhbXBsZSBub25jZQ==', 's3pPLMBiTxaQ9kYGzzhZRbK+xOo=')
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider UnframeMessageProvider
|
|
*/
|
|
public function testUnframeMessage($message, $framed) {
|
|
$frame = new Frame;
|
|
$frame->addBuffer(base64_decode($framed));
|
|
|
|
$this->assertEquals($message, $frame->getPayload());
|
|
}
|
|
|
|
public static function UnframeMessageProvider() {
|
|
return array(
|
|
array('Hello World!', 'gYydAIfa1WXrtvIg0LXvbOP7')
|
|
, array('!@#$%^&*()-=_+[]{}\|/.,<>`~', 'gZv+h96r38f9j9vZ+IHWrvOWoayF9oX6gtfRqfKXwOeg')
|
|
, array('ಠ_ಠ', 'gYfnSpu5B/g75gf4Ow==')
|
|
, array("The quick brown fox jumps over the lazy dog. All work and no play makes Chris a dull boy. I'm trying to get past 128 characters for a unit test here...", 'gf4Amahb14P8M7Kj2S6+4MN7tfHHLLmjzjSvo8IuuvPbe7j1zSn398A+9+/JIa6jzDSwrYh7lu/Ee6Ds2jD34sY/9+3He6fvySL37skwsvCIGL/xwSj34og/ou/Ee7Xs0XX3o+F8uqPcKa7qxjz398d7sObce6fi2y/3sppj9+DAOqXiyy+y8dt7sezae7aj3TW+94gvsvDce7/m2j75rYY=')
|
|
);
|
|
}
|
|
|
|
public function testUnframeMatchesPreFraming() {
|
|
$string = 'Hello World!';
|
|
$framed = $this->_version->frame($string);
|
|
|
|
$frame = new Frame;
|
|
$frame->addBuffer($framed);
|
|
|
|
$this->assertEquals($string, $frame->getPayload());
|
|
}
|
|
|
|
public static $good_rest = 'GET /chat HTTP/1.1';
|
|
|
|
public static $good_header = array(
|
|
'Host' => 'server.example.com'
|
|
, 'Upgrade' => 'websocket'
|
|
, 'Connection' => 'Upgrade'
|
|
, 'Sec-WebSocket-Key' => 'dGhlIHNhbXBsZSBub25jZQ=='
|
|
, 'Origin' => 'http://example.com'
|
|
, 'Sec-WebSocket-Protocol' => 'chat, superchat'
|
|
, 'Sec-WebSocket-Version' => 13
|
|
);
|
|
|
|
/**
|
|
* A helper function to try and quickly put together a valid WebSocket HTTP handshake
|
|
* but optionally replace a piece to an invalid value for failure testing
|
|
*/
|
|
public static function getAndSpliceHeader($key = null, $val = null) {
|
|
$headers = static::$good_header;
|
|
|
|
if (null !== $key && null !== $val) {
|
|
$headers[$key] = $val;
|
|
}
|
|
|
|
$header = '';
|
|
foreach ($headers as $key => $val) {
|
|
if (!empty($key)) {
|
|
$header .= "{$key}: ";
|
|
}
|
|
|
|
$header .= "{$val}\r\n";
|
|
}
|
|
$header .= "\r\n";
|
|
|
|
return $header;
|
|
}
|
|
|
|
public static function headerHandshakeProvider() {
|
|
return array(
|
|
array(false, "GET /test HTTP/1.0\r\n" . static::getAndSpliceHeader())
|
|
, array(true, static::$good_rest . "\r\n" . static::getAndSpliceHeader())
|
|
, array(false, "POST / HTTP:/1.1\r\n" . static::getAndSpliceHeader())
|
|
, array(false, static::$good_rest . "\r\n" . static::getAndSpliceHeader('Upgrade', 'useless'))
|
|
, array(false, "GET /ಠ_ಠ HTTP/1.1\r\n" . static::getAndSpliceHeader())
|
|
, array(true, static::$good_rest . "\r\n" . static::getAndSpliceHeader('Connection', 'Herp, Upgrade, Derp'))
|
|
);
|
|
}
|
|
|
|
/**
|
|
* @dataProvider headerHandshakeProvider
|
|
*/
|
|
public function testVariousHeadersToCheckHandshakeTolerance($pass, $header) {
|
|
$request = RequestFactory::fromMessage($header);
|
|
|
|
if ($pass) {
|
|
$this->assertTrue(is_array($this->_version->handshake($request)));
|
|
} else {
|
|
$this->setExpectedException('InvalidArgumentException');
|
|
$this->_version->handshake($request);
|
|
}
|
|
}
|
|
} |