47 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			47 lines
		
	
	
		
			1.2 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
| <?php
 | |
| namespace Ratchet\Http;
 | |
| use Ratchet\AbstractMessageComponentTestCase;
 | |
| 
 | |
| /**
 | |
|  * @covers Ratchet\Http\OriginCheck
 | |
|  */
 | |
| class OriginCheckTest extends AbstractMessageComponentTestCase {
 | |
|     protected $_reqStub;
 | |
| 
 | |
|     public function setUp() {
 | |
|         $this->_reqStub = $this->getMock('Psr\Http\Message\RequestInterface');
 | |
|         $this->_reqStub->expects($this->any())->method('getHeader')->will($this->returnValue(['localhost']));
 | |
| 
 | |
|         parent::setUp();
 | |
| 
 | |
|         $this->_serv->allowedOrigins[] = 'localhost';
 | |
|     }
 | |
| 
 | |
|     protected function doOpen($conn) {
 | |
|         $this->_serv->onOpen($conn, $this->_reqStub);
 | |
|     }
 | |
| 
 | |
|     public function getConnectionClassString() {
 | |
|         return '\Ratchet\ConnectionInterface';
 | |
|     }
 | |
| 
 | |
|     public function getDecoratorClassString() {
 | |
|         return '\Ratchet\Http\OriginCheck';
 | |
|     }
 | |
| 
 | |
|     public function getComponentClassString() {
 | |
|         return '\Ratchet\Http\HttpServerInterface';
 | |
|     }
 | |
| 
 | |
|     public function testCloseOnNonMatchingOrigin() {
 | |
|         $this->_serv->allowedOrigins = ['socketo.me'];
 | |
|         $this->_conn->expects($this->once())->method('close');
 | |
| 
 | |
|         $this->_serv->onOpen($this->_conn, $this->_reqStub);
 | |
|     }
 | |
| 
 | |
|     public function testOnMessage() {
 | |
|         $this->passthroughMessageTest('Hello World!');
 | |
|     }
 | |
| }
 | 
