From e36717259beffc603eb22c6c2151c95a9f3d6d30 Mon Sep 17 00:00:00 2001
From: James Cole <thegrumpydictator@gmail.com>
Date: Sun, 14 Jun 2015 11:51:47 +0200
Subject: [PATCH] Put tags in test data [skip ci]

---
 database/seeds/TestDataSeeder.php | 49 +++++++++++++++++++++++++++++--
 1 file changed, 47 insertions(+), 2 deletions(-)

diff --git a/database/seeds/TestDataSeeder.php b/database/seeds/TestDataSeeder.php
index c729c8c2da..d4b3c2f022 100644
--- a/database/seeds/TestDataSeeder.php
+++ b/database/seeds/TestDataSeeder.php
@@ -9,6 +9,7 @@ use FireflyIII\Models\BudgetLimit;
 use FireflyIII\Models\Category;
 use FireflyIII\Models\PiggyBank;
 use FireflyIII\Models\PiggyBankEvent;
+use FireflyIII\Models\Tag;
 use FireflyIII\Models\Transaction;
 use FireflyIII\Models\TransactionCurrency;
 use FireflyIII\Models\TransactionGroup;
@@ -91,6 +92,7 @@ class TestDataSeeder extends Seeder
         $this->createBills();
         $this->createExpenseAccounts();
         $this->createRevenueAccounts();
+        $this->createTags();
 
         $current = clone $this->_yearAgoStartOfMonth;
         while ($current <= $this->_startOfMonth) {
@@ -434,6 +436,20 @@ class TestDataSeeder extends Seeder
 
     }
 
+    /**
+     *
+     */
+    public function createTags()
+    {
+        $user = User::whereEmail('thegrumpydictator@gmail.com')->first();
+
+        Tag::create(
+            ['tag' => 'TagOne', 'tagMode' => 'nothing', 'user_id' => $user->id]);
+
+        Tag::create(['tag' => 'TagTwo', 'tagMode' => 'nothing', 'user_id' => $user->id]);
+        Tag::create(['tag' => 'TagThree', 'tagMode' => 'nothing', 'user_id' => $user->id]);
+    }
+
     /**
      * @param Carbon $date
      */
@@ -448,6 +464,8 @@ class TestDataSeeder extends Seeder
         $phone      = $this->findAccount('Phone agency');
         $employer   = $this->findAccount('Employer');
         $bills      = $this->findBudget('Bills');
+        $tagOne     = $this->findTag('TagOne');
+        $tagTwo     = $this->findTag('TagTwo');
         $house      = $this->findCategory('House');
         $withdrawal = TransactionType::whereType('Withdrawal')->first();
         $deposit    = TransactionType::whereType('Deposit')->first();
@@ -469,16 +487,22 @@ class TestDataSeeder extends Seeder
             ['from' => $checking, 'to' => $television, 'amount' => 50, 'transactionType' => $withdrawal, 'description' => 'TV for ' . $formatted,
              'date' => $cur, 'transactionCurrency' => $euro, 'budget' => $bills, 'category' => $house,]
         );
-        $this->createJournal(
+        $jrnl = $this->createJournal(
             ['from' => $checking, 'to' => $phone, 'amount' => 50, 'transactionType' => $withdrawal, 'description' => 'Phone bill for ' . $formatted,
              'date' => $cur, 'transactionCurrency' => $euro, 'budget' => $bills, 'category' => $house,]
         );
 
+        $jrnl->tags()->save($tagOne);
+
+
         // two transactions. One without a budget, one without a category.
-        $this->createJournal(
+        $jrnl = $this->createJournal(
             ['from'        => $checking, 'to' => $phone, 'amount' => 10, 'transactionType' => $withdrawal,
              'description' => 'Extra charges on phone bill for ' . $formatted, 'date' => $cur, 'transactionCurrency' => $euro, 'category' => $house]
         );
+
+        $jrnl->tags()->save($tagTwo);
+
         $this->createJournal(
             ['from'        => $checking, 'to' => $television, 'amount' => 5, 'transactionType' => $withdrawal,
              'description' => 'Extra charges on TV bill for ' . $formatted, 'date' => $cur, 'transactionCurrency' => $euro, 'budget' => $bills]
@@ -516,6 +540,27 @@ class TestDataSeeder extends Seeder
         return null;
     }
 
+    /**
+     * @param $tag
+     *
+     * @return Tag|null
+     */
+    protected function findTag($tagName)
+    {
+        // account
+        $user = User::whereEmail('thegrumpydictator@gmail.com')->first();
+        /** @var Tag $tag */
+        foreach (Tag::get() as $tag) {
+            if ($tag->tag == $tagName && $user->id == $tag->user_id) {
+                return $tag;
+                break;
+            }
+        }
+
+        return null;
+    }
+
+
     /**
      * @param $name
      *