[github/wallabag/wallabag.git] / src / Wallabag / ImportBundle / Tests / Import / PocketImportTest.php

<?php

namespace Wallabag\ImportBundle\Tests\Import;

use Wallabag\UserBundle\Entity\User;
use Wallabag\ImportBundle\Import\PocketImport;
use GuzzleHttp\Client;
use GuzzleHttp\Subscriber\Mock;
use GuzzleHttp\Message\Response;
use GuzzleHttp\Stream\Stream;
use Monolog\Logger;
use Monolog\Handler\TestHandler;

class PocketImportMock extends PocketImport
{
    public function getAccessToken()
    {
        return $this->accessToken;
    }
}

class PocketImportTest extends \PHPUnit_Framework_TestCase
{
    protected $token;
    protected $user;
    protected $em;
    protected $contentProxy;
    protected $logHandler;

    private function getPocketImport($consumerKey = 'ConsumerKey')
    {
        $this->user = new User();

        $this->tokenStorage = $this->getMockBuilder('Symfony\Component\Security\Core\Authentication\Token\Storage\TokenStorageInterface')
            ->disableOriginalConstructor()
            ->getMock();

        $token = $this->getMockBuilder('Symfony\Component\Security\Core\Authentication\Token\TokenInterface')
            ->disableOriginalConstructor()
            ->getMock();

        $this->contentProxy = $this->getMockBuilder('Wallabag\CoreBundle\Helper\ContentProxy')
            ->disableOriginalConstructor()
            ->getMock();

        $token->expects($this->once())
            ->method('getUser')
            ->willReturn($this->user);

        $this->tokenStorage->expects($this->once())
            ->method('getToken')
            ->willReturn($token);

        $this->em = $this->getMockBuilder('Doctrine\ORM\EntityManager')
            ->disableOriginalConstructor()
            ->getMock();

        $pocket = new PocketImportMock(
            $this->tokenStorage,
            $this->em,
            $this->contentProxy,
            $consumerKey
        );

        $this->logHandler = new TestHandler();
        $logger = new Logger('test', array($this->logHandler));
        $pocket->setLogger($logger);

        return $pocket;
    }

    public function testInit()
    {
        $pocketImport = $this->getPocketImport();

        $this->assertEquals('Pocket', $pocketImport->getName());
        $this->assertEquals('This importer will import all your <a href="https://getpocket.com">Pocket</a> data.', $pocketImport->getDescription());
    }

    public function testOAuthRequest()
    {
        $client = new Client();

        $mock = new Mock([
            new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['code' => 'wunderbar_code']))),
        ]);

        $client->getEmitter()->attach($mock);

        $pocketImport = $this->getPocketImport();
        $pocketImport->setClient($client);

        $code = $pocketImport->getRequestToken('http://0.0.0.0/redirect');

        $this->assertEquals('wunderbar_code', $code);
    }

    public function testOAuthRequestBadResponse()
    {
        $client = new Client();

        $mock = new Mock([
            new Response(403),
        ]);

        $client->getEmitter()->attach($mock);

        $pocketImport = $this->getPocketImport();
        $pocketImport->setClient($client);

        $code = $pocketImport->getRequestToken('http://0.0.0.0/redirect');

        $this->assertFalse($code);

        $records = $this->logHandler->getRecords();
        $this->assertContains('PocketImport: Failed to request token', $records[0]['message']);
        $this->assertEquals('ERROR', $records[0]['level_name']);
    }

    public function testOAuthAuthorize()
    {
        $client = new Client();

        $mock = new Mock([
            new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['access_token' => 'wunderbar_token']))),
        ]);

        $client->getEmitter()->attach($mock);

        $pocketImport = $this->getPocketImport();
        $pocketImport->setClient($client);

        $res = $pocketImport->authorize('wunderbar_code');

        $this->assertTrue($res);
        $this->assertEquals('wunderbar_token', $pocketImport->getAccessToken());
    }

    public function testOAuthAuthorizeBadResponse()
    {
        $client = new Client();

        $mock = new Mock([
            new Response(403),
        ]);

        $client->getEmitter()->attach($mock);

        $pocketImport = $this->getPocketImport();
        $pocketImport->setClient($client);

        $res = $pocketImport->authorize('wunderbar_code');

        $this->assertFalse($res);

        $records = $this->logHandler->getRecords();
        $this->assertContains('PocketImport: Failed to authorize client', $records[0]['message']);
        $this->assertEquals('ERROR', $records[0]['level_name']);
    }

    /**
     * Will sample results from https://getpocket.com/developer/docs/v3/retrieve.
     */
    public function testImport()
    {
        $client = new Client();

        $mock = new Mock([
            new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['access_token' => 'wunderbar_token']))),
            new Response(200, ['Content-Type' => 'application/json'], Stream::factory('
                {
                    "status": 1,
                    "list": {
                        "229279689": {
                            "item_id": "229279689",
                            "resolved_id": "229279689",
                            "given_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview",
                            "given_title": "The Massive Ryder Cup Preview - The Triangle Blog - Grantland",
                            "favorite": "1",
                            "status": "1",
                            "resolved_title": "The Massive Ryder Cup Preview",
                            "resolved_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview",
                            "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.",
                            "is_article": "1",
                            "has_video": "1",
                            "has_image": "1",
                            "word_count": "3197",
                            "images": {
                                "1": {
                                    "item_id": "229279689",
                                    "image_id": "1",
                                    "src": "http://a.espncdn.com/combiner/i?img=/photo/2012/0927/grant_g_ryder_cr_640.jpg&w=640&h=360",
                                    "width": "0",
                                    "height": "0",
                                    "credit": "Jamie Squire/Getty Images",
                                    "caption": ""
                                }
                            },
                            "videos": {
                                "1": {
                                    "item_id": "229279689",
                                    "video_id": "1",
                                    "src": "http://www.youtube.com/v/Er34PbFkVGk?version=3&hl=en_US&rel=0",
                                    "width": "420",
                                    "height": "315",
                                    "type": "1",
                                    "vid": "Er34PbFkVGk"
                                }
                            },
                            "tags": {
                                "grantland": {
                                    "item_id": "1147652870",
                                    "tag": "grantland"
                                },
                                "Ryder Cup": {
                                    "item_id": "1147652870",
                                    "tag": "Ryder Cup"
                                }
                            }
                        },
                        "229279690": {
                            "item_id": "229279689",
                            "resolved_id": "229279689",
                            "given_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview",
                            "given_title": "The Massive Ryder Cup Preview - The Triangle Blog - Grantland",
                            "favorite": "1",
                            "status": "1",
                            "resolved_title": "The Massive Ryder Cup Preview",
                            "resolved_url": "http://www.grantland.com/blog/the-triangle/post/_/id/38347/ryder-cup-preview",
                            "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.",
                            "is_article": "1",
                            "has_video": "0",
                            "has_image": "0",
                            "word_count": "3197"
                        }
                    }
                }
            ')),
        ]);

        $client->getEmitter()->attach($mock);

        $pocketImport = $this->getPocketImport();

        $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
            ->disableOriginalConstructor()
            ->getMock();

        $entryRepo->expects($this->exactly(2))
            ->method('existByUrlAndUserId')
            ->will($this->onConsecutiveCalls(false, true));

        $tag = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Tag')
            ->disableOriginalConstructor()
            ->getMock();

        $tagRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\TagRepository')
            ->disableOriginalConstructor()
            ->getMock();

        $tagRepo->expects($this->exactly(2))
            ->method('findOneByLabelAndUserId')
            ->will($this->onConsecutiveCalls(false, $tag));

        $this->em
            ->expects($this->any())
            ->method('getRepository')
            ->will($this->onConsecutiveCalls($entryRepo, $tagRepo, $tagRepo, $entryRepo));

        $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry')
            ->disableOriginalConstructor()
            ->getMock();

        $this->contentProxy
            ->expects($this->once())
            ->method('updateEntry')
            ->willReturn($entry);

        $pocketImport->setClient($client);
        $pocketImport->authorize('wunderbar_code');

        $res = $pocketImport->import();

        $this->assertTrue($res);
        $this->assertEquals(['skipped' => 1, 'imported' => 1], $pocketImport->getSummary());
    }

    public function testImportBadResponse()
    {
        $client = new Client();

        $mock = new Mock([
            new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['access_token' => 'wunderbar_token']))),
            new Response(403),
        ]);

        $client->getEmitter()->attach($mock);

        $pocketImport = $this->getPocketImport();
        $pocketImport->setClient($client);
        $pocketImport->authorize('wunderbar_code');

        $res = $pocketImport->import();

        $this->assertFalse($res);

        $records = $this->logHandler->getRecords();
        $this->assertContains('PocketImport: Failed to import', $records[0]['message']);
        $this->assertEquals('ERROR', $records[0]['level_name']);
    }
}
Commit	Line	Data
7ec2897e JB	1	<?php
	2
	3	namespace Wallabag\ImportBundle\Tests\Import;
	4
	5	use Wallabag\UserBundle\Entity\User;
	6	use Wallabag\ImportBundle\Import\PocketImport;
7ec2897e JB	7	use GuzzleHttp\Client;
	8	use GuzzleHttp\Subscriber\Mock;
	9	use GuzzleHttp\Message\Response;
	10	use GuzzleHttp\Stream\Stream;
252ebd60 JB	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	}
7ec2897e JB	21
	22	class PocketImportTest extends \PHPUnit_Framework_TestCase
	23	{
	24	protected $token;
	25	protected $user;
7ec2897e	26	protected $em;
252ebd60 JB	27	protected $contentProxy;
252ebd60 JB	28	protected $logHandler;
7ec2897e JB	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
252ebd60 JB	42	$this->contentProxy = $this->getMockBuilder('Wallabag\CoreBundle\Helper\ContentProxy')
	43	->disableOriginalConstructor()
	44	->getMock();
	45
7ec2897e JB	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
7ec2897e JB	54	$this->em = $this->getMockBuilder('Doctrine\ORM\EntityManager')
	55	->disableOriginalConstructor()
	56	->getMock();
	57
252ebd60	58	$pocket = new PocketImportMock(
7ec2897e	59	$this->tokenStorage,
7ec2897e	60	$this->em,
252ebd60	61	$this->contentProxy,
7ec2897e JB	62	$consumerKey
7ec2897e JB	63	);
252ebd60 JB	64
	65	$this->logHandler = new TestHandler();
	66	$logger = new Logger('test', array($this->logHandler));
	67	$pocket->setLogger($logger);
	68
	69	return $pocket;
7ec2897e JB	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([
252ebd60	85	new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['code' => 'wunderbar_code']))),
7ec2897e JB	86	]);
	87
	88	$client->getEmitter()->attach($mock);
	89
	90	$pocketImport = $this->getPocketImport();
	91	$pocketImport->setClient($client);
	92
252ebd60	93	$code = $pocketImport->getRequestToken('http://0.0.0.0/redirect');
7ec2897e	94
252ebd60 JB	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']);
7ec2897e JB	118	}
	119
	120	public function testOAuthAuthorize()
	121	{
	122	$client = new Client();
	123
	124	$mock = new Mock([
252ebd60	125	new Response(200, ['Content-Type' => 'application/json'], Stream::factory(json_encode(['access_token' => 'wunderbar_token']))),
7ec2897e JB	126	]);
	127
	128	$client->getEmitter()->attach($mock);
	129
	130	$pocketImport = $this->getPocketImport();
	131	$pocketImport->setClient($client);
	132
252ebd60	133	$res = $pocketImport->authorize('wunderbar_code');
7ec2897e	134
252ebd60 JB	135	$this->assertTrue($res);
252ebd60 JB	136	$this->assertEquals('wunderbar_token', $pocketImport->getAccessToken());
7ec2897e JB	137	}
7ec2897e JB	138
252ebd60 JB	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	*/
7ec2897e JB	164	public function testImport()
	165	{
	166	$client = new Client();
	167
	168	$mock = new Mock([
252ebd60 JB	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),
7ec2897e JB	295	]);
	296
	297	$client->getEmitter()->attach($mock);
	298
	299	$pocketImport = $this->getPocketImport();
	300	$pocketImport->setClient($client);
252ebd60 JB	301	$pocketImport->authorize('wunderbar_code');
	302
	303	$res = $pocketImport->import();
7ec2897e	304
252ebd60	305	$this->assertFalse($res);
7ec2897e	306
252ebd60 JB	307	$records = $this->logHandler->getRecords();
	308	$this->assertContains('PocketImport: Failed to import', $records[0]['message']);
	309	$this->assertEquals('ERROR', $records[0]['level_name']);
7ec2897e JB	310	}
7ec2897e JB	311	}