CreateInvoiceStep.php

Go to the documentation of this file.

<?php
namespace Magento\Sales\Test\TestStep;

use Magento\Checkout\Test\Fixture\Cart;
use Magento\Mtf\TestStep\TestStepInterface;
use Magento\Sales\Test\Fixture\OrderInjectable;
use Magento\Sales\Test\Page\Adminhtml\OrderIndex;
use Magento\Sales\Test\Page\Adminhtml\OrderInvoiceNew;
use Magento\Sales\Test\Page\Adminhtml\OrderInvoiceView;
use Magento\Sales\Test\Page\Adminhtml\SalesOrderView;
use Magento\Sales\Test\TestStep\Utils\CompareQtyTrait;
use Magento\Shipping\Test\Page\Adminhtml\OrderShipmentView;

class CreateInvoiceStep implements TestStepInterface
{
    use CompareQtyTrait;

    private $orderIndex;

    private $salesOrderView;

    private $orderInvoiceNew;

    private $orderInvoiceView;

    private $orderShipmentView;

    private $order;

    private $paymentAction;

    private $cart;

    public function __construct(
        Cart $cart,
        OrderIndex $orderIndex,
        SalesOrderView $salesOrderView,
        OrderInvoiceNew $orderInvoiceNew,
        OrderInvoiceView $orderInvoiceView,
        OrderInjectable $order,
        OrderShipmentView $orderShipmentView,
        $paymentAction = 'authorize'
    ) {
        $this->cart = $cart;
        $this->orderIndex = $orderIndex;
        $this->salesOrderView = $salesOrderView;
        $this->orderInvoiceNew = $orderInvoiceNew;
        $this->orderInvoiceView = $orderInvoiceView;
        $this->order = $order;
        $this->orderShipmentView = $orderShipmentView;
        $this->paymentAction = $paymentAction;
    }

    public function run()
    {
        if ($this->paymentAction == 'sale') {
            return null;
        }
        $this->orderIndex->open();
        $this->orderIndex->getSalesOrderGrid()->searchAndOpen(['id' => $this->order->getId()]);
        $invoicesData = $this->order->getInvoice() !== null ? $this->order->getInvoice() : ['invoiceData' => []];
        foreach ($invoicesData as $invoiceData) {
            $this->salesOrderView->getPageActions()->invoice();

            $items = $this->getItems();
            $this->orderInvoiceNew->getFormBlock()->fillProductData($invoiceData, $items);
            if ($this->compare($this->cart->getItems(), $invoiceData)) {
                $this->orderInvoiceNew->getFormBlock()->updateQty();
            }

            $this->orderInvoiceNew->getFormBlock()->fillFormData($invoiceData);
            $this->orderInvoiceNew->getFormBlock()->submit();
            $shipmentIds = $this->getShipmentIds();
        }
        $invoiceIds = $this->getInvoiceIds();

        return [
            'ids' => [
                'invoiceIds' => $invoiceIds,
                'shipmentIds' => isset($shipmentIds) ? $shipmentIds : null,
            ]
        ];
    }

    protected function getInvoiceIds()
    {
        $this->salesOrderView->getOrderForm()->openTab('invoices');
        return $this->salesOrderView->getOrderForm()->getTab('invoices')->getGridBlock()->getIds();
    }

    protected function getShipmentIds()
    {
        $this->salesOrderView->getOrderForm()->openTab('shipments');
        return $this->salesOrderView->getOrderForm()->getTab('shipments')->getGridBlock()->getIds();
    }

    protected function getItems()
    {
        return $this->cart->getItems();
    }
}