1 files changed, 85 insertions, 1 deletions

diff --git a/tests/phpunit/includes/deferred/LinksUpdateTest.php b/tests/phpunit/includes/deferred/LinksUpdateTest.php
index 02f6b2ab2d5a..efbfe6f6a9e8 100644
--- a/tests/phpunit/includes/deferred/LinksUpdateTest.php
+++ b/tests/phpunit/includes/deferred/LinksUpdateTest.php
@@ -19,7 +19,8 @@ class LinksUpdateTest extends MediaWikiTestCase {
 				'externallinks',
 				'imagelinks',
 				'templatelinks',
-				'iwlinks'
+				'iwlinks',
+				'recentchanges',
 			)
 		);
 	}
@@ -41,6 +42,12 @@ class LinksUpdateTest extends MediaWikiTestCase {
 		);
 	}
 
+	public function addDBData() {
+		$this->insertPage( 'Testing' );
+		$this->insertPage( 'Some_other_page' );
+		$this->insertPage( 'Template:TestingTemplate' );
+	}
+
 	protected function makeTitleAndParserOutput( $name, $id ) {
 		$t = Title::newFromText( $name );
 		$t->mArticleID = $id; # XXX: this is fugly
@@ -133,6 +140,61 @@ class LinksUpdateTest extends MediaWikiTestCase {
 		) );
 	}
 
+	public function testOnAddingAndRemovingCategory_recentChangesRowIsAdded() {
+		$this->setMwGlobals( 'wgCategoryCollation', 'uppercase' );
+
+		$title = Title::newFromText( 'Testing' );
+		$wikiPage = new WikiPage( $title );
+		$wikiPage->doEditContent( new WikitextContent( '[[Category:Foo]]' ), 'added category' );
+
+		$this->assertRecentChangeByCategorization(
+			$title,
+			$wikiPage->getParserOutput( new ParserOptions() ),
+			Title::newFromText( 'Category:Foo' ),
+			array( array( 'Foo', '[[:Testing]] added to category' ) )
+		);
+
+		$wikiPage->doEditContent( new WikitextContent( '[[Category:Bar]]' ), 'added category' );
+		$this->assertRecentChangeByCategorization(
+			$title,
+			$wikiPage->getParserOutput( new ParserOptions() ),
+			Title::newFromText( 'Category:Foo' ),
+			array(
+				array( 'Foo', '[[:Testing]] added to category' ),
+				array( 'Foo', '[[:Testing]] removed from category' ),
+			)
+		);
+
+		$this->assertRecentChangeByCategorization(
+			$title,
+			$wikiPage->getParserOutput( new ParserOptions() ),
+			Title::newFromText( 'Category:Bar' ),
+			array(
+				array( 'Bar', '[[:Testing]] added to category' ),
+			)
+		);
+	}
+
+	public function testOnAddingAndRemovingCategoryToTemplates_embeddingPagesAreIgnored() {
+		$this->setMwGlobals( 'wgCategoryCollation', 'uppercase' );
+
+		$templateTitle = Title::newFromText( 'Template:TestingTemplate' );
+		$templatePage = new WikiPage( $templateTitle );
+
+		$wikiPage = new WikiPage( Title::newFromText( 'Testing' ) );
+		$wikiPage->doEditContent( new WikitextContent( '{{TestingTemplate}}' ), 'added template' );
+		$otherWikiPage = new WikiPage( Title::newFromText( 'Some_other_page' ) );
+		$otherWikiPage->doEditContent( new WikitextContent( '{{TestingTemplate}}' ), 'added template' );
+		$templatePage->doEditContent( new WikitextContent( '[[Category:Foo]]' ), 'added category' );
+
+		$this->assertRecentChangeByCategorization(
+			$templateTitle,
+			$templatePage->getParserOutput( new ParserOptions() ),
+			Title::newFromText( 'Foo' ),
+			array( array( 'Foo', '[[:Template:TestingTemplate]] and 2 pages added to category' ) )
+		);
+	}
+
 	/**
 	 * @covers ParserOutput::addInterwikiLink
 	 */
@@ -263,4 +325,26 @@ class LinksUpdateTest extends MediaWikiTestCase {
 		$this->assertSelect( $table, $fields, $condition, $expectedRows );
 		return $update;
 	}
+
+	protected function assertRecentChangeByCategorization(
+		Title $pageTitle, ParserOutput $parserOutput, Title $categoryTitle, $expectedRows
+	) {
+		$update = new LinksUpdate( $pageTitle, $parserOutput );
+		$revision = Revision::newFromTitle( $pageTitle );
+		$update->setRevision( $revision );
+		$update->beginTransaction();
+		$update->doUpdate();
+		$update->commitTransaction();
+
+		$this->assertSelect(
+			'recentchanges',
+			'rc_title, rc_comment',
+			array(
+				'rc_type' => RC_CATEGORIZE,
+				'rc_namespace' => NS_CATEGORY,
+				'rc_title' => $categoryTitle->getDBkey()
+			),
+			$expectedRows
+		);
+	}
 }