ExtensionAttributesGenerationTest.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\Framework\Api\ExtensionAttribute;

use Magento\Catalog\Api\Data\ProductInterface;
use Magento\Catalog\Api\Data\ProductExtensionInterface;
use Magento\Customer\Api\Data\CustomerInterface;
use Magento\Customer\Api\Data\CustomerExtensionInterface;

class ExtensionAttributesGenerationTest extends \PHPUnit\Framework\TestCase
{
    public function testAttributeObjectGenerationForExtensibleModel()
    {
        $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();
        $product = $objectManager->get(ProductInterface::class);

        $extensionAttributes = $product->getExtensionAttributes();
        $this->assertInstanceOf(ProductExtensionInterface::class, $extensionAttributes);

        $stockItemExtensionAttribute = $extensionAttributes->getStockItem();
        $this->assertNull($stockItemExtensionAttribute);
    }

    public function testAttributeObjectGenerationForExtensibleObject()
    {
        $objectManager = \Magento\TestFramework\Helper\Bootstrap::getObjectManager();
        $customer = $objectManager->get(CustomerInterface::class);

        $extensionAttributes = $customer->getExtensionAttributes();
        $this->assertInstanceOf(CustomerExtensionInterface::class, $extensionAttributes);
    }
}