RenameWebsiteToStockLinkTest.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\InventorySales\Test\Integration\Website;

use Magento\InventorySalesApi\Model\GetAssignedStockIdForWebsiteInterface;
use Magento\Store\Model\Website;
use Magento\Store\Model\ResourceModel\Website as WebsiteResource;
use Magento\Store\Model\WebsiteFactory;
use Magento\TestFramework\Helper\Bootstrap;
use PHPUnit\Framework\TestCase;

class RenameWebsiteToStockLinkTest extends TestCase
{
    private $websiteFactory;

    private $getAssignedStockIdForWebsite;

    private $websiteResource;

    protected function setUp()
    {
        $this->websiteFactory = Bootstrap::getObjectManager()->get(WebsiteFactory::class);
        $this->getAssignedStockIdForWebsite = Bootstrap::getObjectManager()->get(
            GetAssignedStockIdForWebsiteInterface::class
        );
        $this->websiteResource = Bootstrap::getObjectManager()->get(WebsiteResource::class);
    }

    public function testRenameWebsiteToStockLink()
    {
        $oldWebsiteCode = 'old_website_code';
        $newWebsiteCode = 'new_website_code';

        $website = $this->websiteFactory->create();
        $website->setCode($oldWebsiteCode);
        $this->websiteResource->save($website);
        $websiteId = $website->getId();

        $website = $this->websiteFactory->create();
        $this->websiteResource->load($website, $websiteId);
        $website->setCode($newWebsiteCode);
        $this->websiteResource->save($website);

        self::assertNull(
            $this->getAssignedStockIdForWebsite->execute($oldWebsiteCode),
            'Old website link was not removed'
        );

        self::assertNotNull(
            $this->getAssignedStockIdForWebsite->execute($newWebsiteCode),
            'Website link was not renamed'
        );
    }
}