]>
Commit | Line | Data |
---|---|---|
1 | <?php | |
2 | ||
3 | namespace Wallabag\ImportBundle\Tests\Import; | |
4 | ||
5 | use Wallabag\UserBundle\Entity\User; | |
6 | use Wallabag\ImportBundle\Import\PocketImport; | |
7 | use GuzzleHttp\Client; | |
8 | use GuzzleHttp\Subscriber\Mock; | |
9 | use GuzzleHttp\Message\Response; | |
10 | use GuzzleHttp\Stream\Stream; | |
11 | use Monolog\Logger; | |
12 | use Monolog\Handler\TestHandler; | |
13 | ||
14 | class PocketImportMock extends PocketImport | |
15 | { | |
16 | public function getAccessToken() | |
17 | { | |
18 | return $this->accessToken; | |
19 | } | |
20 | } | |
21 | ||
22 | class PocketImportTest extends \PHPUnit_Framework_TestCase | |
23 | { | |
24 | protected $token; | |
25 | protected $user; | |
26 | protected $em; | |
27 | protected $contentProxy; | |
28 | protected $logHandler; | |
29 | ||
30 | private function getPocketImport($consumerKey = 'ConsumerKey') | |
31 | { | |
32 | $this->user = new User(); | |
33 | ||
34 | $this->tokenStorage = $this->getMockBuilder('Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface') | |
35 | ->disableOriginalConstructor() | |
36 | ->getMock(); | |
37 | ||
38 | $token = $this->getMockBuilder('Symfony\Component\Security\Core\Authentication\Token\TokenInterface') | |
39 | ->disableOriginalConstructor() | |
40 | ->getMock(); | |
41 | ||
42 | $this->contentProxy = $this->getMockBuilder('Wallabag\CoreBundle\Helper\ContentProxy') | |
43 | ->disableOriginalConstructor() | |
44 | ->getMock(); | |
45 | ||
46 | $token->expects($this->once()) | |
47 | ->method('getUser') | |
48 | ->willReturn($this->user); | |
49 | ||
50 | $this->tokenStorage->expects($this->once()) | |
51 | ->method('getToken') | |
52 | ->willReturn($token); | |
53 | ||
54 | $this->em = $this->getMockBuilder('Doctrine\ORM\EntityManager') | |
55 | ->disableOriginalConstructor() | |
56 | ->getMock(); | |
57 | ||
58 | $pocket = new PocketImportMock( | |
59 | $this->tokenStorage, | |
60 | $this->em, | |
61 | $this->contentProxy, | |
62 | $consumerKey | |
63 | ); | |
64 | ||
65 | $this->logHandler = new TestHandler(); | |
66 | $logger = new Logger('test', array($this->logHandler)); | |
67 | $pocket->setLogger($logger); | |
68 | ||
69 | return $pocket; | |
70 | } | |
71 | ||
72 | public function testInit() | |
73 | { | |
74 | $pocketImport = $this->getPocketImport(); | |
75 | ||
76 | $this->assertEquals('Pocket', $pocketImport->getName()); | |
77 | $this->assertEquals('This importer will import all your <a href="https://getpocket.com">Pocket</a> data.', $pocketImport->getDescription()); | |
78 | } | |
79 | ||
80 | public function testOAuthRequest() | |
81 | { | |
82 | $client = new Client(); | |
83 | ||
84 | $mock = new Mock([ | |
85 | new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['code' => 'wunderbar_code']))), | |
86 | ]); | |
87 | ||
88 | $client->getEmitter()->attach($mock); | |
89 | ||
90 | $pocketImport = $this->getPocketImport(); | |
91 | $pocketImport->setClient($client); | |
92 | ||
93 | $code = $pocketImport->getRequestToken('http://0.0.0.0/redirect'); | |
94 | ||
95 | $this->assertEquals('wunderbar_code', $code); | |
96 | } | |
97 | ||
98 | public function testOAuthRequestBadResponse() | |
99 | { | |
100 | $client = new Client(); | |
101 | ||
102 | $mock = new Mock([ | |
103 | new Response(403), | |
104 | ]); | |
105 | ||
106 | $client->getEmitter()->attach($mock); | |
107 | ||
108 | $pocketImport = $this->getPocketImport(); | |
109 | $pocketImport->setClient($client); | |
110 | ||
111 | $code = $pocketImport->getRequestToken('http://0.0.0.0/redirect'); | |
112 | ||
113 | $this->assertFalse($code); | |
114 | ||
115 | $records = $this->logHandler->getRecords(); | |
116 | $this->assertContains('PocketImport: Failed to request token', $records[0]['message']); | |
117 | $this->assertEquals('ERROR', $records[0]['level_name']); | |
118 | } | |
119 | ||
120 | public function testOAuthAuthorize() | |
121 | { | |
122 | $client = new Client(); | |
123 | ||
124 | $mock = new Mock([ | |
125 | new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['access_token' => 'wunderbar_token']))), | |
126 | ]); | |
127 | ||
128 | $client->getEmitter()->attach($mock); | |
129 | ||
130 | $pocketImport = $this->getPocketImport(); | |
131 | $pocketImport->setClient($client); | |
132 | ||
133 | $res = $pocketImport->authorize('wunderbar_code'); | |
134 | ||
135 | $this->assertTrue($res); | |
136 | $this->assertEquals('wunderbar_token', $pocketImport->getAccessToken()); | |
137 | } | |
138 | ||
139 | public function testOAuthAuthorizeBadResponse() | |
140 | { | |
141 | $client = new Client(); | |
142 | ||
143 | $mock = new Mock([ | |
144 | new Response(403), | |
145 | ]); | |
146 | ||
147 | $client->getEmitter()->attach($mock); | |
148 | ||
149 | $pocketImport = $this->getPocketImport(); | |
150 | $pocketImport->setClient($client); | |
151 | ||
152 | $res = $pocketImport->authorize('wunderbar_code'); | |
153 | ||
154 | $this->assertFalse($res); | |
155 | ||
156 | $records = $this->logHandler->getRecords(); | |
157 | $this->assertContains('PocketImport: Failed to authorize client', $records[0]['message']); | |
158 | $this->assertEquals('ERROR', $records[0]['level_name']); | |
159 | } | |
160 | ||
161 | /** | |
162 | * Will sample results from https://getpocket.com/developer/docs/v3/retrieve. | |
163 | */ | |
164 | public function testImport() | |
165 | { | |
166 | $client = new Client(); | |
167 | ||
168 | $mock = new Mock([ | |
169 | new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['access_token' => 'wunderbar_token']))), | |
170 | new Response(200, ['Content-Type' => 'application/json'], Stream::factory(' | |
171 | { | |
172 | "status": 1, | |
173 | "list": { | |
174 | "229279689": { | |
175 | "item_id": "229279689", | |
176 | "resolved_id": "229279689", | |
177 | "given_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview", | |
178 | "given_title": "The Massive Ryder Cup Preview - The Triangle Blog - Grantland", | |
179 | "favorite": "1", | |
180 | "status": "1", | |
181 | "resolved_title": "The Massive Ryder Cup Preview", | |
182 | "resolved_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview", | |
183 | "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.", | |
184 | "is_article": "1", | |
185 | "has_video": "1", | |
186 | "has_image": "1", | |
187 | "word_count": "3197", | |
188 | "images": { | |
189 | "1": { | |
190 | "item_id": "229279689", | |
191 | "image_id": "1", | |
192 | "src": "http://a.espncdn.com/combiner/i?img=/photo/2012/0927/grant_g_ryder_cr_640.jpg&w=640&h=360", | |
193 | "width": "0", | |
194 | "height": "0", | |
195 | "credit": "Jamie Squire/Getty Images", | |
196 | "caption": "" | |
197 | } | |
198 | }, | |
199 | "videos": { | |
200 | "1": { | |
201 | "item_id": "229279689", | |
202 | "video_id": "1", | |
203 | "src": "http://www.youtube.com/v/Er34PbFkVGk?version=3&hl=en_US&rel=0", | |
204 | "width": "420", | |
205 | "height": "315", | |
206 | "type": "1", | |
207 | "vid": "Er34PbFkVGk" | |
208 | } | |
209 | }, | |
210 | "tags": { | |
211 | "grantland": { | |
212 | "item_id": "1147652870", | |
213 | "tag": "grantland" | |
214 | }, | |
215 | "Ryder Cup": { | |
216 | "item_id": "1147652870", | |
217 | "tag": "Ryder Cup" | |
218 | } | |
219 | } | |
220 | }, | |
221 | "229279690": { | |
222 | "item_id": "229279689", | |
223 | "resolved_id": "229279689", | |
224 | "given_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview", | |
225 | "given_title": "The Massive Ryder Cup Preview - The Triangle Blog - Grantland", | |
226 | "favorite": "1", | |
227 | "status": "1", | |
228 | "resolved_title": "The Massive Ryder Cup Preview", | |
229 | "resolved_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview", | |
230 | "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.", | |
231 | "is_article": "1", | |
232 | "has_video": "0", | |
233 | "has_image": "0", | |
234 | "word_count": "3197" | |
235 | } | |
236 | } | |
237 | } | |
238 | ')), | |
239 | ]); | |
240 | ||
241 | $client->getEmitter()->attach($mock); | |
242 | ||
243 | $pocketImport = $this->getPocketImport(); | |
244 | ||
245 | $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository') | |
246 | ->disableOriginalConstructor() | |
247 | ->getMock(); | |
248 | ||
249 | $entryRepo->expects($this->exactly(2)) | |
250 | ->method('existByUrlAndUserId') | |
251 | ->will($this->onConsecutiveCalls(false, true)); | |
252 | ||
253 | $tag = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Tag') | |
254 | ->disableOriginalConstructor() | |
255 | ->getMock(); | |
256 | ||
257 | $tagRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\TagRepository') | |
258 | ->disableOriginalConstructor() | |
259 | ->getMock(); | |
260 | ||
261 | $tagRepo->expects($this->exactly(2)) | |
262 | ->method('findOneByLabelAndUserId') | |
263 | ->will($this->onConsecutiveCalls(false, $tag)); | |
264 | ||
265 | $this->em | |
266 | ->expects($this->any()) | |
267 | ->method('getRepository') | |
268 | ->will($this->onConsecutiveCalls($entryRepo, $tagRepo, $tagRepo, $entryRepo)); | |
269 | ||
270 | $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry') | |
271 | ->disableOriginalConstructor() | |
272 | ->getMock(); | |
273 | ||
274 | $this->contentProxy | |
275 | ->expects($this->once()) | |
276 | ->method('updateEntry') | |
277 | ->willReturn($entry); | |
278 | ||
279 | $pocketImport->setClient($client); | |
280 | $pocketImport->authorize('wunderbar_code'); | |
281 | ||
282 | $res = $pocketImport->import(); | |
283 | ||
284 | $this->assertTrue($res); | |
285 | $this->assertEquals(['skipped' => 1, 'imported' => 1], $pocketImport->getSummary()); | |
286 | } | |
287 | ||
288 | public function testImportBadResponse() | |
289 | { | |
290 | $client = new Client(); | |
291 | ||
292 | $mock = new Mock([ | |
293 | new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['access_token' => 'wunderbar_token']))), | |
294 | new Response(403), | |
295 | ]); | |
296 | ||
297 | $client->getEmitter()->attach($mock); | |
298 | ||
299 | $pocketImport = $this->getPocketImport(); | |
300 | $pocketImport->setClient($client); | |
301 | $pocketImport->authorize('wunderbar_code'); | |
302 | ||
303 | $res = $pocketImport->import(); | |
304 | ||
305 | $this->assertFalse($res); | |
306 | ||
307 | $records = $this->logHandler->getRecords(); | |
308 | $this->assertContains('PocketImport: Failed to import', $records[0]['message']); | |
309 | $this->assertEquals('ERROR', $records[0]['level_name']); | |
310 | } | |
311 | } |