kenmirrors/firefly-iii
1
0
Fork 0
mirror of https://github.com/firefly-iii/firefly-iii.git synced 2025-10-12 15:35:15 +00:00
Code Issues Projects Releases Wiki Activity

This should fix the tests.

Browse Source
This commit is contained in:
James Cole
2016-02-04 11:07:34 +01:00
parent f0bb462f1c
commit 93d9c44585
3 changed files with 55 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

47
app/Support/Migration/TestData.php
View File

@@ -14,6 +14,7 @@ use FireflyIII\Models\Account;
use FireflyIII\Models\AccountMeta;
use FireflyIII\Models\Bill;
use FireflyIII\Models\Budget;
use FireflyIII\Models\BudgetLimit;
use FireflyIII\Models\PiggyBank;
use FireflyIII\Models\PiggyBankEvent;
use FireflyIII\Models\Rule;
@@ -31,6 +32,7 @@ use Log;
*/
class TestData
{
/**
* @param User $user
*/
@@ -103,6 +105,31 @@ class TestData
}
/**
* @param User $user
* @param Carbon $current
* @param $name
* @param $amount
*/
public static function createBudgetLimit(User $user, Carbon $current, $name, $amount)
{
$start = clone $current;
$end = clone $current;
$budget = self::findBudget($user, $name);
$start->startOfMonth();
$end->endOfMonth();
BudgetLimit::create(
[
'budget_id' => $budget->id,
'startdate' => $start->format('Y-m-d'),
'amount' => $amount,
'repeats' => 0,
'repeat_freq' => 'monthly',
]
);
}
/**
* @param User $user
*/
@@ -110,6 +137,7 @@ class TestData
{
Budget::firstOrCreateEncrypted(['name' => 'Groceries', 'user_id' => $user->id]);
Budget::firstOrCreateEncrypted(['name' => 'Bills', 'user_id' => $user->id]);
Budget::firstOrCreateEncrypted(['name' => 'Car', 'user_id' => $user->id]);
// some empty budgets.
foreach (['A', 'B', 'C', 'D', "E"] as $letter) {
@@ -378,5 +406,24 @@ class TestData
return null;
}
/**
* @param User $user
* @param $name
*
* @return Budget|null
*/
public static function findBudget(User $user, $name)
{
/** @var Budget $budget */
foreach (Budget::get() as $budget) {
if ($budget->name == $name && $user->id == $budget->user_id) {
return $budget;
break;
}
}
return null;
}
}