93 lines
1.9 KiB
PHP
93 lines
1.9 KiB
PHP
<?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\Tag;
|
|
use Doctrine\Persistence\ObjectManager;
|
|
|
|
/**
|
|
* Defines the sample data to load in during controller tests.
|
|
*/
|
|
final class TagFixtures implements TestFixture
|
|
{
|
|
/**
|
|
* @var string[]
|
|
*/
|
|
private array $tagArray = [];
|
|
private ?\Closure $callback = null;
|
|
|
|
/**
|
|
* Will be called prior to persisting the object.
|
|
*/
|
|
public function setCallback(\Closure $callback): void
|
|
{
|
|
$this->callback = $callback;
|
|
}
|
|
|
|
/**
|
|
* @return string[]
|
|
*/
|
|
public function getTagArray(): array
|
|
{
|
|
return $this->tagArray;
|
|
}
|
|
|
|
public function addTagNameToCreate(string $name): void
|
|
{
|
|
$this->tagArray[] = $name;
|
|
}
|
|
|
|
/**
|
|
* @param string[] $tagArray
|
|
*/
|
|
public function setTagArray(array $tagArray): void
|
|
{
|
|
$this->tagArray = $tagArray;
|
|
}
|
|
|
|
public function importAmount(int $amount): void
|
|
{
|
|
$tags = [];
|
|
for ($i = 0; $i <= $amount; $i++) {
|
|
$tags[] = (string) $i;
|
|
}
|
|
$this->setTagArray($tags);
|
|
}
|
|
|
|
/**
|
|
* @return Tag[]
|
|
*/
|
|
public function load(ObjectManager $manager): array
|
|
{
|
|
$created = [];
|
|
|
|
foreach ($this->getTagArray() as $tagName) {
|
|
$tag = $this->createTagEntry($tagName);
|
|
|
|
if (null !== $this->callback) {
|
|
\call_user_func($this->callback, $tag);
|
|
}
|
|
$manager->persist($tag);
|
|
$created[] = $tag;
|
|
}
|
|
$manager->flush();
|
|
|
|
return $created;
|
|
}
|
|
|
|
private function createTagEntry(string $tagName): Tag
|
|
{
|
|
$tagObject = new Tag();
|
|
$tagObject->setName($tagName);
|
|
|
|
return $tagObject;
|
|
}
|
|
}
|