X-Git-Url: https://scripts.mit.edu/gitweb/autoinstallsdev/mediawiki.git/blobdiff_plain/19e297c21b10b1b8a3acad5e73fc71dcb35db44a..6932310fd58ebef145fa01eb76edf7150284d8ea:/tests/phpunit/includes/deferred/DeferredUpdatesTest.php

diff --git a/tests/phpunit/includes/deferred/DeferredUpdatesTest.php b/tests/phpunit/includes/deferred/DeferredUpdatesTest.php
new file mode 100644
index 00000000..3b423563
--- /dev/null
+++ b/tests/phpunit/includes/deferred/DeferredUpdatesTest.php
@@ -0,0 +1,221 @@
+<?php
+
+class DeferredUpdatesTest extends MediaWikiTestCase {
+
+	/**
+	 * @covers DeferredUpdates::getPendingUpdates
+	 */
+	public function testGetPendingUpdates() {
+		# Prevent updates from running
+		$this->setMwGlobals( 'wgCommandLineMode', false );
+
+		$pre = DeferredUpdates::PRESEND;
+		$post = DeferredUpdates::POSTSEND;
+		$all = DeferredUpdates::ALL;
+
+		$update = $this->getMock( DeferrableUpdate::class );
+		$update->expects( $this->never() )
+			->method( 'doUpdate' );
+
+		DeferredUpdates::addUpdate( $update, $pre );
+		$this->assertCount( 1, DeferredUpdates::getPendingUpdates( $pre ) );
+		$this->assertCount( 0, DeferredUpdates::getPendingUpdates( $post ) );
+		$this->assertCount( 1, DeferredUpdates::getPendingUpdates( $all ) );
+		$this->assertCount( 1, DeferredUpdates::getPendingUpdates() );
+		DeferredUpdates::clearPendingUpdates();
+		$this->assertCount( 0, DeferredUpdates::getPendingUpdates() );
+
+		DeferredUpdates::addUpdate( $update, $post );
+		$this->assertCount( 0, DeferredUpdates::getPendingUpdates( $pre ) );
+		$this->assertCount( 1, DeferredUpdates::getPendingUpdates( $post ) );
+		$this->assertCount( 1, DeferredUpdates::getPendingUpdates( $all ) );
+		$this->assertCount( 1, DeferredUpdates::getPendingUpdates() );
+		DeferredUpdates::clearPendingUpdates();
+		$this->assertCount( 0, DeferredUpdates::getPendingUpdates() );
+	}
+
+	public function testDoUpdatesWeb() {
+		$this->setMwGlobals( 'wgCommandLineMode', false );
+
+		$updates = [
+			'1' => "deferred update 1;\n",
+			'2' => "deferred update 2;\n",
+			'2-1' => "deferred update 1 within deferred update 2;\n",
+			'2-2' => "deferred update 2 within deferred update 2;\n",
+			'3' => "deferred update 3;\n",
+			'3-1' => "deferred update 1 within deferred update 3;\n",
+			'3-2' => "deferred update 2 within deferred update 3;\n",
+			'3-1-1' => "deferred update 1 within deferred update 1 within deferred update 3;\n",
+			'3-2-1' => "deferred update 1 within deferred update 2 with deferred update 3;\n",
+		];
+		DeferredUpdates::addCallableUpdate(
+			function () use ( $updates ) {
+				echo $updates['1'];
+			}
+		);
+		DeferredUpdates::addCallableUpdate(
+			function () use ( $updates ) {
+				echo $updates['2'];
+				DeferredUpdates::addCallableUpdate(
+					function () use ( $updates ) {
+						echo $updates['2-1'];
+					}
+				);
+				DeferredUpdates::addCallableUpdate(
+					function () use ( $updates ) {
+						echo $updates['2-2'];
+					}
+				);
+			}
+		);
+		DeferredUpdates::addCallableUpdate(
+			function () use ( $updates ) {
+				echo $updates['3'];
+				DeferredUpdates::addCallableUpdate(
+					function () use ( $updates ) {
+						echo $updates['3-1'];
+						DeferredUpdates::addCallableUpdate(
+							function () use ( $updates ) {
+								echo $updates['3-1-1'];
+							}
+						);
+					}
+				);
+				DeferredUpdates::addCallableUpdate(
+					function () use ( $updates ) {
+						echo $updates['3-2'];
+						DeferredUpdates::addCallableUpdate(
+							function () use ( $updates ) {
+								echo $updates['3-2-1'];
+							}
+						);
+					}
+				);
+			}
+		);
+
+		$this->assertEquals( 3, DeferredUpdates::pendingUpdatesCount() );
+
+		$this->expectOutputString( implode( '', $updates ) );
+
+		DeferredUpdates::doUpdates();
+
+		$x = null;
+		$y = null;
+		DeferredUpdates::addCallableUpdate(
+			function () use ( &$x ) {
+				$x = 'Sherity';
+			},
+			DeferredUpdates::PRESEND
+		);
+		DeferredUpdates::addCallableUpdate(
+			function () use ( &$y ) {
+				$y = 'Marychu';
+			},
+			DeferredUpdates::POSTSEND
+		);
+
+		$this->assertNull( $x, "Update not run yet" );
+		$this->assertNull( $y, "Update not run yet" );
+
+		DeferredUpdates::doUpdates( 'run', DeferredUpdates::PRESEND );
+		$this->assertEquals( "Sherity", $x, "PRESEND update ran" );
+		$this->assertNull( $y, "POSTSEND update not run yet" );
+
+		DeferredUpdates::doUpdates( 'run', DeferredUpdates::POSTSEND );
+		$this->assertEquals( "Marychu", $y, "POSTSEND update ran" );
+	}
+
+	public function testDoUpdatesCLI() {
+		$this->setMwGlobals( 'wgCommandLineMode', true );
+		$updates = [
+			'1' => "deferred update 1;\n",
+			'2' => "deferred update 2;\n",
+			'2-1' => "deferred update 1 within deferred update 2;\n",
+			'2-2' => "deferred update 2 within deferred update 2;\n",
+			'3' => "deferred update 3;\n",
+			'3-1' => "deferred update 1 within deferred update 3;\n",
+			'3-2' => "deferred update 2 within deferred update 3;\n",
+			'3-1-1' => "deferred update 1 within deferred update 1 within deferred update 3;\n",
+			'3-2-1' => "deferred update 1 within deferred update 2 with deferred update 3;\n",
+		];
+
+		wfGetLBFactory()->commitMasterChanges( __METHOD__ ); // clear anything
+
+		DeferredUpdates::addCallableUpdate(
+			function () use ( $updates ) {
+				echo $updates['1'];
+			}
+		);
+		DeferredUpdates::addCallableUpdate(
+			function () use ( $updates ) {
+				echo $updates['2'];
+				DeferredUpdates::addCallableUpdate(
+					function () use ( $updates ) {
+						echo $updates['2-1'];
+					}
+				);
+				DeferredUpdates::addCallableUpdate(
+					function () use ( $updates ) {
+						echo $updates['2-2'];
+					}
+				);
+			}
+		);
+		DeferredUpdates::addCallableUpdate(
+			function () use ( $updates ) {
+				echo $updates['3'];
+				DeferredUpdates::addCallableUpdate(
+					function () use ( $updates ) {
+						echo $updates['3-1'];
+						DeferredUpdates::addCallableUpdate(
+							function () use ( $updates ) {
+								echo $updates['3-1-1'];
+							}
+						);
+					}
+				);
+				DeferredUpdates::addCallableUpdate(
+					function () use ( $updates ) {
+						echo $updates['3-2'];
+						DeferredUpdates::addCallableUpdate(
+							function () use ( $updates ) {
+								echo $updates['3-2-1'];
+							}
+						);
+					}
+				);
+			}
+		);
+
+		$this->expectOutputString( implode( '', $updates ) );
+
+		DeferredUpdates::doUpdates();
+	}
+
+	public function testPresendAddOnPostsendRun() {
+		$this->setMwGlobals( 'wgCommandLineMode', true );
+
+		$x = false;
+		$y = false;
+		wfGetLBFactory()->commitMasterChanges( __METHOD__ ); // clear anything
+
+		DeferredUpdates::addCallableUpdate(
+			function () use ( &$x, &$y ) {
+				$x = true;
+				DeferredUpdates::addCallableUpdate(
+					function () use ( &$y ) {
+						$y = true;
+					},
+					DeferredUpdates::PRESEND
+				);
+			},
+			DeferredUpdates::POSTSEND
+		);
+
+		DeferredUpdates::doUpdates();
+
+		$this->assertTrue( $x, "Outer POSTSEND update ran" );
+		$this->assertTrue( $y, "Nested PRESEND update ran" );
+	}
+}