ThrottledErrorTest.php

Go to the documentation of this file.

<?php
 
class ThrottledErrorTest extends MediaWikiTestCase {
 
    public function testExceptionSetsStatusCode() {
        $this->setMwGlobals( 'wgOut', $this->getMockWgOut() );
        try {
            throw new ThrottledError();
        } catch ( ThrottledError $e ) {
            ob_start();
            $e->report();
            $text = ob_get_clean();
            $this->assertContains( $e->getText(), $text );
        }
    }
 
    private function getMockWgOut() {
        $mock = $this->getMockBuilder( OutputPage::class )
            ->disableOriginalConstructor()
            ->getMock();
        $mock->expects( $this->once() )
            ->method( 'setStatusCode' )
            ->with( 429 );
        return $mock;
    }
 
}