TddWizard\Fixtures\Sales\ShipmentFixturePool

Complexity

Total Complexity

5

Size/Duplication

Total Lines	32
Duplicated Lines	0 %

Test Coverage

Coverage

100%

Importance

Changes	1
Bugs	0	Features	0

2 Methods

Rating	Name	Duplication	Size	Complexity
A	get()	0	9	3
A	add()	0	6	2

<?php
declare(strict_types=1);

namespace TddWizard\Fixtures\Sales;

use Magento\Sales\Api\Data\ShipmentInterface;

class ShipmentFixturePool
{

    /**
     * @var ShipmentFixture[]
     */
    private $shipmentFixtures = [];

    public function add(ShipmentInterface $shipment, string $key = null): void
    {
        if ($key === null) {
            $this->shipmentFixtures[] = new ShipmentFixture($shipment);
        } else {
            $this->shipmentFixtures[$key] = new ShipmentFixture($shipment);
        }
    }

    /**
     * Returns shipment fixture by key, or last added if key not specified
     *
     * @param string|null $key
     * @return ShipmentFixture
     */
    public function get(string $key = null): ShipmentFixture
    {
        if ($key === null) {
            $key = \array_key_last($this->shipmentFixtures);
        }
        if (!array_key_exists($key, $this->shipmentFixtures)) {
            throw new \OutOfBoundsException('No matching shipment found in fixture pool');
        }
        return $this->shipmentFixtures[$key];
    }

}