Add partner relation to payment model

app/Entities/Payments/Payment.php

@@ -2,6 +2,8 @@
 
 namespace App\Entities\Payments;
 
+use App\Entities\Partners\Customer;
+use App\Entities\Partners\Vendor;
 use App\Entities\Payments\PaymentPresenter;
 use App\Entities\Projects\Project;
 use App\Entities\Users\User;
@@ -26,6 +28,15 @@ class Payment extends Model
         return $this->belongsTo(User::class, 'customer_id');
     }
 
+    public function partner()
+    {
+        if ($this->in_out == 1) {
+            return $this->belongsTo(Customer::class, 'customer_id');
+        }
+
+        return $this->belongsTo(Vendor::class, 'customer_id');
+    }
+
     public function type()
     {
         return paymentTypes($this->type_id);

database/factories/ModelFactory.php

@@ -3,7 +3,6 @@
 use App\Entities\Invoices\Invoice;
 use App\Entities\Partners\Customer;
 use App\Entities\Partners\Vendor;
-use App\Entities\Payments\Payment;
 use App\Entities\Projects\Feature;
 use App\Entities\Projects\Project;
 use App\Entities\Projects\Task;
@@ -45,26 +44,6 @@ $factory->define(Project::class, function (Faker\Generator $faker) {
     ];
 });
 
-$factory->define(Payment::class, function (Faker\Generator $faker) {
-
-    return [
-        'project_id' => function () {
-            return factory(Project::class)->create()->id;
-        },
-        'amount' => rand(1, 5) * 500000,
-        'in_out' => rand(0, 1),
-        'type_id' => rand(1, 3),
-        'date' => $faker->dateTimeBetween('-1 year', '-1 month')->format('Y-m-d'),
-        'description' => $faker->paragraph,
-        'owner_id' => function () {
-            return factory(User::class)->create()->id;
-        },
-        'customer_id' => function () {
-            return factory(User::class)->create()->id;
-        },
-    ];
-});
-
 $factory->define(Subscription::class, function (Faker\Generator $faker) {
 
     $startDate = Carbon::parse($faker->dateTimeBetween('-1 year', '-1 month')->format('Y-m-d'));

database/factories/PaymentFactory.php

@@ -0,0 +1,46 @@
+<?php
+
+use App\Entities\Partners\Customer;
+use App\Entities\Partners\Vendor;
+use App\Entities\Payments\Payment;
+use App\Entities\Projects\Project;
+use App\Entities\Users\User;
+use Faker\Generator as Faker;
+
+$factory->define(Payment::class, function (Faker $faker) {
+
+    return [
+        'project_id'  => function () {
+            return factory(Project::class)->create()->id;
+        },
+        'amount'      => 10000,
+        'in_out'      => 1,
+        'type_id'     => rand(1, 3),
+        'date'        => $faker->dateTimeBetween('-1 year', '-1 month')->format('Y-m-d'),
+        'description' => $faker->paragraph,
+        'owner_id'    => function () {
+            return factory(User::class)->create()->id;
+        },
+        'customer_id' => function () {
+            return factory(Customer::class)->create()->id;
+        },
+    ];
+});
+
+$factory->state(Payment::class, 'income', function (Faker $faker) {
+    return [
+        'in_out'      => 1,
+        'customer_id' => function () {
+            return factory(Customer::class)->create()->id;
+        },
+    ];
+});
+
+$factory->state(Payment::class, 'expanse', function (Faker $faker) {
+    return [
+        'in_out'      => 2,
+        'customer_id' => function () {
+            return factory(Vendor::class)->create()->id;
+        },
+    ];
+});

tests/Unit/Models/PaymentTest.php

@@ -2,16 +2,20 @@
 
 namespace Tests\Unit\Models;
 
+use App\Entities\Partners\Customer;
+use App\Entities\Partners\Vendor;
 use App\Entities\Payments\Payment;
-use App\Entities\Users\User;
 use Tests\TestCase;
 
 class PaymentTest extends TestCase
 {
     /** @test */
-    public function it_has_customer_relation()
+    public function it_has_partner_relation()
     {
-        $payment = factory(Payment::class)->create();
-        $this->assertTrue($payment->customer instanceof User);
+        $payment = factory(Payment::class)->states('income')->create(['in_out' => 1]);
+        $this->assertTrue($payment->partner instanceof Customer);
+
+        $payment = factory(Payment::class)->states('expanse')->create(['in_out' => 0]);
+        $this->assertTrue($payment->partner instanceof Vendor);
     }
 }