ShipmentRemoveTrackTest.php

Go to the documentation of this file.

<?php
namespace Magento\Sales\Service\V1;

use Magento\Framework\Api\SearchCriteriaBuilder;
use Magento\Framework\Webapi\Rest\Request;
use Magento\Sales\Api\Data\ShipmentTrackInterface;
use Magento\Sales\Api\ShipmentTrackRepositoryInterface;
use Magento\Sales\Model\Order\Shipment\Track;
use Magento\Sales\Model\Order\Shipment\TrackFactory;
use Magento\Sales\Model\ResourceModel\Order\Shipment\Collection;
use Magento\TestFramework\Helper\Bootstrap;
use Magento\TestFramework\TestCase\WebapiAbstract;

class ShipmentRemoveTrackTest extends WebapiAbstract
{
    const SERVICE_READ_NAME = 'salesShipmentTrackRepositoryV1';

    const SERVICE_VERSION = 'V1';

    const SHIPMENT_INCREMENT_ID = '100000001';

    private $objectManager;

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

    public function testShipmentRemoveTrack()
    {
        $shipmentCollection = $this->objectManager->get(Collection::class);
        $shipment = $shipmentCollection->getFirstItem();

        $trackEntity = $this->objectManager->get(TrackFactory::class)
            ->create(
                [
                    'data' => [
                        ShipmentTrackInterface::ENTITY_ID => null,
                        ShipmentTrackInterface::ORDER_ID => $shipment->getOrderId(),
                        ShipmentTrackInterface::PARENT_ID => $shipment->getId(),
                        ShipmentTrackInterface::WEIGHT => 20,
                        ShipmentTrackInterface::QTY => 5,
                        ShipmentTrackInterface::TRACK_NUMBER => 2,
                        ShipmentTrackInterface::DESCRIPTION => 'Shipment description',
                        ShipmentTrackInterface::TITLE => 'Shipment title',
                        ShipmentTrackInterface::CARRIER_CODE => Track::CUSTOM_CARRIER_CODE,
                    ]
                ]
            );

        $trackEntity = $this->objectManager->get(ShipmentTrackRepositoryInterface::class)
            ->save($trackEntity);

        $serviceInfo = [
            'rest' => [
                'resourcePath' => '/V1/shipment/track/' . $trackEntity->getEntityId(),
                'httpMethod' => Request::HTTP_METHOD_DELETE,
            ],
            'soap' => [
                'service' => self::SERVICE_READ_NAME,
                'serviceVersion' => self::SERVICE_VERSION,
                'operation' => self::SERVICE_READ_NAME . 'deleteById',
            ],
        ];

        $result = $this->_webApiCall($serviceInfo, ['id' => $trackEntity->getEntityId()]);

        self::assertTrue($result);
        $this->assertNoAvailableTrackItems($shipment->getId());
    }

    private function assertNoAvailableTrackItems($shipmentId)
    {
        $searchCriteriaBuilder = $this->objectManager->get(SearchCriteriaBuilder::class);
        $searchCriteria = $searchCriteriaBuilder->addFilter(ShipmentTrackInterface::PARENT_ID, $shipmentId)
            ->create();

        $items = $this->objectManager->get(ShipmentTrackRepositoryInterface::class)
            ->getList($searchCriteria)
            ->getItems();

        self::assertEmpty($items);
    }
}