wiki.techinc.nl/tests/phpunit/includes/api/query/ApiQueryTokensTest.php
Lucas Werkmeister 47f598b94b Use continuation when meta=tokens doesn’t fit in response
The implementation and test are both based on similar code for
meta=siteinfo. I’ve checked manually that, when no continuation is
necessary, the output looks the same as before in all formats, despite
the different way that we now set it.

Bug: T309921
Change-Id: I3b4b7aecdeb6461ae5bd8dc188e5ba49a119858e
2022-06-04 17:43:38 +02:00

65 lines
1.9 KiB
PHP

<?php
/**
* @group API
* @group medium
* @covers ApiQueryTokens
*/
class ApiQueryTokensTest extends ApiTestCase {
public function testGetCsrfToken() {
$params = [
'action' => 'query',
'meta' => 'tokens',
'type' => 'csrf',
];
$user = $this->getTestUser()->getUser();
$apiResult = $this->doApiRequest( $params, null, false, $user );
$this->assertArrayHasKey( 'query', $apiResult[0] );
$this->assertArrayHasKey( 'tokens', $apiResult[0]['query'] );
$this->assertArrayHasKey( 'csrftoken', $apiResult[0]['query']['tokens'] );
$this->assertStringEndsWith( '+\\', $apiResult[0]['query']['tokens']['csrftoken'] );
}
public function testGetAllTokens() {
$params = [
'action' => 'query',
'meta' => 'tokens',
'type' => '*',
];
$user = $this->getTestUser()->getUser();
$apiResult = $this->doApiRequest( $params, null, false, $user );
$this->assertArrayHasKey( 'query', $apiResult[0] );
$this->assertArrayHasKey( 'tokens', $apiResult[0]['query'] );
// MW core has 7 token types (createaccount, csrf, login, patrol, rollback, userrights, watch)
$this->assertGreaterThanOrEqual( 7, count( $apiResult[0]['query']['tokens'] ) );
}
public function testContinuation(): void {
// one token is 42 characters, so 100 is enough for 2 tokens but not 3
$size = 100;
$this->setMwGlobals( 'wgAPIMaxResultSize', $size );
[ $result ] = $this->doApiRequest( [
'action' => 'query',
'meta' => 'tokens',
'type' => 'csrf|patrol|watch',
] );
$this->assertSame(
wfMessage( 'apiwarn-truncatedresult', Message::numParam( $size ) )
->text(),
$result['warnings']['result']['warnings']
);
$this->assertSame( [ 'csrftoken', 'patroltoken' ], array_keys( $result['query']['tokens'] ) );
$this->assertTrue( $result['batchcomplete'], 'batchcomplete should be true' );
$this->assertSame( [ 'type' => 'watch', 'continue' => '-||' ], $result['continue'] );
}
}