7854dd5fc9
The functionality of creating title mocks is generally useful and this will also allow to make HandlerTestTrait more narrow. Bug: T264058 Change-Id: I76eca48dfcff65a6203fccde5366912a2d66c495
131 lines
3.7 KiB
PHP
131 lines
3.7 KiB
PHP
<?php
|
|
|
|
namespace MediaWiki\Tests\Rest\Handler;
|
|
|
|
use File;
|
|
use MockTitleTrait;
|
|
use PHPUnit\Framework\MockObject\MockObject;
|
|
use RepoGroup;
|
|
use ThumbnailImage;
|
|
use Title;
|
|
|
|
/**
|
|
* A trait providing utility functions for mocking media-related objects.
|
|
*
|
|
* @package MediaWiki\Tests\Rest\Handler
|
|
*/
|
|
trait MediaTestTrait {
|
|
|
|
use HandlerTestTrait;
|
|
use MockTitleTrait;
|
|
|
|
/**
|
|
* @param Title|string $title
|
|
*
|
|
* @return File|MockObject
|
|
*/
|
|
private function makeMissingMockFile( $title ) {
|
|
$title = $title instanceof Title
|
|
? $title
|
|
: $this->makeMockTitle( 'File:' . $title, [ 'namespace' => NS_FILE ] );
|
|
|
|
/** @var MockObject|File $file */
|
|
$file = $this->createNoOpMock(
|
|
File::class,
|
|
[ 'getTitle', 'exists', 'getDescriptionUrl' ]
|
|
);
|
|
$file->method( 'getTitle' )->willReturn( $title );
|
|
$file->method( 'exists' )->willReturn( false );
|
|
$file->method( 'getDescriptionUrl' )->willReturn(
|
|
'https://example.com/wiki/' . $title->getPrefixedDBkey()
|
|
);
|
|
|
|
return $file;
|
|
}
|
|
|
|
/**
|
|
* @param Title|string $title
|
|
*
|
|
* @return File|MockObject
|
|
*/
|
|
private function makeMockFile( $title ) {
|
|
$title = $title instanceof Title
|
|
? $title
|
|
: $this->makeMockTitle( 'File:' . $title, [ 'namespace' => NS_FILE ] );
|
|
|
|
/** @var MockObject|File $file */
|
|
$file = $this->createNoOpMock(
|
|
File::class,
|
|
[ 'getTitle', 'getDescriptionUrl', 'exists', 'userCan', 'getUser', 'getTimestamp',
|
|
'getMediaType', 'getSize', 'getHeight', 'getWidth', 'getDisplayWidthHeight',
|
|
'getLength', 'getUrl', 'allowInlineDisplay', 'transform', 'getSha1' ]
|
|
);
|
|
$file->method( 'getTitle' )->willReturn( $title );
|
|
$file->method( 'exists' )->willReturn( true );
|
|
$file->method( 'userCan' )->willReturn( true );
|
|
$file->method( 'getUser' )->willReturnCallback( function ( $type ) {
|
|
return $type === 'id' ? 7 : 'Alice';
|
|
} );
|
|
$file->method( 'getTimestamp' )->willReturn( '20200102030405' );
|
|
$file->method( 'getMediaType' )->willReturn( 'test' );
|
|
$file->method( 'getSize' )->willReturn( 12345 );
|
|
$file->method( 'getHeight' )->willReturn( 400 );
|
|
$file->method( 'getWidth' )->willReturn( 600 );
|
|
$file->method( 'getDisplayWidthHeight' )->willReturn( [ 600, 400 ] );
|
|
$file->method( 'getLength' )->willReturn( 678 );
|
|
$file->method( 'getSha1' )->willReturn( 'DEADBEEF' );
|
|
$file->method( 'allowInlineDisplay' )->willReturn( true );
|
|
$file->method( 'getUrl' )->willReturn(
|
|
'https://media.example.com/static/' . $title->getDBkey()
|
|
);
|
|
$file->method( 'getDescriptionUrl' )->willReturn(
|
|
'https://example.com/wiki/' . $title->getPrefixedDBkey()
|
|
);
|
|
|
|
$thumbnail = new ThumbnailImage(
|
|
$file,
|
|
'https://media.example.com/static/thumb/' . $title->getDBkey(),
|
|
false,
|
|
[ 'width' => 64, 'height' => 64, ]
|
|
);
|
|
$file->method( 'transform' )->willReturn( $thumbnail );
|
|
|
|
return $file;
|
|
}
|
|
|
|
/**
|
|
* @return MockObject|RepoGroup
|
|
*/
|
|
private function makeMockRepoGroup() {
|
|
$findFile = function ( $title ) {
|
|
$title = $title instanceof Title
|
|
? $title
|
|
: $this->makeMockTitle( 'File:' . $title, [ 'namespace' => NS_FILE ] );
|
|
|
|
if ( preg_match( '/missing/i', $title->getText() )
|
|
|| $title->getNamespace() !== NS_FILE
|
|
) {
|
|
return $this->makeMissingMockFile( $title );
|
|
} else {
|
|
return $this->makeMockFile( $title );
|
|
}
|
|
};
|
|
|
|
$findFiles = function ( array $inputItems ) use ( $findFile ) {
|
|
$files = [];
|
|
foreach ( $inputItems as $item ) {
|
|
$files[] = ( $findFile )( $item['title'] );
|
|
}
|
|
|
|
return $files;
|
|
};
|
|
|
|
/** @var RepoGroup|MockObject $repoGroup */
|
|
$repoGroup = $this->createNoOpMock( RepoGroup::class, [ 'findFiles', 'findFile' ] );
|
|
$repoGroup->method( 'findFile' )->willReturnCallback( $findFile );
|
|
$repoGroup->method( 'findFiles' )->willReturnCallback( $findFiles );
|
|
|
|
return $repoGroup;
|
|
}
|
|
|
|
}
|