3 namespace Tests\Wallabag\ImportBundle\Import
;
5 use Wallabag\ImportBundle\Import\WallabagV2Import
;
6 use Wallabag\UserBundle\Entity\User
;
7 use Wallabag\CoreBundle\Entity\Entry
;
9 use Monolog\Handler\TestHandler
;
11 class WallabagV2ImportTest
extends \PHPUnit_Framework_TestCase
15 protected $logHandler;
16 protected $contentProxy;
18 private function getWallabagV2Import($unsetUser = false)
20 $this->user
= new User();
22 $this->em
= $this->getMockBuilder('Doctrine\ORM\EntityManager')
23 ->disableOriginalConstructor()
26 $this->contentProxy
= $this->getMockBuilder('Wallabag\CoreBundle\Helper\ContentProxy')
27 ->disableOriginalConstructor()
30 $wallabag = new WallabagV2Import($this->em
, $this->contentProxy
);
32 $this->logHandler
= new TestHandler();
33 $logger = new Logger('test', [$this->logHandler
]);
34 $wallabag->setLogger($logger);
36 if (false === $unsetUser) {
37 $wallabag->setUser($this->user
);
43 public function testInit()
45 $wallabagV2Import = $this->getWallabagV2Import();
47 $this->assertEquals('wallabag v2', $wallabagV2Import->getName());
48 $this->assertNotEmpty($wallabagV2Import->getUrl());
49 $this->assertEquals('import.wallabag_v2.description', $wallabagV2Import->getDescription());
52 public function testImport()
54 $wallabagV2Import = $this->getWallabagV2Import();
55 $wallabagV2Import->setFilepath(__DIR__
.'/../fixtures/wallabag-v2.json');
57 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
58 ->disableOriginalConstructor()
61 $entryRepo->expects($this->exactly(24))
62 ->method('findByUrlAndUserId')
63 ->will($this->onConsecutiveCalls(false, true, false));
66 ->expects($this->any())
67 ->method('getRepository')
68 ->willReturn($entryRepo);
71 ->expects($this->exactly(2))
72 ->method('updateEntry')
73 ->willReturn(new Entry($this->user
));
75 $res = $wallabagV2Import->import();
77 $this->assertTrue($res);
78 $this->assertEquals(['skipped' => 22, 'imported' => 2], $wallabagV2Import->getSummary());
81 public function testImportAndMarkAllAsRead()
83 $wallabagV2Import = $this->getWallabagV2Import();
84 $wallabagV2Import->setFilepath(__DIR__
.'/../fixtures/wallabag-v2-read.json');
86 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
87 ->disableOriginalConstructor()
90 $entryRepo->expects($this->exactly(2))
91 ->method('findByUrlAndUserId')
92 ->will($this->onConsecutiveCalls(false, false));
95 ->expects($this->any())
96 ->method('getRepository')
97 ->willReturn($entryRepo);
100 ->expects($this->exactly(2))
101 ->method('updateEntry')
102 ->willReturn(new Entry($this->user
));
104 // check that every entry persisted are archived
106 ->expects($this->any())
108 ->with($this->callback(function ($persistedEntry) {
109 return $persistedEntry->isArchived();
112 $res = $wallabagV2Import->setMarkAsRead(true)->import();
114 $this->assertTrue($res);
116 $this->assertEquals(['skipped' => 0, 'imported' => 2], $wallabagV2Import->getSummary());
119 public function testImportWithRabbit()
121 $wallabagV2Import = $this->getWallabagV2Import();
122 $wallabagV2Import->setFilepath(__DIR__
.'/../fixtures/wallabag-v2.json');
124 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
125 ->disableOriginalConstructor()
128 $entryRepo->expects($this->never())
129 ->method('findByUrlAndUserId');
132 ->expects($this->never())
133 ->method('getRepository');
136 ->expects($this->never())
137 ->method('updateEntry');
139 $producer = $this->getMockBuilder('OldSound\RabbitMqBundle\RabbitMq\Producer')
140 ->disableOriginalConstructor()
144 ->expects($this->exactly(24))
147 $wallabagV2Import->setRabbitmqProducer($producer);
149 $res = $wallabagV2Import->setMarkAsRead(true)->import();
151 $this->assertTrue($res);
152 $this->assertEquals(['skipped' => 0, 'imported' => 24], $wallabagV2Import->getSummary());
155 public function testImportBadFile()
157 $wallabagV1Import = $this->getWallabagV2Import();
158 $wallabagV1Import->setFilepath(__DIR__
.'/../fixtures/wallabag-v2.jsonx');
160 $res = $wallabagV1Import->import();
162 $this->assertFalse($res);
164 $records = $this->logHandler
->getRecords();
165 $this->assertContains('WallabagImport: unable to read file', $records[0]['message']);
166 $this->assertEquals('ERROR', $records[0]['level_name']);
169 public function testImportUserNotDefined()
171 $wallabagV1Import = $this->getWallabagV2Import(true);
172 $wallabagV1Import->setFilepath(__DIR__
.'/../fixtures/wallabag-v2.json');
174 $res = $wallabagV1Import->import();
176 $this->assertFalse($res);
178 $records = $this->logHandler
->getRecords();
179 $this->assertContains('WallabagImport: user is not defined', $records[0]['message']);
180 $this->assertEquals('ERROR', $records[0]['level_name']);
183 public function testImportEmptyFile()
185 $wallabagV2Import = $this->getWallabagV2Import();
186 $wallabagV2Import->setFilepath(__DIR__
.'/../fixtures/wallabag-v2-empty.json');
188 $res = $wallabagV2Import->import();
190 $this->assertFalse($res);
191 $this->assertEquals(['skipped' => 0, 'imported' => 0], $wallabagV2Import->getSummary());
194 public function testImportWithExceptionFromGraby()
196 $wallabagV2Import = $this->getWallabagV2Import();
197 $wallabagV2Import->setFilepath(__DIR__
.'/../fixtures/wallabag-v2.json');
199 $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
200 ->disableOriginalConstructor()
203 $entryRepo->expects($this->exactly(24))
204 ->method('findByUrlAndUserId')
205 ->will($this->onConsecutiveCalls(false, true, false));
208 ->expects($this->any())
209 ->method('getRepository')
210 ->willReturn($entryRepo);
213 ->expects($this->exactly(2))
214 ->method('updateEntry')
215 ->will($this->throwException(new \
Exception()));
217 $res = $wallabagV2Import->import();
219 $this->assertTrue($res);
220 $this->assertEquals(['skipped' => 24, 'imported' => 0], $wallabagV2Import->getSummary());