3 namespace Tests\Wallabag\CoreBundle\Helper
;
6 use GuzzleHttp\Message\Response
;
7 use GuzzleHttp\Stream\Stream
;
8 use GuzzleHttp\Subscriber\Mock
;
9 use Monolog\Handler\TestHandler
;
11 use PHPUnit\Framework\TestCase
;
12 use Wallabag\CoreBundle\Helper\DownloadImages
;
14 class DownloadImagesTest
extends TestCase
16 public function dataForSuccessImage()
20 '<div><img src="http://i.imgur.com/T9qgcHc.jpg" /></div>',
21 'http://imgur.com/gallery/WxtWY',
24 '<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&ssl=1" /></div>',
25 'https://www.tvaddons.ag/realdebrid-kodi-jarvis/',
31 * @dataProvider dataForSuccessImage
33 public function testProcessHtml($html, $url)
35 $client = new Client();
38 new Response(200, ['content-type' => 'image/png'], Stream
::factory(file_get_contents(__DIR__
. '/../fixtures/unnamed.png'))),
41 $client->getEmitter()->attach($mock);
43 $logHandler = new TestHandler();
44 $logger = new Logger('test', [$logHandler]);
46 $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
48 $res = $download->processHtml(123, $html, $url);
50 // this the base path of all image (since it's calculated using the entry id: 123)
51 $this->assertContains('http://wallabag.io/assets/images/9/b/9b0ead26/', $res);
54 public function testProcessHtmlWithBadImage()
56 $client = new Client();
59 new Response(200, ['content-type' => 'application/json'], Stream
::factory('')),
62 $client->getEmitter()->attach($mock);
64 $logHandler = new TestHandler();
65 $logger = new Logger('test', [$logHandler]);
67 $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
68 $res = $download->processHtml(123, '<div><img src="http://i.imgur.com/T9qgcHc.jpg" /></div>', 'http://imgur.com/gallery/WxtWY');
70 $this->assertContains('http://i.imgur.com/T9qgcHc.jpg', $res, 'Image were not replace because of content-type');
73 public function singleImage()
76 ['image/pjpeg', 'jpeg'],
77 ['image/jpeg', 'jpeg'],
84 * @dataProvider singleImage
86 public function testProcessSingleImage($header, $extension)
88 $client = new Client();
91 new Response(200, ['content-type' => $header], Stream
::factory(file_get_contents(__DIR__
. '/../fixtures/unnamed.png'))),
94 $client->getEmitter()->attach($mock);
96 $logHandler = new TestHandler();
97 $logger = new Logger('test', [$logHandler]);
99 $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
100 $res = $download->processSingleImage(123, 'T9qgcHc.jpg', 'http://imgur.com/gallery/WxtWY');
102 $this->assertContains('/assets/images/9/b/9b0ead26/ebe60399.' . $extension, $res);
105 public function testProcessSingleImageWithBadUrl()
107 $client = new Client();
110 new Response(404, []),
113 $client->getEmitter()->attach($mock);
115 $logHandler = new TestHandler();
116 $logger = new Logger('test', [$logHandler]);
118 $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
119 $res = $download->processSingleImage(123, 'T9qgcHc.jpg', 'http://imgur.com/gallery/WxtWY');
121 $this->assertFalse($res, 'Image can not be found, so it will not be replaced');
124 public function testProcessSingleImageWithBadImage()
126 $client = new Client();
129 new Response(200, ['content-type' => 'image/png'], Stream
::factory('')),
132 $client->getEmitter()->attach($mock);
134 $logHandler = new TestHandler();
135 $logger = new Logger('test', [$logHandler]);
137 $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
138 $res = $download->processSingleImage(123, 'http://i.imgur.com/T9qgcHc.jpg', 'http://imgur.com/gallery/WxtWY');
140 $this->assertFalse($res, 'Image can not be loaded, so it will not be replaced');
143 public function testProcessSingleImageFailAbsolute()
145 $client = new Client();
148 new Response(200, ['content-type' => 'image/png'], Stream
::factory(file_get_contents(__DIR__
. '/../fixtures/unnamed.png'))),
151 $client->getEmitter()->attach($mock);
153 $logHandler = new TestHandler();
154 $logger = new Logger('test', [$logHandler]);
156 $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
157 $res = $download->processSingleImage(123, '/i.imgur.com/T9qgcHc.jpg', 'imgur.com/gallery/WxtWY');
159 $this->assertFalse($res, 'Absolute image can not be determined, so it will not be replaced');
162 public function testProcessRealImage()
164 $client = new Client();
167 new Response(200, ['content-type' => null], Stream
::factory(file_get_contents(__DIR__
. '/../fixtures/image-no-content-type.jpg'))),
170 $client->getEmitter()->attach($mock);
172 $logHandler = new TestHandler();
173 $logger = new Logger('test', [$logHandler]);
175 $download = new DownloadImages($client, sys_get_temp_dir() . '/wallabag_test', 'http://wallabag.io/', $logger);
177 $res = $download->processSingleImage(
179 'https://cdn.theconversation.com/files/157200/article/width926/gsj2rjp2-1487348607.jpg',
180 'https://theconversation.com/conversation-avec-gerald-bronner-ce-nest-pas-la-post-verite-qui-nous-menace-mais-lextension-de-notre-credulite-73089'
183 $this->assertContains('http://wallabag.io/assets/images/9/b/9b0ead26/', $res, 'Content-Type was empty but data is ok for an image');
184 $this->assertContains('DownloadImages: Checking extension (alternative)', $logHandler->getRecords()[3]['message']);