tests/phpunit/includes/media/PNGTest.php

   1 <?php
   2 class PNGHandlerTest extends MediaWikiTestCase {
   3
   4         /** @var PNGHandler */
   5         protected $handler;
   6         /** @var FSRepo */
   7         protected $repo;
   8         /** @var FSFileBackend */
   9         protected $backend;
  10         /** @var string */
  11         protected $filePath;
  12
  13         protected function setUp() {
  14                 parent::setUp();
  15
  16                 $this->filePath = __DIR__ . '/../../data/media';
  17                 $this->backend = new FSFileBackend( array(
  18                         'name' => 'localtesting',
  19                         'lockManager' => 'nullLockManager',
  20                         'containerPaths' => array( 'data' => $this->filePath )
  21                 ) );
  22                 $this->repo = new FSRepo( array(
  23                         'name' => 'temp',
  24                         'url' => 'http://localhost/thumbtest',
  25                         'backend' => $this->backend
  26                 ) );
  27                 $this->handler = new PNGHandler();
  28         }
  29
  30         /**
  31          * @covers PNGHandler::getMetadata
  32          */
  33         public function testInvalidFile() {
  34                 $res = $this->handler->getMetadata( null, $this->filePath . '/README' );
  35                 $this->assertEquals( PNGHandler::BROKEN_FILE, $res );
  36         }
  37
  38         /**
  39          * @param $filename String basename of the file to check
  40          * @param $expected boolean Expected result.
  41          * @dataProvider provideIsAnimated
  42          * @covers PNGHandler::isAnimatedImage
  43          */
  44         public function testIsAnimanted( $filename, $expected ) {
  45                 $file = $this->dataFile( $filename, 'image/png' );
  46                 $actual = $this->handler->isAnimatedImage( $file );
  47                 $this->assertEquals( $expected, $actual );
  48         }
  49
  50         public static function provideIsAnimated() {
  51                 return array(
  52                         array( 'Animated_PNG_example_bouncing_beach_ball.png', true ),
  53                         array( '1bit-png.png', false ),
  54                 );
  55         }
  56
  57         /**
  58          * @param $filename String
  59          * @param $expected Integer Total image area
  60          * @dataProvider provideGetImageArea
  61          * @covers PNGHandler::getImageArea
  62          */
  63         public function testGetImageArea( $filename, $expected ) {
  64                 $file = $this->dataFile( $filename, 'image/png' );
  65                 $actual = $this->handler->getImageArea( $file, $file->getWidth(), $file->getHeight() );
  66                 $this->assertEquals( $expected, $actual );
  67         }
  68
  69         public static function provideGetImageArea() {
  70                 return array(
  71                         array( '1bit-png.png', 2500 ),
  72                         array( 'greyscale-png.png', 2500 ),
  73                         array( 'Png-native-test.png', 126000 ),
  74                         array( 'Animated_PNG_example_bouncing_beach_ball.png', 10000 ),
  75                 );
  76         }
  77
  78         /**
  79          * @param $metadata String Serialized metadata
  80          * @param $expected Integer One of the class constants of PNGHandler
  81          * @dataProvider provideIsMetadataValid
  82          * @covers PNGHandler::isMetadataValid
  83          */
  84         public function testIsMetadataValid( $metadata, $expected ) {
  85                 $actual = $this->handler->isMetadataValid( null, $metadata );
  86                 $this->assertEquals( $expected, $actual );
  87         }
  88
  89         public static function provideIsMetadataValid() {
  90                 return array(
  91                         array( PNGHandler::BROKEN_FILE, PNGHandler::METADATA_GOOD ),
  92                         array( '', PNGHandler::METADATA_BAD ),
  93                         array( null, PNGHandler::METADATA_BAD ),
  94                         array( 'Something invalid!', PNGHandler::METADATA_BAD ),
  95                         array( 'a:6:{s:10:"frameCount";i:0;s:9:"loopCount";i:1;s:8:"duration";d:0;s:8:"bitDepth";i:8;s:9:"colorType";s:10:"truecolour";s:8:"metadata";a:1:{s:15:"_MW_PNG_VERSION";i:1;}}', PNGHandler::METADATA_GOOD ),
  96                 );
  97         }
  98
  99         /**
 100          * @param $filename String
 101          * @param $expected String Serialized array
 102          * @dataProvider provideGetMetadata
 103          * @covers PNGHandler::getMetadata
 104          */
 105         public function testGetMetadata( $filename, $expected ) {
 106                 $file = $this->dataFile( $filename, 'image/png' );
 107                 $actual = $this->handler->getMetadata( $file, "$this->filePath/$filename" );
 108 //              $this->assertEquals( unserialize( $expected ), unserialize( $actual ) );
 109                 $this->assertEquals( ( $expected ), ( $actual ) );
 110         }
 111
 112         public static function provideGetMetadata() {
 113                 return array(
 114                         array( 'rgb-na-png.png', 'a:6:{s:10:"frameCount";i:0;s:9:"loopCount";i:1;s:8:"duration";d:0;s:8:"bitDepth";i:8;s:9:"colorType";s:10:"truecolour";s:8:"metadata";a:1:{s:15:"_MW_PNG_VERSION";i:1;}}' ),
 115                         array( 'xmp.png', 'a:6:{s:10:"frameCount";i:0;s:9:"loopCount";i:1;s:8:"duration";d:0;s:8:"bitDepth";i:1;s:9:"colorType";s:14:"index-coloured";s:8:"metadata";a:2:{s:12:"SerialNumber";s:9:"123456789";s:15:"_MW_PNG_VERSION";i:1;}}' ),
 116                 );
 117         }
 118
 119         private function dataFile( $name, $type ) {
 120                 return new UnregisteredLocalFile( false, $this->repo,
 121                         "mwstore://localtesting/data/$name", $type );
 122         }
 123 }