Janisbiz\LightOrm\Tests\Unit\Dms\MySQL\QueryBuilder\Traits\AbstractTraitTestCase

Complexity

Total Complexity

4

Size/Duplication

Total Lines	37
Duplicated Lines	0 %

Importance

Changes

0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	assertObjectUsesTrait()	0	14	2
A	getTraits()	0	8	2

<?php declare(strict_types=1);

namespace Janisbiz\LightOrm\Tests\Unit\Dms\MySQL\QueryBuilder\Traits;

use PHPUnit\Framework\TestCase;

abstract class AbstractTraitTestCase extends TestCase
{
    /**
     * @param string $trait
     * @param string|object $object
     * @param string $message
     */
    protected function assertObjectUsesTrait(string $trait, $object, string $message = '')
    {
        $traits = [];
        $this->getTraits($object, $traits);

        parent::assertThat(
            \in_array($trait, $traits),
            static::isTrue(),
            \mb_strlen($message)
                ? $message
                : \sprintf(
                    'Failed asserting that "%s" is used in object "%s"',
                    $trait,
                    \get_class($object)

It seems like $object can also be of type string; however, parameter $object of get_class() does only seem to accept object, maybe add an additional type check?

                )
        );
    }

    /**
     * @param string|object $class
     * @param array $traits
     */
    private function getTraits($class, array &$traits)
    {
        $classTraits = \array_keys((new \ReflectionClass($class))->getTraits());

        $traits = \array_merge($traits, $classTraits);

        foreach ($classTraits as $classTrait) {
            $this->getTraits($classTrait, $traits);
        }
    }
}