yii2/tests/unit/framework/web/ResponseTest.php

<?php

namespace yii\web;

use yiiunit\framework\web\ResponseTest;

/**
 * Mock PHP header function to check for sent headers
 * @param string $string
 * @param bool $replace
 * @param int $httpResponseCode
 */
function header($string, $replace = true, $httpResponseCode = null) {
	ResponseTest::$headers[] = $string;
	// TODO implement replace

	if ($httpResponseCode !== null) {
		ResponseTest::$httpResponseCode = $httpResponseCode;
	}
}

namespace yiiunit\framework\web;

use yii\helpers\StringHelper;
use yii\web\Response;

class ResponseTest extends \yiiunit\TestCase
{
	public static $headers = array();
	public static $httpResponseCode = 200;

	protected function setUp()
	{
		parent::setUp();
		$this->mockApplication();
		$this->reset();
	}

	protected function reset()
	{
		static::$headers = array();
		static::$httpResponseCode = 200;
	}

	public function rightRanges()
	{
		// TODO test more cases for range requests and check for rfc compatibility
		// http://www.w3.org/Protocols/rfc2616/rfc2616.txt
		return array(
			array('0-5', '0-5', 6, '12ёж'),
			array('2-', '2-66', 65, 'ёжик3456798áèabcdefghijklmnopqrstuvwxyz!"§$%&/(ёжик)=?'),
			array('-12', '55-66', 12, '(ёжик)=?'),
		);
	}

	/**
	 * @dataProvider rightRanges
	 */
	public function testSendFileRanges($rangeHeader, $expectedHeader, $length, $expectedFile)
	{
		$content = $this->generateTestFileContent();
		$_SERVER['HTTP_RANGE'] = 'bytes=' . $rangeHeader;
		$sent = $this->runSendFile('testFile.txt', $content, null);

		$this->assertEquals($expectedFile, $sent);
		$this->assertTrue(in_array('HTTP/1.1 206 Partial Content', static::$headers));
		$this->assertTrue(in_array('Accept-Ranges: bytes', static::$headers));
		$this->assertArrayHasKey('Content-Range: bytes ' . $expectedHeader . '/' . StringHelper::strlen($content), array_flip(static::$headers));
		$this->assertTrue(in_array('Content-Type: text/plain', static::$headers));
		$this->assertTrue(in_array('Content-Length: ' . $length, static::$headers));
	}

	public function wrongRanges()
	{
		// TODO test more cases for range requests and check for rfc compatibility
		// http://www.w3.org/Protocols/rfc2616/rfc2616.txt
		return array(
			array('1-2,3-5,6-10'),	// multiple range request not supported
			array('5-1'),			// last-byte-pos value is less than its first-byte-pos value
			array('-100000'),		// last-byte-pos bigger then content length
			array('10000-'),			// first-byte-pos bigger then content length
		);
	}

	/**
	 * @dataProvider wrongRanges
	 */
	public function testSendFileWrongRanges($rangeHeader)
	{
		$this->setExpectedException('yii\base\HttpException', 'Requested Range Not Satisfiable');

		$content = $this->generateTestFileContent();
		$_SERVER['HTTP_RANGE'] = 'bytes=' . $rangeHeader;
		$this->runSendFile('testFile.txt', $content, null);
	}

	protected function generateTestFileContent()
	{
		return '12ёжик3456798áèabcdefghijklmnopqrstuvwxyz!"§$%&/(ёжик)=?';
	}

	protected function runSendFile($fileName, $content, $mimeType)
	{
		ob_start();
		ob_implicit_flush(false);
		$response = new Response();
		$response->sendFile($fileName, $content, $mimeType, false);
		$file = ob_get_clean();
		return $file;
	}
}