From 0fd7dabbc15b230b7af42d05f5f847957c91566b Mon Sep 17 00:00:00 2001
From: James Cole <james@firefly-iii.org>
Date: Mon, 24 Aug 2020 07:38:33 +0200
Subject: [PATCH] Restore missing functions.

---
 tests/Traits/CollectsValues.php | 47 ++++++++++++++++++++++++++++++++-
 1 file changed, 46 insertions(+), 1 deletion(-)

diff --git a/tests/Traits/CollectsValues.php b/tests/Traits/CollectsValues.php
index f65ba9366e..71507e3f71 100644
--- a/tests/Traits/CollectsValues.php
+++ b/tests/Traits/CollectsValues.php
@@ -28,6 +28,10 @@ use DB;
 use FireflyIII\Exceptions\FireflyException;
 use FireflyIII\Models\Account;
 use FireflyIII\Models\AccountType;
+use FireflyIII\Models\Bill;
+use FireflyIII\Models\Budget;
+use FireflyIII\Models\Category;
+use FireflyIII\Models\Tag;
 use FireflyIII\Models\TransactionCurrency;
 use FireflyIII\Models\TransactionJournal;
 use FireflyIII\Models\TransactionType;
@@ -54,6 +58,39 @@ trait CollectsValues
         return User::where('email', 'no_admin@firefly')->first();
     }
 
+    /**
+     * @return Budget
+     */
+    public function getRandomBudget(): Budget
+    {
+        return $this->user()->budgets()->inRandomOrder()->first();
+    }
+
+    /**
+     * @return Category
+     */
+    public function getRandomCategory(): Category
+    {
+        return $this->user()->categories()->inRandomOrder()->first();
+    }
+
+    /**
+     * @return Bill
+     */
+    public function getRandomBill(): Bill
+    {
+        return $this->user()->bills()->inRandomOrder()->first();
+    }
+
+
+    /**
+     * @return Tag
+     */
+    public function getRandomTag(): Tag
+    {
+        return $this->user()->tags()->inRandomOrder()->first();
+    }
+
     /**
      * @return TransactionJournal
      */
@@ -62,6 +99,14 @@ trait CollectsValues
         return $this->getRandomJournal(TransactionType::WITHDRAWAL);
     }
 
+    /**
+     * @return TransactionJournal
+     */
+    public function getRandomDeposit(): TransactionJournal
+    {
+        return $this->getRandomJournal(TransactionType::DEPOSIT);
+    }
+
     /**
      * @param string $type
      * @return TransactionJournal
@@ -90,7 +135,7 @@ trait CollectsValues
             throw new FireflyException(sprintf('Cannot find suitable %s to use.', $type));
         }
 
-        return TransactionJournal::find((int)$result->transaction_journal_id);
+        return TransactionJournal::find((int) $result->transaction_journal_id);
 
     }