tests/Wallabag/ImportBundle/Import/FirefoxImportTest.php

   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
  64         $entryRepo->expects($this->exactly(4))
  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
  78             ->expects($this->exactly(4))
  79             ->method('updateEntry')
  80             ->willReturn($entry);
  81
  82         $res = $firefoxImport->import();
  83
  84         $this->assertTrue($res);
  85         $this->assertEquals(['skipped' => 0, 'imported' => 4, 'queued' => 0], $firefoxImport->getSummary());
  86     }
  87
  88     public function testImportAndMarkAllAsRead()
  89     {
  90         $firefoxImport = $this->getFirefoxImport();
  91         $firefoxImport->setFilepath(__DIR__.'/../fixtures/readability-read.json');
  92
  93         $entryRepo = $this->getMockBuilder('Wallabag\CoreBundle\Repository\EntryRepository')
  94             ->disableOriginalConstructor()
  95             ->getMock();
  96
  97         $entryRepo->expects($this->exactly(2))
  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
 123         $this->assertEquals(['skipped' => 1, 'imported' => 1, 'queued' => 0], $firefoxImport->getSummary());
 124     }
 125
 126     public function testImportWithRabbit()
 127     {
 128         $firefoxImport = $this->getFirefoxImport();
 129         $firefoxImport->setFilepath(__DIR__.'/../fixtures/readability.json');
 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
 155             ->expects($this->exactly(4))
 156             ->method('publish');
 157
 158         $firefoxImport->setProducer($producer);
 159
 160         $res = $readabilityImport->setMarkAsRead(true)->import();
 161
 162         $this->assertTrue($res);
 163         $this->assertEquals(['skipped' => 0, 'imported' => 0, 'queued' => 4], $firefoxImport->getSummary());
 164     }
 165
 166     public function testImportWithRedis()
 167     {
 168         $firefoxImport = $this->getReadabilityImport();
 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);
 201         $this->assertEquals(['skipped' => 0, 'imported' => 0, 'queued' => 4], $firefoxImport->getSummary());
 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();
 216         $this->assertContains('FirefoxImport: unable to read file', $records[0]['message']);
 217         $this->assertEquals('ERROR', $records[0]['level_name']);
 218     }
 219
 220     public function testImportUserNotDefined()
 221     {
 222         $firefoxImport = $this->getFirefoxImport(true);
 223         $firefoxImport->setFilepath(__DIR__.'/../fixtures/readability.json');
 224
 225         $res = $firefoxImport->import();
 226
 227         $this->assertFalse($res);
 228
 229         $records = $this->logHandler->getRecords();
 230         $this->assertContains('FirefoxImport: user is not defined', $records[0]['message']);
 231         $this->assertEquals('ERROR', $records[0]['level_name']);
 232     }
 233 }