Restore missing functions.

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);
 
     }