tests/Wallabag/CoreBundle/Helper/DownloadImagesTest.php

   1 <?php
   2
   3 namespace Tests\Wallabag\CoreBundle\Helper;
   4
   5 use GuzzleHttp\Client;
   6 use GuzzleHttp\Psr7\Response;
   7 use GuzzleHttp\Stream\Stream;
   8 use GuzzleHttp\Subscriber\Mock;
   9 use Http\Mock\Client as HttpMockClient;
  10 use Monolog\Handler\TestHandler;
  11 use Monolog\Logger;
  12 use PHPUnit\Framework\TestCase;
  13 use Wallabag\CoreBundle\Helper\DownloadImages;
  14
  15 class DownloadImagesTest extends TestCase
  16 {
  17     public function dataForSuccessImage()
  18     {
  19         return [
  20             'imgur' => [
  21                 '<div><img src="http://i.imgur.com/T9qgcHc.jpg" /></div>',
  22                 'http://imgur.com/gallery/WxtWY',
  23             ],
  24             'image with &' => [
  25                 '<div><img src="https://i2.wp.com/www.tvaddons.ag/wp-content/uploads/2017/01/Screen-Shot-2017-01-07-at-10.17.40-PM.jpg?w=640&amp;ssl=1" /></div>',
  26                 'https://www.tvaddons.ag/realdebrid-kodi-jarvis/',
  27             ],
  28         ];
  29     }
  30
  31     /**
  32      * @dataProvider dataForSuccessImage
  33      */
  34     public function testProcessHtml($html, $url)
  35     {
  36         $httpMockClient = new HttpMockClient();
  37
  38         $httpMockClient->addResponse(new Response(200, ['content-type' => 'image/png'], file_get_contents(__DIR__ . '/../fixtures/unnamed.png')));
  39
  40         $logHandler = new TestHandler();
  41         $logger = new Logger('test', [$logHandler]);
  42
  43         $download = new DownloadImages($httpMockClient, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
  44
  45         $res = $download->processHtml(123, $html, $url);
  46
  47         // this the base path of all image (since it's calculated using the entry id: 123)
  48         $this->assertContains('http://wallabag.io/assets/images/9/b/9b0ead26/', $res);
  49     }
  50
  51     public function testProcessHtmlWithBadImage()
  52     {
  53         $httpMockClient = new HttpMockClient();
  54         $httpMockClient->addResponse(new Response(200, ['content-type' => 'application/json'], ''));
  55
  56         $logHandler = new TestHandler();
  57         $logger = new Logger('test', [$logHandler]);
  58
  59         $download = new DownloadImages($httpMockClient, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
  60         $res = $download->processHtml(123, '<div><img src="http://i.imgur.com/T9qgcHc.jpg" /></div>', 'http://imgur.com/gallery/WxtWY');
  61
  62         $this->assertContains('http://i.imgur.com/T9qgcHc.jpg', $res, 'Image were not replace because of content-type');
  63     }
  64
  65     public function singleImage()
  66     {
  67         return [
  68             ['image/pjpeg', 'jpeg'],
  69             ['image/jpeg', 'jpeg'],
  70             ['image/png', 'png'],
  71             ['image/gif', 'gif'],
  72         ];
  73     }
  74
  75     /**
  76      * @dataProvider singleImage
  77      */
  78     public function testProcessSingleImage($header, $extension)
  79     {
  80         $httpMockClient = new HttpMockClient();
  81         $httpMockClient->addResponse(new Response(200, ['content-type' => $header], file_get_contents(__DIR__ . '/../fixtures/unnamed.png')));
  82
  83         $logHandler = new TestHandler();
  84         $logger = new Logger('test', [$logHandler]);
  85
  86         $download = new DownloadImages($httpMockClient, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
  87         $res = $download->processSingleImage(123, 'T9qgcHc.jpg', 'http://imgur.com/gallery/WxtWY');
  88
  89         $this->assertContains('/assets/images/9/b/9b0ead26/ebe60399.' . $extension, $res);
  90     }
  91
  92     public function testProcessSingleImageWithBadUrl()
  93     {
  94         $httpMockClient = new HttpMockClient();
  95         $httpMockClient->addResponse(new Response(404, []));
  96
  97         $logHandler = new TestHandler();
  98         $logger = new Logger('test', [$logHandler]);
  99
 100         $download = new DownloadImages($httpMockClient, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
 101         $res = $download->processSingleImage(123, 'T9qgcHc.jpg', 'http://imgur.com/gallery/WxtWY');
 102
 103         $this->assertFalse($res, 'Image can not be found, so it will not be replaced');
 104     }
 105
 106     public function testProcessSingleImageWithBadImage()
 107     {
 108         $httpMockClient = new HttpMockClient();
 109         $httpMockClient->addResponse(new Response(200, ['content-type' => 'image/png'], ''));
 110
 111         $logHandler = new TestHandler();
 112         $logger = new Logger('test', [$logHandler]);
 113
 114         $download = new DownloadImages($httpMockClient, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
 115         $res = $download->processSingleImage(123, 'http://i.imgur.com/T9qgcHc.jpg', 'http://imgur.com/gallery/WxtWY');
 116
 117         $this->assertFalse($res, 'Image can not be loaded, so it will not be replaced');
 118     }
 119
 120     public function testProcessSingleImageFailAbsolute()
 121     {
 122         $httpMockClient = new HttpMockClient();
 123         $httpMockClient->addResponse(new Response(200, ['content-type' => 'image/png'], file_get_contents(__DIR__ . '/../fixtures/unnamed.png')));
 124
 125         $logHandler = new TestHandler();
 126         $logger = new Logger('test', [$logHandler]);
 127
 128         $download = new DownloadImages($httpMockClient, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
 129         $res = $download->processSingleImage(123, '/i.imgur.com/T9qgcHc.jpg', 'imgur.com/gallery/WxtWY');
 130
 131         $this->assertFalse($res, 'Absolute image can not be determined, so it will not be replaced');
 132     }
 133
 134     public function testProcessRealImage()
 135     {
 136         $httpMockClient = new HttpMockClient();
 137         $httpMockClient->addResponse(new Response(200, ['content-type' => null], file_get_contents(__DIR__ . '/../fixtures/image-no-content-type.jpg')));
 138
 139         $logHandler = new TestHandler();
 140         $logger = new Logger('test', [$logHandler]);
 141
 142         $download = new DownloadImages($httpMockClient, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
 143
 144         $res = $download->processSingleImage(
 145             123,
 146             'https://cdn.theconversation.com/files/157200/article/width926/gsj2rjp2-1487348607.jpg',
 147             'https://theconversation.com/conversation-avec-gerald-bronner-ce-nest-pas-la-post-verite-qui-nous-menace-mais-lextension-de-notre-credulite-73089'
 148         );
 149
 150         $this->assertContains('http://wallabag.io/assets/images/9/b/9b0ead26/', $res, 'Content-Type was empty but data is ok for an image');
 151         $this->assertContains('DownloadImages: Checking extension (alternative)', $logHandler->getRecords()[3]['message']);
 152     }
 153
 154     public function testProcessImageWithSrcset()
 155     {
 156         $client = new Client();
 157
 158         $mock = new Mock([
 159             new Response(200, ['content-type' => 'image/jpeg'], Stream::factory(file_get_contents(__DIR__ . '/../fixtures/image-no-content-type.jpg'))),
 160             new Response(200, ['content-type' => 'image/jpeg'], Stream::factory(file_get_contents(__DIR__ . '/../fixtures/image-no-content-type.jpg'))),
 161             new Response(200, ['content-type' => 'image/jpeg'], Stream::factory(file_get_contents(__DIR__ . '/../fixtures/image-no-content-type.jpg'))),
 162         ]);
 163
 164         $client->getEmitter()->attach($mock);
 165
 166         $logHandler = new TestHandler();
 167         $logger = new Logger('test', [$logHandler]);
 168
 169         $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
 170         $res = $download->processHtml(123, '<p><img class="alignnone wp-image-1153" src="http://piketty.blog.lemonde.fr/files/2017/10/F1FR-530x375.jpg" alt="" width="628" height="444" srcset="http://piketty.blog.lemonde.fr/files/2017/10/F1FR-530x375.jpg 530w, http://piketty.blog.lemonde.fr/files/2017/10/F1FR-768x543.jpg 768w, http://piketty.blog.lemonde.fr/files/2017/10/F1FR-900x636.jpg 900w" sizes="(max-width: 628px) 100vw, 628px" /></p>', 'http://piketty.blog.lemonde.fr/2017/10/12/budget-2018-la-jeunesse-sacrifiee/');
 171
 172         $this->assertNotContains('http://piketty.blog.lemonde.fr/', $res, 'Image srcset attribute were not replaced');
 173     }
 174
 175     public function testProcessImageWithTrickySrcset()
 176     {
 177         $client = new Client();
 178
 179         $mock = new Mock([
 180             new Response(200, ['content-type' => 'image/jpeg'], Stream::factory(file_get_contents(__DIR__ . '/../fixtures/image-no-content-type.jpg'))),
 181             new Response(200, ['content-type' => 'image/jpeg'], Stream::factory(file_get_contents(__DIR__ . '/../fixtures/image-no-content-type.jpg'))),
 182             new Response(200, ['content-type' => 'image/jpeg'], Stream::factory(file_get_contents(__DIR__ . '/../fixtures/image-no-content-type.jpg'))),
 183         ]);
 184
 185         $client->getEmitter()->attach($mock);
 186
 187         $logHandler = new TestHandler();
 188         $logger = new Logger('test', [$logHandler]);
 189
 190         $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
 191         $res = $download->processHtml(123, '<figure id="post-257260" class="align-none media-257260"><img src="https://cdn.css-tricks.com/wp-content/uploads/2017/08/the-critical-request.png" srcset="https://res.cloudinary.com/css-tricks/image/upload/c_scale,w_1000,f_auto,q_auto/v1501594717/the-critical-request_bqdfaa.png 1000w, https://res.cloudinary.com/css-tricks/image/upload/c_scale,w_200,f_auto,q_auto/v1501594717/the-critical-request_bqdfaa.png 200w" sizes="(min-width: 1850px) calc( (100vw - 555px) / 3 )
 192        (min-width: 1251px) calc( (100vw - 530px) / 2 )
 193        (min-width: 1086px) calc(100vw - 480px)
 194        (min-width: 626px)  calc(100vw - 335px)
 195                            calc(100vw - 30px)" alt="" /></figure>', 'https://css-tricks.com/the-critical-request/');
 196
 197         $this->assertNotContains('f_auto,q_auto', $res, 'Image srcset attribute were not replaced');
 198     }
 199
 200     public function testProcessImageWithNullPath()
 201     {
 202         $client = new Client();
 203
 204         $mock = new Mock([
 205             new Response(200, ['content-type' => null], Stream::factory(file_get_contents(__DIR__ . '/../fixtures/image-no-content-type.jpg'))),
 206         ]);
 207
 208         $client->getEmitter()->attach($mock);
 209
 210         $logHandler = new TestHandler();
 211         $logger = new Logger('test', [$logHandler]);
 212
 213         $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
 214
 215         $res = $download->processSingleImage(
 216             123,
 217             null,
 218             'https://framablog.org/2018/06/30/engagement-atypique/'
 219         );
 220         $this->assertFalse($res);
 221     }
 222 }