Improved data fixtures with Faker (#234)

2018-07-24 19:24:58 +02:00
parent 20a2d94996
commit 5ccc1c991b
12 changed files with 362 additions and 442 deletions
--- a/src/DataFixtures/CustomerFixtures.php
+++ b/src/DataFixtures/CustomerFixtures.php
@@ -0,0 +1,132 @@
+<?php
+
+/*
+ * This file is part of the Kimai time-tracking app.
+ *
+ * For the full copyright and license information, please view the LICENSE
+ * file that was distributed with this source code.
+ */
+
+namespace App\DataFixtures;
+
+use App\Entity\Activity;
+use App\Entity\Customer;
+use App\Entity\Project;
+use Doctrine\Bundle\FixturesBundle\Fixture;
+use Doctrine\Common\Persistence\ObjectManager;
+use Faker\Factory;
+use Faker\Generator;
+
+/**
+ * Defines the sample data to load in the database when running the unit and
+ * functional tests or while development.
+ *
+ * Execute this command to load the data:
+ * bin/console doctrine:fixtures:load
+ */
+class CustomerFixtures extends Fixture
+{
+    public const MIN_CUSTOMERS = 5;
+    public const MAX_CUSTOMERS = 15;
+    public const MIN_BUDGET = 0;
+    public const MAX_BUDGET = 100000;
+    public const MIN_PROJECTS_PER_CUSTOMER = 10;
+    public const MAX_PROJECTS_PER_CUSTOMER = 50;
+    public const MIN_ACTIVITIES_PER_PROJECT = 0;
+    public const MAX_ACTIVITIES_PER_PROJECT = 15;
+
+    /**
+     * @param ObjectManager $manager
+     */
+    public function load(ObjectManager $manager)
+    {
+        $faker = Factory::create();
+
+        $amountCustomers = rand(self::MIN_CUSTOMERS, self::MAX_CUSTOMERS);
+        for ($c = 1; $c <= $amountCustomers; $c++) {
+            $visibleCustomer = 0 != $c % 3;
+            $customer = $this->createCustomer($faker, $visibleCustomer);
+
+            $projectForCustomer = rand(self::MIN_PROJECTS_PER_CUSTOMER, self::MAX_PROJECTS_PER_CUSTOMER);
+            for ($p = 1; $p <= $projectForCustomer; $p++) {
+                $visibleProject = 0 != $p % 3;
+                $project = $this->createProject($faker, $customer, $visibleProject);
+                $manager->persist($project);
+
+                $activityForProject = rand(self::MIN_ACTIVITIES_PER_PROJECT, self::MAX_ACTIVITIES_PER_PROJECT);
+                for ($a = 1; $a <= $activityForProject; $a++) {
+                    $visibleActivity = 0 != $a % 3;
+                    $activity = $this->createActivity($faker, $project, $visibleActivity);
+                    $manager->persist($activity);
+                }
+            }
+
+            $manager->persist($customer);
+
+            $manager->flush();
+            $manager->clear();
+        }
+    }
+
+    /**
+     * @param Generator $faker
+     * @param boolean $visible
+     * @return Customer
+     */
+    private function createCustomer(Generator $faker, $visible)
+    {
+        $entry = new Customer();
+        $entry
+            ->setCurrency($faker->currencyCode)
+            ->setName($faker->company . ($visible ? '' : ' (x)'))
+            ->setAddress($faker->address)
+            ->setComment($faker->text)
+            ->setNumber('C-' . $faker->ean8)
+            ->setCountry($faker->countryCode)
+            ->setTimezone($faker->timezone)
+            ->setVisible($visible)
+        ;
+
+        return $entry;
+    }
+
+    /**
+     * @param Generator $faker
+     * @param Customer $customer
+     * @param boolean $visible
+     * @return Project
+     */
+    private function createProject(Generator $faker, Customer $customer, $visible)
+    {
+        $entry = new Project();
+
+        $entry
+            ->setName($faker->catchPhrase . ($visible ? '' : ' (x)'))
+            ->setBudget(rand(self::MIN_BUDGET, self::MAX_BUDGET))
+            ->setComment($faker->text)
+            ->setCustomer($customer)
+            ->setVisible($visible)
+        ;
+
+        return $entry;
+    }
+
+    /**
+     * @param Generator $faker
+     * @param Project $project
+     * @param boolean $visible
+     * @return Activity
+     */
+    private function createActivity(Generator $faker, Project $project, $visible)
+    {
+        $entry = new Activity();
+        $entry
+            ->setName($faker->bs . ($visible ? '' : ' (x)'))
+            ->setProject($project)
+            ->setComment($faker->text)
+            ->setVisible($visible)
+        ;
+
+        return $entry;
+    }
+}