CreditmemoFactoryTest.php

Go to the documentation of this file.

<?php
namespace Magento\Sales\Model\Order;

use Magento\Framework\Api\SearchCriteriaBuilder;
use Magento\Framework\ObjectManagerInterface;
use Magento\Sales\Api\InvoiceRepositoryInterface;
use Magento\Sales\Model\Order;
use Magento\TestFramework\Helper\Bootstrap;

class CreditmemoFactoryTest extends \PHPUnit\Framework\TestCase
{
    const ORDER_ITEM_ID_PLACEHOLDER = 'id_item_';

    private $objectManager;

    private $creditmemoFactory;

    protected function setUp()
    {
        $this->objectManager = Bootstrap::getObjectManager();
        $this->creditmemoFactory = $this->objectManager->create(CreditmemoFactory::class);
    }

    public function testCreateByOrder(array $creditmemoData, $expectedQty)
    {
        $order = $this->getOrder('100000001');
        $creditmemoData = $this->prepareCreditMemoData($order, $creditmemoData);
        $creditmemo = $this->creditmemoFactory->createByOrder($order, $creditmemoData);
        self::assertEquals($expectedQty, $creditmemo->getTotalQty(), 'Creditmemo has wrong total qty.');
    }

    public function createByOrderDataProvider(): array
    {
        return [
            [
                'creditmemoData' => [
                    'qtys' => [
                        self::ORDER_ITEM_ID_PLACEHOLDER . '2' => 2,
                        self::ORDER_ITEM_ID_PLACEHOLDER . '3' => 10,
                    ],
                ],
                'expectedQty' => 12,
            ],
            [
                'creditmemoData' => [
                    'qtys' => [
                        self::ORDER_ITEM_ID_PLACEHOLDER . '1' => 2,
                        self::ORDER_ITEM_ID_PLACEHOLDER . '2' => 2,
                        self::ORDER_ITEM_ID_PLACEHOLDER . '3' => 10,
                    ],
                ],
                'expectedQty' => 14,
            ],
        ];
    }

    public function testCreateByInvoice()
    {
        $invoice = $this->getInvoice('100000001');
        $creditmemo = $this->creditmemoFactory->createByInvoice($invoice);
        self::assertEquals(14, $creditmemo->getTotalQty(), 'Creditmemo has wrong total qty.');
    }

    private function prepareCreditMemoData(Order $order, array $creditmemoData): array
    {
        $result = [];
        $orderItems = $order->getAllItems();
        foreach ($creditmemoData['qtys'] as $key => $item) {
            $result[$orderItems[$this->prepareOrderItemKey($key)]->getId()] = $item;
        }
        $creditmemoData['qtys'] = $result;

        return $creditmemoData;
    }

    private function prepareOrderItemKey($key)
    {
        return str_replace(self::ORDER_ITEM_ID_PLACEHOLDER, '', $key) - 1;
    }

    private function getOrder(string $incrementId): Order
    {
        $order = $this->objectManager->create(Order::class);
        $order->loadByIncrementId($incrementId);

        return $order;
    }

    private function getInvoice(string $incrementId): Invoice
    {
        $searchCriteriaBuilder = $this->objectManager->get(SearchCriteriaBuilder::class);
        $searchCriteria = $searchCriteriaBuilder
            ->addFilter('increment_id', $incrementId)
            ->create();
        $repository = $this->objectManager->get(InvoiceRepositoryInterface::class);
        $items = $repository->getList($searchCriteria)->getItems();

        return array_pop($items);
    }
}