]>
Commit | Line | Data |
---|---|---|
59201088 TC |
1 | <?php |
2 | ||
3 | namespace Tests\Wallabag\ImportBundle\Import; | |
4 | ||
5 | use Wallabag\ImportBundle\Import\FirefoxImport; | |
6 | use Wallabag\UserBundle\Entity\User; | |
7 | use Wallabag\CoreBundle\Entity\Entry; | |
8 | use Wallabag\ImportBundle\Redis\Producer; | |
9 | use Monolog\Logger; | |
10 | use Monolog\Handler\TestHandler; | |
11 | use Simpleue\Queue\RedisQueue; | |
12 | use M6Web\Component\RedisMock\RedisMockFactory; | |
13 | ||
14 | class FirefoxImportTest extends \PHPUnit_Framework_TestCase | |
15 | { | |
16 | protected $user; | |
17 | protected $em; | |
18 | protected $logHandler; | |
19 | protected $contentProxy; | |
20 | ||
21 | private function getFirefoxImport($unsetUser = false) | |
22 | { | |
23 | $this->user = new User(); | |
24 | ||
25 | $this->em = $this->getMockBuilder('Doctrine\ORM\EntityManager') | |
26 | ->disableOriginalConstructor() | |
27 | ->getMock(); | |
28 | ||
29 | $this->contentProxy = $this->getMockBuilder('Wallabag\CoreBundle\Helper\ContentProxy') | |
30 | ->disableOriginalConstructor() | |
31 | ->getMock(); | |
32 | ||
33 | $wallabag = new FirefoxImport($this->em, $this->contentProxy); | |
34 | ||
35 | $this->logHandler = new TestHandler(); | |
36 | $logger = new Logger('test', [$this->logHandler]); | |
37 | $wallabag->setLogger($logger); | |
38 | ||
39 | if (false === $unsetUser) { | |
40 | $wallabag->setUser($this->user); | |
41 | } | |
42 | ||
43 | return $wallabag; | |
44 | } | |
45 | ||
46 | public function testInit() | |
47 | { | |
48 | $firefoxImport = $this->getFirefoxImport(); | |
49 | ||
50 | $this->assertEquals('Firefox', $firefoxImport->getName()); | |
51 | $this->assertNotEmpty($firefoxImport->getUrl()); | |
52 | $this->assertEquals('import.firefox.description', $firefoxImport->getDescription()); | |
53 | } | |
54 | ||
55 | public function testImport() | |
56 | { | |
57 | $firefoxImport = $this->getFirefoxImport(); | |
58 | $firefoxImport->setFilepath(__DIR__.'/../fixtures/firefox-bookmarks.json'); | |
59 | ||
60 | $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository') | |
61 | ->disableOriginalConstructor() | |
62 | ->getMock(); | |
63 | ||
12d93e68 | 64 | $entryRepo->expects($this->exactly(2)) |
59201088 TC |
65 | ->method('findByUrlAndUserId') |
66 | ->willReturn(false); | |
67 | ||
68 | $this->em | |
69 | ->expects($this->any()) | |
70 | ->method('getRepository') | |
71 | ->willReturn($entryRepo); | |
72 | ||
73 | $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry') | |
74 | ->disableOriginalConstructor() | |
75 | ->getMock(); | |
76 | ||
77 | $this->contentProxy | |
12d93e68 | 78 | ->expects($this->exactly(2)) |
59201088 TC |
79 | ->method('updateEntry') |
80 | ->willReturn($entry); | |
81 | ||
82 | $res = $firefoxImport->import(); | |
83 | ||
84 | $this->assertTrue($res); | |
12d93e68 | 85 | $this->assertEquals(['skipped' => 0, 'imported' => 2, 'queued' => 0], $firefoxImport->getSummary()); |
59201088 TC |
86 | } |
87 | ||
88 | public function testImportAndMarkAllAsRead() | |
89 | { | |
90 | $firefoxImport = $this->getFirefoxImport(); | |
64b1229b | 91 | $firefoxImport->setFilepath(__DIR__.'/../fixtures/firefox-bookmarks.json'); |
59201088 TC |
92 | |
93 | $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository') | |
94 | ->disableOriginalConstructor() | |
95 | ->getMock(); | |
96 | ||
12d93e68 | 97 | $entryRepo->expects($this->exactly(2)) |
59201088 TC |
98 | ->method('findByUrlAndUserId') |
99 | ->will($this->onConsecutiveCalls(false, true)); | |
100 | ||
101 | $this->em | |
102 | ->expects($this->any()) | |
103 | ->method('getRepository') | |
104 | ->willReturn($entryRepo); | |
105 | ||
106 | $this->contentProxy | |
107 | ->expects($this->exactly(1)) | |
108 | ->method('updateEntry') | |
109 | ->willReturn(new Entry($this->user)); | |
110 | ||
111 | // check that every entry persisted are archived | |
112 | $this->em | |
113 | ->expects($this->any()) | |
114 | ->method('persist') | |
115 | ->with($this->callback(function ($persistedEntry) { | |
116 | return $persistedEntry->isArchived(); | |
117 | })); | |
118 | ||
119 | $res = $firefoxImport->setMarkAsRead(true)->import(); | |
120 | ||
121 | $this->assertTrue($res); | |
122 | ||
12d93e68 | 123 | $this->assertEquals(['skipped' => 1, 'imported' => 1, 'queued' => 0], $firefoxImport->getSummary()); |
59201088 TC |
124 | } |
125 | ||
126 | public function testImportWithRabbit() | |
127 | { | |
128 | $firefoxImport = $this->getFirefoxImport(); | |
64b1229b | 129 | $firefoxImport->setFilepath(__DIR__.'/../fixtures/firefox-bookmarks.json'); |
59201088 TC |
130 | |
131 | $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository') | |
132 | ->disableOriginalConstructor() | |
133 | ->getMock(); | |
134 | ||
135 | $entryRepo->expects($this->never()) | |
136 | ->method('findByUrlAndUserId'); | |
137 | ||
138 | $this->em | |
139 | ->expects($this->never()) | |
140 | ->method('getRepository'); | |
141 | ||
142 | $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry') | |
143 | ->disableOriginalConstructor() | |
144 | ->getMock(); | |
145 | ||
146 | $this->contentProxy | |
147 | ->expects($this->never()) | |
148 | ->method('updateEntry'); | |
149 | ||
150 | $producer = $this->getMockBuilder('OldSound\RabbitMqBundle\RabbitMq\Producer') | |
151 | ->disableOriginalConstructor() | |
152 | ->getMock(); | |
153 | ||
154 | $producer | |
64b1229b | 155 | ->expects($this->exactly(1)) |
59201088 TC |
156 | ->method('publish'); |
157 | ||
158 | $firefoxImport->setProducer($producer); | |
159 | ||
64b1229b | 160 | $res = $firefoxImport->setMarkAsRead(true)->import(); |
59201088 TC |
161 | |
162 | $this->assertTrue($res); | |
64b1229b | 163 | $this->assertEquals(['skipped' => 0, 'imported' => 0, 'queued' => 1], $firefoxImport->getSummary()); |
59201088 TC |
164 | } |
165 | ||
166 | public function testImportWithRedis() | |
167 | { | |
64b1229b | 168 | $firefoxImport = $this->getFirefoxImport(); |
59201088 TC |
169 | $firefoxImport->setFilepath(__DIR__.'/../fixtures/firefox-bookmarks.json'); |
170 | ||
171 | $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository') | |
172 | ->disableOriginalConstructor() | |
173 | ->getMock(); | |
174 | ||
175 | $entryRepo->expects($this->never()) | |
176 | ->method('findByUrlAndUserId'); | |
177 | ||
178 | $this->em | |
179 | ->expects($this->never()) | |
180 | ->method('getRepository'); | |
181 | ||
182 | $entry = $this->getMockBuilder('Wallabag\CoreBundle\Entity\Entry') | |
183 | ->disableOriginalConstructor() | |
184 | ->getMock(); | |
185 | ||
186 | $this->contentProxy | |
187 | ->expects($this->never()) | |
188 | ->method('updateEntry'); | |
189 | ||
190 | $factory = new RedisMockFactory(); | |
191 | $redisMock = $factory->getAdapter('Predis\Client', true); | |
192 | ||
193 | $queue = new RedisQueue($redisMock, 'firefox'); | |
194 | $producer = new Producer($queue); | |
195 | ||
196 | $firefoxImport->setProducer($producer); | |
197 | ||
198 | $res = $firefoxImport->setMarkAsRead(true)->import(); | |
199 | ||
200 | $this->assertTrue($res); | |
64b1229b | 201 | $this->assertEquals(['skipped' => 0, 'imported' => 0, 'queued' => 1], $firefoxImport->getSummary()); |
59201088 TC |
202 | |
203 | $this->assertNotEmpty($redisMock->lpop('firefox')); | |
204 | } | |
205 | ||
206 | public function testImportBadFile() | |
207 | { | |
208 | $firefoxImport = $this->getFirefoxImport(); | |
209 | $firefoxImport->setFilepath(__DIR__.'/../fixtures/wallabag-v1.jsonx'); | |
210 | ||
211 | $res = $firefoxImport->import(); | |
212 | ||
213 | $this->assertFalse($res); | |
214 | ||
215 | $records = $this->logHandler->getRecords(); | |
64b1229b | 216 | $this->assertContains('Wallabag Browser Import: unable to read file', $records[0]['message']); |
59201088 TC |
217 | $this->assertEquals('ERROR', $records[0]['level_name']); |
218 | } | |
219 | ||
220 | public function testImportUserNotDefined() | |
221 | { | |
222 | $firefoxImport = $this->getFirefoxImport(true); | |
64b1229b | 223 | $firefoxImport->setFilepath(__DIR__.'/../fixtures/firefox-bookmarks.json'); |
59201088 TC |
224 | |
225 | $res = $firefoxImport->import(); | |
226 | ||
227 | $this->assertFalse($res); | |
228 | ||
229 | $records = $this->logHandler->getRecords(); | |
64b1229b | 230 | $this->assertContains('Wallabag Browser Import: user is not defined', $records[0]['message']); |
59201088 TC |
231 | $this->assertEquals('ERROR', $records[0]['level_name']); |
232 | } | |
233 | } |