Add test for ApiCreateTempUserTrait

Bug: T349223
Change-Id: Idcce57a45357c64df099b5c1297bf6ad61d46957

1 files changed, 79 insertions, 0 deletions

diff --git a/tests/phpunit/includes/api/ApiCreateTempUserTraitTest.php b/tests/phpunit/includes/api/ApiCreateTempUserTraitTest.php
new file mode 100644
index 000000000000..1a4d50753e8e
--- /dev/null
+++ b/tests/phpunit/includes/api/ApiCreateTempUserTraitTest.php
@@ -0,0 +1,79 @@
+<?php
+
+use MediaWiki\HookContainer\HookRunner;
+use MediaWiki\Request\WebRequest;
+use Wikimedia\TestingAccessWrapper;
+
+/**
+ * @covers ApiCreateTempUserTrait
+ */
+class ApiCreateTempUserTraitTest extends MediaWikiIntegrationTestCase {
+
+	/**
+	 * @dataProvider provideGetTempUserRedirectUrl
+	 */
+	public function testGetTempUserRedirectUrl( $params, $expected ) {
+		$this->setTemporaryHook(
+			'TempUserCreatedRedirect',
+			static function (
+				$session,
+				$user,
+				$returnTo,
+				$returnToQuery,
+				$returnToAnchor,
+				&$redirectUrl
+			) {
+				$redirectUrl = $returnTo . $returnToQuery . $returnToAnchor;
+				return false;
+			}
+		);
+
+		$mock = $this->getMockForTrait( ApiCreateTempUserTrait::class );
+		$mock->method( 'getHookRunner' )
+			->willReturn( new HookRunner( $this->getServiceContainer()->getHookContainer() ) );
+		$mock->method( 'getRequest' )
+			->willReturn( $this->createMock( WebRequest::class ) );
+
+		$url = TestingAccessWrapper::newFromObject( $mock )
+			->getTempUserRedirectUrl( $params, $this->createMock( User::class ) );
+
+		$this->assertSame( $expected, $url );
+	}
+
+	public static function provideGetTempUserRedirectUrl() {
+		return [
+			'Default params' => [
+				[
+					'returnto' => '',
+					'returntoquery' => '',
+					'returntoanchor' => '',
+				],
+				'',
+			],
+			'Missing returnto' => [
+				[
+					'returnto' => null,
+					'returntoquery' => '',
+					'returntoanchor' => '',
+				],
+				'',
+			],
+			'Params are parsed correctly' => [
+				[
+					'returnto' => 'Base',
+					'returntoquery' => 'Query',
+					'returntoanchor' => 'Anchor',
+				],
+				'BaseQuery#Anchor',
+			],
+			'Params are parsed correctly with anchor #' => [
+				[
+					'returnto' => 'Base',
+					'returntoquery' => 'Query',
+					'returntoanchor' => '#Anchor',
+				],
+				'BaseQuery#Anchor',
+			],
+		];
+	}
+}