3 namespace Tests\Wallabag\ImportBundle\Import
;
5 use Wallabag\ImportBundle\Import\ReadabilityImport
;
6 use Wallabag\UserBundle\Entity\User
;
7 use Wallabag\CoreBundle\Entity\Entry
;
8 use Wallabag\ImportBundle\Redis\Producer
;
10 use Monolog\Handler\TestHandler
;
11 use Simpleue\Queue\RedisQueue
;
12 use M6Web\Component\RedisMock\RedisMockFactory
;
14 class ReadabilityImportTest
extends \PHPUnit_Framework_TestCase
18 protected $logHandler;
19 protected $contentProxy;
21 private function getReadabilityImport($unsetUser = false, $dispatched = 0)
23 $this->user
= new User();
25 $this->em
= $this->getMockBuilder('Doctrine\ORM\EntityManager')
26 ->disableOriginalConstructor()
29 $this->contentProxy
= $this->getMockBuilder('Wallabag\CoreBundle\Helper\ContentProxy')
30 ->disableOriginalConstructor()
33 $dispatcher = $this->getMockBuilder('Symfony\Component\EventDispatcher\EventDispatcher')
34 ->disableOriginalConstructor()
38 ->expects($this->exactly($dispatched))
41 $wallabag = new ReadabilityImport($this->em
, $this->contentProxy
, $dispatcher);
43 $this->logHandler
= new TestHandler();
44 $logger = new Logger('test', [$this->logHandler
]);
45 $wallabag->setLogger($logger);
47 if (false === $unsetUser) {
48 $wallabag->setUser($this->user
);
54 public function testInit()
56 $readabilityImport = $this->getReadabilityImport();
58 $this->assertEquals('Readability', $readabilityImport->getName());
59 $this->assertNotEmpty($readabilityImport->getUrl());
60 $this->assertEquals('import.readability.description', $readabilityImport->getDescription());
63 public function testImport()
65 $readabilityImport = $this->getReadabilityImport(false, 24);
66 $readabilityImport->setFilepath(__DIR__
.'/../fixtures/readability.json');
68 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
69 ->disableOriginalConstructor()
72 $entryRepo->expects($this->exactly(24))
73 ->method('findByUrlAndUserId')
77 ->expects($this->any())
78 ->method('getRepository')
79 ->willReturn($entryRepo);
81 $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry')
82 ->disableOriginalConstructor()
86 ->expects($this->exactly(24))
87 ->method('updateEntry')
90 $res = $readabilityImport->import();
92 $this->assertTrue($res);
93 $this->assertEquals(['skipped' => 0, 'imported' => 24, 'queued' => 0], $readabilityImport->getSummary());
96 public function testImportAndMarkAllAsRead()
98 $readabilityImport = $this->getReadabilityImport(false, 1);
99 $readabilityImport->setFilepath(__DIR__
.'/../fixtures/readability-read.json');
101 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
102 ->disableOriginalConstructor()
105 $entryRepo->expects($this->exactly(2))
106 ->method('findByUrlAndUserId')
107 ->will($this->onConsecutiveCalls(false, true));
110 ->expects($this->any())
111 ->method('getRepository')
112 ->willReturn($entryRepo);
115 ->expects($this->exactly(1))
116 ->method('updateEntry')
117 ->willReturn(new Entry($this->user
));
119 // check that every entry persisted are archived
121 ->expects($this->any())
123 ->with($this->callback(function ($persistedEntry) {
124 return $persistedEntry->isArchived();
127 $res = $readabilityImport->setMarkAsRead(true)->import();
129 $this->assertTrue($res);
131 $this->assertEquals(['skipped' => 1, 'imported' => 1, 'queued' => 0], $readabilityImport->getSummary());
134 public function testImportWithRabbit()
136 $readabilityImport = $this->getReadabilityImport();
137 $readabilityImport->setFilepath(__DIR__
.'/../fixtures/readability.json');
139 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
140 ->disableOriginalConstructor()
143 $entryRepo->expects($this->never())
144 ->method('findByUrlAndUserId');
147 ->expects($this->never())
148 ->method('getRepository');
150 $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry')
151 ->disableOriginalConstructor()
155 ->expects($this->never())
156 ->method('updateEntry');
158 $producer = $this->getMockBuilder('OldSound\RabbitMqBundle\RabbitMq\Producer')
159 ->disableOriginalConstructor()
163 ->expects($this->exactly(24))
166 $readabilityImport->setProducer($producer);
168 $res = $readabilityImport->setMarkAsRead(true)->import();
170 $this->assertTrue($res);
171 $this->assertEquals(['skipped' => 0, 'imported' => 0, 'queued' => 24], $readabilityImport->getSummary());
174 public function testImportWithRedis()
176 $readabilityImport = $this->getReadabilityImport();
177 $readabilityImport->setFilepath(__DIR__
.'/../fixtures/readability.json');
179 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
180 ->disableOriginalConstructor()
183 $entryRepo->expects($this->never())
184 ->method('findByUrlAndUserId');
187 ->expects($this->never())
188 ->method('getRepository');
190 $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry')
191 ->disableOriginalConstructor()
195 ->expects($this->never())
196 ->method('updateEntry');
198 $factory = new RedisMockFactory();
199 $redisMock = $factory->getAdapter('Predis\Client', true);
201 $queue = new RedisQueue($redisMock, 'readability');
202 $producer = new Producer($queue);
204 $readabilityImport->setProducer($producer);
206 $res = $readabilityImport->setMarkAsRead(true)->import();
208 $this->assertTrue($res);
209 $this->assertEquals(['skipped' => 0, 'imported' => 0, 'queued' => 24], $readabilityImport->getSummary());
211 $this->assertNotEmpty($redisMock->lpop('readability'));
214 public function testImportBadFile()
216 $readabilityImport = $this->getReadabilityImport();
217 $readabilityImport->setFilepath(__DIR__
.'/../fixtures/wallabag-v1.jsonx');
219 $res = $readabilityImport->import();
221 $this->assertFalse($res);
223 $records = $this->logHandler
->getRecords();
224 $this->assertContains('ReadabilityImport: unable to read file', $records[0]['message']);
225 $this->assertEquals('ERROR', $records[0]['level_name']);
228 public function testImportUserNotDefined()
230 $readabilityImport = $this->getReadabilityImport(true);
231 $readabilityImport->setFilepath(__DIR__
.'/../fixtures/readability.json');
233 $res = $readabilityImport->import();
235 $this->assertFalse($res);
237 $records = $this->logHandler
->getRecords();
238 $this->assertContains('ReadabilityImport: user is not defined', $records[0]['message']);
239 $this->assertEquals('ERROR', $records[0]['level_name']);