From b66daad3d3ee81685a5d3a117b68e18355317d5a Mon Sep 17 00:00:00 2001
From: James Cole <thegrumpydictator@gmail.com>
Date: Sat, 28 Apr 2018 05:40:23 +0200
Subject: [PATCH] Fix coverage.

---
 tests/Unit/Factory/PiggyBankEventFactoryTest.php | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/tests/Unit/Factory/PiggyBankEventFactoryTest.php b/tests/Unit/Factory/PiggyBankEventFactoryTest.php
index 44e2ec87f5..5e1fda97f2 100644
--- a/tests/Unit/Factory/PiggyBankEventFactoryTest.php
+++ b/tests/Unit/Factory/PiggyBankEventFactoryTest.php
@@ -72,21 +72,22 @@ class PiggyBankEventFactoryTest extends TestCase
     }
 
     /**
+     * Test for withdrawal where piggy has no repetition.
+     *
      * @covers \FireflyIII\Factory\PiggyBankEventFactory
      */
-    public function testCreateNoRep()
+    public function testCreateNoRep(): void
     {
         /** @var TransactionJournal $transfer */
-        $transfer   = $this->user()->transactionJournals()->where('transaction_type_id', 3)->first();
-        $piggy      = $this->user()->piggyBanks()->first();
-        $repetition = new PiggyBankRepetition;
-        $repos      = $this->mock(PiggyBankRepositoryInterface::class);
+        $transfer = $this->user()->transactionJournals()->where('transaction_type_id', 3)->first();
+        $piggy    = $this->user()->piggyBanks()->first();
+        $repos    = $this->mock(PiggyBankRepositoryInterface::class);
         /** @var PiggyBankEventFactory $factory */
         $factory = app(PiggyBankEventFactory::class);
 
         // mock:
         $repos->shouldReceive('setUser');
-        $repos->shouldReceive('getRepetition')->andReturn($repetition);
+        $repos->shouldReceive('getRepetition')->andReturn(null);
         $repos->shouldReceive('getExactAmount')->andReturn('0');
 
         $this->assertNull($factory->create($transfer, $piggy));