3 namespace Wallabag\ImportBundle\Tests\Import
;
5 use Wallabag\UserBundle\Entity\User
;
6 use Wallabag\ImportBundle\Import\PocketImport
;
8 use GuzzleHttp\Subscriber\Mock
;
9 use GuzzleHttp\Message\Response
;
10 use GuzzleHttp\Stream\Stream
;
12 use Monolog\Handler\TestHandler
;
14 class PocketImportMock
extends PocketImport
16 public function getAccessToken()
18 return $this->accessToken
;
22 class PocketImportTest
extends \PHPUnit_Framework_TestCase
27 protected $contentProxy;
28 protected $logHandler;
30 private function getPocketImport($consumerKey = 'ConsumerKey')
32 $this->user
= new User();
34 $this->tokenStorage
= $this->getMockBuilder('Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface')
35 ->disableOriginalConstructor()
38 $token = $this->getMockBuilder('Symfony\Component\Security\Core\Authentication\Token\TokenInterface')
39 ->disableOriginalConstructor()
42 $this->contentProxy
= $this->getMockBuilder('Wallabag\CoreBundle\Helper\ContentProxy')
43 ->disableOriginalConstructor()
46 $token->expects($this->once())
48 ->willReturn($this->user
);
50 $this->tokenStorage
->expects($this->once())
54 $this->em
= $this->getMockBuilder('Doctrine\ORM\EntityManager')
55 ->disableOriginalConstructor()
58 $pocket = new PocketImportMock(
65 $this->logHandler
= new TestHandler();
66 $logger = new Logger('test', array($this->logHandler
));
67 $pocket->setLogger($logger);
72 public function testInit()
74 $pocketImport = $this->getPocketImport();
76 $this->assertEquals('Pocket', $pocketImport->getName());
77 $this->assertNotEmpty($pocketImport->getUrl());
78 $this->assertContains('This importer will import all your Pocket data.', $pocketImport->getDescription());
81 public function testOAuthRequest()
83 $client = new Client();
86 new Response(200, ['Content-Type' => 'application/json'], Stream
::factory(json_encode(['code' => 'wunderbar_code']))),
89 $client->getEmitter()->attach($mock);
91 $pocketImport = $this->getPocketImport();
92 $pocketImport->setClient($client);
94 $code = $pocketImport->getRequestToken('http://0.0.0.0/redirect');
96 $this->assertEquals('wunderbar_code', $code);
99 public function testOAuthRequestBadResponse()
101 $client = new Client();
107 $client->getEmitter()->attach($mock);
109 $pocketImport = $this->getPocketImport();
110 $pocketImport->setClient($client);
112 $code = $pocketImport->getRequestToken('http://0.0.0.0/redirect');
114 $this->assertFalse($code);
116 $records = $this->logHandler
->getRecords();
117 $this->assertContains('PocketImport: Failed to request token', $records[0]['message']);
118 $this->assertEquals('ERROR', $records[0]['level_name']);
121 public function testOAuthAuthorize()
123 $client = new Client();
126 new Response(200, ['Content-Type' => 'application/json'], Stream
::factory(json_encode(['access_token' => 'wunderbar_token']))),
129 $client->getEmitter()->attach($mock);
131 $pocketImport = $this->getPocketImport();
132 $pocketImport->setClient($client);
134 $res = $pocketImport->authorize('wunderbar_code');
136 $this->assertTrue($res);
137 $this->assertEquals('wunderbar_token', $pocketImport->getAccessToken());
140 public function testOAuthAuthorizeBadResponse()
142 $client = new Client();
148 $client->getEmitter()->attach($mock);
150 $pocketImport = $this->getPocketImport();
151 $pocketImport->setClient($client);
153 $res = $pocketImport->authorize('wunderbar_code');
155 $this->assertFalse($res);
157 $records = $this->logHandler
->getRecords();
158 $this->assertContains('PocketImport: Failed to authorize client', $records[0]['message']);
159 $this->assertEquals('ERROR', $records[0]['level_name']);
163 * Will sample results from https://getpocket.com/developer/docs/v3/retrieve.
165 public function testImport()
167 $client = new Client();
170 new Response(200, ['Content-Type' => 'application/json'], Stream
::factory(json_encode(['access_token' => 'wunderbar_token']))),
171 new Response(200, ['Content-Type' => 'application/json'], Stream
::factory('
176 "item_id": "229279689",
177 "resolved_id": "229279689",
178 "given_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview",
179 "given_title": "The Massive Ryder Cup Preview - The Triangle Blog - Grantland",
182 "resolved_title": "The Massive Ryder Cup Preview",
183 "resolved_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview",
184 "excerpt": "The list of things I love about the Ryder Cup is so long that it could fill a (tedious) novel, and golf fans can probably guess most of them.",
188 "word_count": "3197",
191 "item_id": "229279689",
193 "src": "http://a.espncdn.com/combiner/i?img=/photo/2012/0927/grant_g_ryder_cr_640.jpg&w=640&h=360",
196 "credit": "Jamie Squire/Getty Images",
202 "item_id": "229279689",
204 "src": "http://www.youtube.com/v/Er34PbFkVGk?version=3&hl=en_US&rel=0",
213 "item_id": "1147652870",
217 "item_id": "1147652870",
223 "item_id": "229279689",
224 "resolved_id": "229279689",
225 "given_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview",
226 "given_title": "The Massive Ryder Cup Preview - The Triangle Blog - Grantland",
229 "resolved_title": "The Massive Ryder Cup Preview",
230 "resolved_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview",
231 "excerpt": "The list of things I love about the Ryder Cup is so long that it could fill a (tedious) novel, and golf fans can probably guess most of them.",
242 $client->getEmitter()->attach($mock);
244 $pocketImport = $this->getPocketImport();
246 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
247 ->disableOriginalConstructor()
250 $entryRepo->expects($this->exactly(2))
251 ->method('findByUrlAndUserId')
252 ->will($this->onConsecutiveCalls(false, true));
254 $tag = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Tag')
255 ->disableOriginalConstructor()
258 $tagRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\TagRepository')
259 ->disableOriginalConstructor()
262 $tagRepo->expects($this->exactly(2))
263 // the method `findOneByLabel` doesn't exist, EntityRepository will then call `_call` method
264 // to magically call the `findOneBy` with ['label' => 'foo']
266 ->will($this->onConsecutiveCalls(false, $tag));
269 ->expects($this->any())
270 ->method('getRepository')
271 ->will($this->onConsecutiveCalls($entryRepo, $tagRepo, $tagRepo, $entryRepo));
273 $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry')
274 ->disableOriginalConstructor()
278 ->expects($this->once())
279 ->method('updateEntry')
280 ->willReturn($entry);
282 $pocketImport->setClient($client);
283 $pocketImport->authorize('wunderbar_code');
285 $res = $pocketImport->import();
287 $this->assertTrue($res);
288 $this->assertEquals(['skipped' => 1, 'imported' => 1], $pocketImport->getSummary());
291 public function testImportBadResponse()
293 $client = new Client();
296 new Response(200, ['Content-Type' => 'application/json'], Stream
::factory(json_encode(['access_token' => 'wunderbar_token']))),
300 $client->getEmitter()->attach($mock);
302 $pocketImport = $this->getPocketImport();
303 $pocketImport->setClient($client);
304 $pocketImport->authorize('wunderbar_code');
306 $res = $pocketImport->import();
308 $this->assertFalse($res);
310 $records = $this->logHandler
->getRecords();
311 $this->assertContains('PocketImport: Failed to import', $records[0]['message']);
312 $this->assertEquals('ERROR', $records[0]['level_name']);