silverstripe-framework/tests/php/Control/HTTPResponseTest.php

<?php

namespace SilverStripe\Control\Tests;

use SilverStripe\Dev\SapphireTest;
use SilverStripe\Control\HTTPResponse;
use SilverStripe\Control\HTTPResponse_Exception;

class HTTPResponseTest extends SapphireTest
{

    public function testStatusDescriptionStripsNewlines()
    {
        $r = new HTTPResponse('my body', 200, "my description \nwith newlines \rand carriage returns");
        $this->assertEquals(
            "my description with newlines and carriage returns",
            $r->getStatusDescription()
        );
    }

    public function testHTTPResponseException()
    {
        $response = new HTTPResponse("Test", 200, 'OK');

        // Confirm that the exception's statusCode and statusDescription take precedence
        try {
            throw new HTTPResponse_Exception($response, 404, 'not even found');
        } catch (HTTPResponse_Exception $e) {
            $this->assertEquals(404, $e->getResponse()->getStatusCode());
            $this->assertEquals('not even found', $e->getResponse()->getStatusDescription());
            return;
        }
        // Fail if we get to here
        $this->assertFalse(true, 'Something went wrong with our test exception');
    }

    public function testExceptionContentPlainByDefault()
    {

        // Confirm that the exception's statusCode and statusDescription take precedence
        try {
            throw new HTTPResponse_Exception("Some content that may be from a hacker", 404, 'not even found');
        } catch (HTTPResponse_Exception $e) {
            $this->assertEquals("text/plain", $e->getResponse()->getHeader("Content-Type"));
            return;
        }
        // Fail if we get to here
        $this->assertFalse(true, 'Something went wrong with our test exception');
    }

    public function testRemoveHeader()
    {
        $response = new HTTPResponse();

        $response->addHeader('X-Animal', 'Monkey');
        $this->assertSame('Monkey', $response->getHeader('X-Animal'));

        $response->removeHeader('X-Animal');
        $this->assertEmpty($response->getHeader('X-Animal'));
    }
}