TagHooksTest.php

Go to the documentation of this file.

<?php
 
use MediaWiki\MediaWikiServices;
 
class TagHooksTest extends MediaWikiTestCase {
    public static function provideValidNames() {
        return [
            [ 'foo' ],
            [ 'foo-bar' ],
            [ 'foo_bar' ],
            [ 'FOO-BAR' ],
            [ 'foo bar' ]
        ];
    }
 
    public static function provideBadNames() {
        return [ [ "foo<bar" ], [ "foo>bar" ], [ "foo\nbar" ], [ "foo\rbar" ] ];
    }
 
    private function getParserOptions() {
        $popt = ParserOptions::newFromUserAndLang( new User,
            MediaWikiServices::getInstance()->getContentLanguage() );
        return $popt;
    }
 
    public function testTagHooks( $tag ) {
        $parser = MediaWikiServices::getInstance()->getParserFactory()->create();
 
        $parser->setHook( $tag, [ $this, 'tagCallback' ] );
        $parserOutput = $parser->parse(
            "Foo<$tag>Bar</$tag>Baz",
            Title::newFromText( 'Test' ),
            $this->getParserOptions()
        );
        $this->assertEquals( "<p>FooOneBaz\n</p>", $parserOutput->getText( [ 'unwrap' => true ] ) );
 
        $parser->mPreprocessor = null; # Break the Parser <-> Preprocessor cycle
    }
 
    public function testBadTagHooks( $tag ) {
        $parser = MediaWikiServices::getInstance()->getParserFactory()->create();
 
        $parser->setHook( $tag, [ $this, 'tagCallback' ] );
        $parser->parse(
            "Foo<$tag>Bar</$tag>Baz",
            Title::newFromText( 'Test' ),
            $this->getParserOptions()
        );
        $this->fail( 'Exception not thrown.' );
    }
 
    public function testFunctionTagHooks( $tag ) {
        $parser = MediaWikiServices::getInstance()->getParserFactory()->create();
 
        $parser->setFunctionTagHook( $tag, [ $this, 'functionTagCallback' ], 0 );
        $parserOutput = $parser->parse(
            "Foo<$tag>Bar</$tag>Baz",
            Title::newFromText( 'Test' ),
            $this->getParserOptions()
        );
        $this->assertEquals( "<p>FooOneBaz\n</p>", $parserOutput->getText( [ 'unwrap' => true ] ) );
 
        $parser->mPreprocessor = null; # Break the Parser <-> Preprocessor cycle
    }
 
    public function testBadFunctionTagHooks( $tag ) {
        $parser = MediaWikiServices::getInstance()->getParserFactory()->create();
 
        $parser->setFunctionTagHook(
            $tag,
            [ $this, 'functionTagCallback' ],
            Parser::SFH_OBJECT_ARGS
        );
        $parser->parse(
            "Foo<$tag>Bar</$tag>Baz",
            Title::newFromText( 'Test' ),
            $this->getParserOptions()
        );
        $this->fail( 'Exception not thrown.' );
    }
 
    function tagCallback( $text, $params, $parser ) {
        return str_rot13( $text );
    }
 
    function functionTagCallback( &$parser, $frame, $code, $attribs ) {
        return str_rot13( $code );
    }
}