TddWizard\Fixtures\Sales\InvoiceBuilder

Complexity

Total Complexity

6

Size/Duplication

Total Lines	80
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	build()	0	14	2
A	__construct()	0	12	1
A	forOrder()	0	13	2
A	withItem()	0	7	1

<?php

namespace TddWizard\Fixtures\Sales;

use Magento\Framework\ObjectManagerInterface;
use Magento\Sales\Api\Data\InvoiceInterface;
use Magento\Sales\Api\Data\InvoiceItemCreationInterfaceFactory;

The type Magento\Sales\Api\Data\InvoiceItemCreationInterfaceFactory was not found. Maybe you did not declare it correctly or list all dependencies?

use Magento\Sales\Api\Data\OrderInterface;
use Magento\Sales\Api\InvoiceOrderInterface;
use Magento\Sales\Api\InvoiceRepositoryInterface;
use Magento\TestFramework\Helper\Bootstrap;

The type Magento\TestFramework\Helper\Bootstrap was not found. Maybe you did not declare it correctly or list all dependencies?

/**
 * Builder to be used by fixtures
 */
class InvoiceBuilder
{
    /**
     * @var InvoiceItemCreationInterfaceFactory
     */
    private $itemFactory;

    /**
     * @var InvoiceOrderInterface
     */
    private $invoiceOrder;

    /**
     * @var InvoiceRepositoryInterface
     */
    private $invoiceRepository;

    /**
     * @var OrderInterface
     */
    private $order;

    /**
     * @var int[]
     */
    private $orderItems;

    public function __construct(
        InvoiceItemCreationInterfaceFactory $itemFactory,
        InvoiceOrderInterface $invoiceOrder,
        InvoiceRepositoryInterface $invoiceRepository,
        OrderInterface $order
    ) {
        $this->itemFactory = $itemFactory;
        $this->invoiceOrder = $invoiceOrder;
        $this->invoiceRepository = $invoiceRepository;
        $this->order = $order;

        $this->orderItems = [];
    }

    public static function forOrder(
        OrderInterface $order,
        ObjectManagerInterface $objectManager = null
    ): InvoiceBuilder {
        if ($objectManager === null) {
            $objectManager = Bootstrap::getObjectManager();
        }

        return new static(
            $objectManager->create(InvoiceItemCreationInterfaceFactory::class),
            $objectManager->create(InvoiceOrderInterface::class),
            $objectManager->create(InvoiceRepositoryInterface::class),
            $order
        );
    }

    public function withItem(int $orderItemId, int $qty): InvoiceBuilder
    {
        $builder = clone $this;

        $builder->orderItems[$orderItemId] = $qty;

        return $builder;
    }

    public function build(): InvoiceInterface
    {
        $invoiceItems = [];

        foreach ($this->orderItems as $orderItemId => $qty) {
            $invoiceItem = $this->itemFactory->create();
            $invoiceItem->setOrderItemId($orderItemId);
            $invoiceItem->setQty($qty);
            $invoiceItems[] = $invoiceItem;
        }

        $invoiceId = $this->invoiceOrder->execute($this->order->getEntityId(), false, $invoiceItems);

        return $this->invoiceRepository->get($invoiceId);
    }
}