diff options
Diffstat (limited to 'src/Wallabag/ImportBundle/Tests/Import/WallabagV1ImportTest.php')
-rw-r--r-- | src/Wallabag/ImportBundle/Tests/Import/WallabagV1ImportTest.php | 10 |
1 files changed, 8 insertions, 2 deletions
diff --git a/src/Wallabag/ImportBundle/Tests/Import/WallabagV1ImportTest.php b/src/Wallabag/ImportBundle/Tests/Import/WallabagV1ImportTest.php index 496cf2d3..3ef852e5 100644 --- a/src/Wallabag/ImportBundle/Tests/Import/WallabagV1ImportTest.php +++ b/src/Wallabag/ImportBundle/Tests/Import/WallabagV1ImportTest.php | |||
@@ -2,8 +2,9 @@ | |||
2 | 2 | ||
3 | namespace Wallabag\ImportBundle\Tests\Import; | 3 | namespace Wallabag\ImportBundle\Tests\Import; |
4 | 4 | ||
5 | use Wallabag\UserBundle\Entity\User; | ||
6 | use Wallabag\ImportBundle\Import\WallabagV1Import; | 5 | use Wallabag\ImportBundle\Import\WallabagV1Import; |
6 | use Wallabag\UserBundle\Entity\User; | ||
7 | use Wallabag\CoreBundle\Entity\Entry; | ||
7 | use Monolog\Logger; | 8 | use Monolog\Logger; |
8 | use Monolog\Handler\TestHandler; | 9 | use Monolog\Handler\TestHandler; |
9 | 10 | ||
@@ -71,7 +72,7 @@ class WallabagV1ImportTest extends \PHPUnit_Framework_TestCase | |||
71 | ->getMock(); | 72 | ->getMock(); |
72 | 73 | ||
73 | $this->contentProxy | 74 | $this->contentProxy |
74 | ->expects($this->once()) | 75 | ->expects($this->exactly(3)) |
75 | ->method('updateEntry') | 76 | ->method('updateEntry') |
76 | ->willReturn($entry); | 77 | ->willReturn($entry); |
77 | 78 | ||
@@ -99,6 +100,11 @@ class WallabagV1ImportTest extends \PHPUnit_Framework_TestCase | |||
99 | ->method('getRepository') | 100 | ->method('getRepository') |
100 | ->willReturn($entryRepo); | 101 | ->willReturn($entryRepo); |
101 | 102 | ||
103 | $this->contentProxy | ||
104 | ->expects($this->exactly(3)) | ||
105 | ->method('updateEntry') | ||
106 | ->willReturn(new Entry($this->user)); | ||
107 | |||
102 | // check that every entry persisted are archived | 108 | // check that every entry persisted are archived |
103 | $this->em | 109 | $this->em |
104 | ->expects($this->any()) | 110 | ->expects($this->any()) |