added team permissions (#996)
This commit is contained in:
144
tests/DataFixtures/TeamFixtures.php
Normal file
144
tests/DataFixtures/TeamFixtures.php
Normal file
@@ -0,0 +1,144 @@
|
||||
<?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\Tests\DataFixtures;
|
||||
|
||||
use App\Entity\Customer;
|
||||
use App\Entity\Team;
|
||||
use App\Entity\User;
|
||||
use Doctrine\Bundle\FixturesBundle\Fixture;
|
||||
use Doctrine\Common\Persistence\ObjectManager;
|
||||
use Faker\Factory;
|
||||
|
||||
/**
|
||||
* Defines the sample data to load in during controller tests.
|
||||
*/
|
||||
class TeamFixtures extends Fixture
|
||||
{
|
||||
/**
|
||||
* @var int
|
||||
*/
|
||||
protected $amount = 0;
|
||||
/**
|
||||
* @var bool
|
||||
*/
|
||||
protected $addCustomer = true;
|
||||
/**
|
||||
* @var User[]
|
||||
*/
|
||||
protected $skipUser = [];
|
||||
/**
|
||||
* @var bool
|
||||
*/
|
||||
protected $addUser = true;
|
||||
|
||||
public function setAddCustomer(bool $useCustomer)
|
||||
{
|
||||
$this->addCustomer = $useCustomer;
|
||||
}
|
||||
|
||||
public function setAddUser(bool $useUser)
|
||||
{
|
||||
$this->addUser = $useUser;
|
||||
}
|
||||
|
||||
public function getAmount(): int
|
||||
{
|
||||
return $this->amount;
|
||||
}
|
||||
|
||||
public function setAmount(int $amount): TeamFixtures
|
||||
{
|
||||
$this->amount = $amount;
|
||||
|
||||
return $this;
|
||||
}
|
||||
|
||||
public function addUserToIgnore(User $user)
|
||||
{
|
||||
$this->skipUser[] = $user;
|
||||
}
|
||||
|
||||
/**
|
||||
* {@inheritdoc}
|
||||
*/
|
||||
public function load(ObjectManager $manager)
|
||||
{
|
||||
$faker = Factory::create();
|
||||
$user = $this->getAllUsers($manager);
|
||||
$customer = $this->getAllCustomers($manager);
|
||||
|
||||
for ($i = 0; $i < $this->amount; $i++) {
|
||||
$lead = null;
|
||||
while (null === $lead) {
|
||||
$tmp = $user[array_rand($user)];
|
||||
if (!in_array($tmp, $this->skipUser)) {
|
||||
$lead = $tmp;
|
||||
}
|
||||
}
|
||||
|
||||
$entity = new Team();
|
||||
$entity
|
||||
->setName($faker->name)
|
||||
->setTeamLead($lead)
|
||||
;
|
||||
|
||||
if ($this->addUser) {
|
||||
$userToAdd = null;
|
||||
while (null === $userToAdd) {
|
||||
$tmp = $user[array_rand($user)];
|
||||
if (!in_array($tmp, $this->skipUser)) {
|
||||
$userToAdd = $tmp;
|
||||
}
|
||||
}
|
||||
$entity->addUser($userToAdd);
|
||||
}
|
||||
|
||||
if ($this->addCustomer) {
|
||||
$entity->addCustomer($customer[array_rand($customer)]);
|
||||
}
|
||||
|
||||
$manager->persist($entity);
|
||||
}
|
||||
|
||||
$manager->flush();
|
||||
}
|
||||
|
||||
/**
|
||||
* @param ObjectManager $manager
|
||||
* @return Customer[]
|
||||
*/
|
||||
protected function getAllCustomers(ObjectManager $manager)
|
||||
{
|
||||
$all = [];
|
||||
/* @var Customer[] $entries */
|
||||
$entries = $manager->getRepository(Customer::class)->findAll();
|
||||
foreach ($entries as $temp) {
|
||||
$all[$temp->getId()] = $temp;
|
||||
}
|
||||
|
||||
return $all;
|
||||
}
|
||||
|
||||
/**
|
||||
* @param ObjectManager $manager
|
||||
* @return User[]
|
||||
*/
|
||||
protected function getAllUsers(ObjectManager $manager)
|
||||
{
|
||||
$all = [];
|
||||
/* @var User[] $entries */
|
||||
$entries = $manager->getRepository(User::class)->findAll();
|
||||
foreach ($entries as $temp) {
|
||||
$all[$temp->getId()] = $temp;
|
||||
}
|
||||
|
||||
return $all;
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user