ORM\Testing\EntityManagerMock

Complexity

Total Complexity

4

Size/Duplication

Total Lines	22
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

1 Method

Rating	Name	Duplication	Size	Complexity
A	fetch()	0	20	4

<?php

namespace ORM\Testing;

use ORM\Entity;
use ORM\EntityManager;
use ORM\Exception\NoEntity;

class EntityManagerMock extends EntityManager
{
    public function fetch($class, $primaryKey = null)
    {
        $reflection = new \ReflectionClass($class);
        if (!$reflection->isSubclassOf(Entity::class)) {
            throw new NoEntity($class . ' is not a subclass of Entity');
        }
        /** @var string|Entity $class */

        if ($primaryKey === null) {
            return new EntityFetcherMock($this, $class);
        }

        $primaryKey = $this->buildPrimaryKey($class, (array)$primaryKey);
        $checksum = $this->buildChecksum($primaryKey);

        if (isset($this->map[$class][$checksum])) {
            return $this->map[$class][$checksum];
        }

        return EntityFetcherMock::retrieve($class, $primaryKey);

It seems like $class can also be of type ORM\Entity; however, parameter $class of ORM\Testing\EntityFetcherMock::retrieve() does only seem to accept string, maybe add an additional type check?

    }
}