MediaWiki 1.30.2

[autoinstallsdev/mediawiki.git] / tests / phpunit / includes / parser / TagHooksTest.php

diff --git a/tests/phpunit/includes/parser/TagHooksTest.php b/tests/phpunit/includes/parser/TagHooksTest.php
new file mode 100644 (file)
index 0000000..06fe272
--- /dev/null
+++ b/tests/phpunit/includes/parser/TagHooksTest.php
@@ -0,0 +1,134 @@
+<?php
+
+/**
+ * @group Database
+ * @group Parser
+ *
+ * @covers Parser
+ * @covers StripState
+ *
+ * @covers Preprocessor_DOM
+ * @covers PPDStack
+ * @covers PPDStackElement
+ * @covers PPDPart
+ * @covers PPFrame_DOM
+ * @covers PPTemplateFrame_DOM
+ * @covers PPCustomFrame_DOM
+ * @covers PPNode_DOM
+ *
+ * @covers Preprocessor_Hash
+ * @covers PPDStack_Hash
+ * @covers PPDStackElement_Hash
+ * @covers PPDPart_Hash
+ * @covers PPFrame_Hash
+ * @covers PPTemplateFrame_Hash
+ * @covers PPCustomFrame_Hash
+ * @covers PPNode_Hash_Tree
+ * @covers PPNode_Hash_Text
+ * @covers PPNode_Hash_Array
+ * @covers PPNode_Hash_Attr
+ */
+class TagHookTest 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() {
+               global $wgContLang;
+               $popt = ParserOptions::newFromUserAndLang( new User, $wgContLang );
+               $popt->setWrapOutputClass( false );
+               return $popt;
+       }
+
+       /**
+        * @dataProvider provideValidNames
+        */
+       public function testTagHooks( $tag ) {
+               global $wgParserConf;
+               $parser = new Parser( $wgParserConf );
+
+               $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() );
+
+               $parser->mPreprocessor = null; # Break the Parser <-> Preprocessor cycle
+       }
+
+       /**
+        * @dataProvider provideBadNames
+        * @expectedException MWException
+        */
+       public function testBadTagHooks( $tag ) {
+               global $wgParserConf;
+               $parser = new Parser( $wgParserConf );
+
+               $parser->setHook( $tag, [ $this, 'tagCallback' ] );
+               $parser->parse(
+                       "Foo<$tag>Bar</$tag>Baz",
+                       Title::newFromText( 'Test' ),
+                       $this->getParserOptions()
+               );
+               $this->fail( 'Exception not thrown.' );
+       }
+
+       /**
+        * @dataProvider provideValidNames
+        */
+       public function testFunctionTagHooks( $tag ) {
+               global $wgParserConf;
+               $parser = new Parser( $wgParserConf );
+
+               $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() );
+
+               $parser->mPreprocessor = null; # Break the Parser <-> Preprocessor cycle
+       }
+
+       /**
+        * @dataProvider provideBadNames
+        * @expectedException MWException
+        */
+       public function testBadFunctionTagHooks( $tag ) {
+               global $wgParserConf;
+               $parser = new Parser( $wgParserConf );
+
+               $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 );
+       }
+}