RenameWebsiteToStockLinkTest.php 2.1 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
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
{
    /**
     * @var WebsiteFactory
     */
    private $websiteFactory;

    /**
     * @var GetAssignedStockIdForWebsiteInterface
     */
    private $getAssignedStockIdForWebsite;

    /**
     * @var WebsiteResource
     */
    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);
    }

    /**
     * @throws \Exception
     * @magentoDbIsolation enabled
     */
    public function testRenameWebsiteToStockLink()
    {
        $oldWebsiteCode = 'old_website_code';
        $newWebsiteCode = 'new_website_code';

        /** @var Website $website */
        $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'
        );
    }
}