From bfa445c322e54e75cd1fbea4e13b2a14bdf4b7e7 Mon Sep 17 00:00:00 2001
From: denisdulici <denisdulici@gmail.com>
Date: Mon, 16 Dec 2019 10:39:09 +0300
Subject: [PATCH] added expense state

---
 database/factories/Transaction.php | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/database/factories/Transaction.php b/database/factories/Transaction.php
index 5e65dd21e..f2ef53cc8 100644
--- a/database/factories/Transaction.php
+++ b/database/factories/Transaction.php
@@ -5,10 +5,10 @@ use App\Models\Banking\Transaction;
 use Faker\Generator as Faker;
 use Illuminate\Http\UploadedFile;
 
-$factory->define(Transaction::class, function (Faker $faker) {
-	$user = User::first();
-	$company = $user->companies()->first();
+$user = User::first();
+$company = $user->companies()->first();
 
+$factory->define(Transaction::class, function (Faker $faker) use ($company) {
 	setting()->setExtraColumns(['company_id' => $company->id]);
 
 	$attachment = UploadedFile::fake()->create('image.jpg');
@@ -22,9 +22,18 @@ $factory->define(Transaction::class, function (Faker $faker) {
 		'currency_code' => setting('default.currency'),
 		'currency_rate' => '1',
 		'description' => $faker->text(5),
-		'category_id' => $company->categories()->type('income')->first()->id,
+		'category_id' => $company->categories()->type('income')->pluck('id')->first(),
 		'reference' => $faker->text(5),
 		'payment_method' => setting('default.payment_method'),
 		//'attachment' => $attachment,
 	];
 });
+
+$factory->state(Transaction::class, 'income', []);
+
+$factory->state(Transaction::class, 'expense', function (Faker $faker) use ($company) {
+    return [
+		'type' => 'expense',
+		'category_id' => $company->categories()->type('expense')->pluck('id')->first(),
+    ];
+});