VirtualProductViewTest.php

Go to the documentation of this file.

<?php
declare(strict_types=1);

namespace Magento\GraphQl\Catalog;

use Magento\Catalog\Api\Data\ProductInterface;
use Magento\Catalog\Api\ProductRepositoryInterface;
use Magento\Framework\EntityManager\MetadataPool;
use Magento\TestFramework\ObjectManager;
use Magento\TestFramework\TestCase\GraphQlAbstract;

class VirtualProductViewTest extends GraphQlAbstract
{
    public function testQueryAllFieldsVirtualProduct()
    {
        $productSku = 'virtual-product';

        $query
            = <<<QUERY
{
   products(filter: {sku: {eq: "$productSku"}})
   {
       items{
           id
           attribute_set_id    
           created_at
           name
           sku
           type_id
           updated_at
           ... on PhysicalProductInterface {
             weight
           }  
           ... on VirtualProduct {
            attribute_set_id
            name
            id
            sku
          
           }
       }
   }
}
QUERY;

        $response = $this->graphQlQuery($query);

        $productRepository = ObjectManager::getInstance()->get(ProductRepositoryInterface::class);
        $product = $productRepository->get($productSku, false, null, true);
        $metadataPool = ObjectManager::getInstance()->get(MetadataPool::class);
        $product->setId(
            $product->getData($metadataPool->getMetadata(ProductInterface::class)->getLinkField())
        );
        $this->assertArrayHasKey('products', $response);
        $this->assertArrayHasKey('items', $response['products']);
        $this->assertEquals(1, count($response['products']['items']));
        $this->assertArrayHasKey(0, $response['products']['items']);
        $this->assertBaseFields($product, $response['products']['items'][0]);
        $this->assertArrayNotHasKey(
            'weight',
            $response['products']['items'][0],
            "response does contain the key weight"
        );
    }

    public function testCannotQueryWeightOnVirtualProductException()
    {
        $productSku = 'virtual-product';

        $query
            = <<<QUERY
{
   products(filter: {sku: {eq: "$productSku"}})
   {
       items{
           id
           attribute_set_id    
           created_at
           name
           sku
           type_id
           updated_at
           ... on PhysicalProductInterface {
             weight
           }  
           ... on VirtualProduct {
            attribute_set_id
            name
            weight
            id
            sku          
           }
       }
   }   
}
QUERY;

        $this->expectException(\Exception::class);
        $this->expectExceptionMessage(
            'GraphQL response contains errors: Cannot query field "weight" on type "VirtualProduct"'
        );
        $this->graphQlQuery($query);
    }

    private function assertBaseFields($product, $actualResponse)
    {
        $assertionMap = [
            ['response_field' => 'attribute_set_id', 'expected_value' => $product->getAttributeSetId()],
            ['response_field' => 'id', 'expected_value' => $product->getId()],
            ['response_field' => 'name', 'expected_value' => $product->getName()],
            ['response_field' => 'sku', 'expected_value' => $product->getSku()],
            ['response_field' => 'type_id', 'expected_value' => $product->getTypeId()]
        ];

        $this->assertResponseFields($actualResponse, $assertionMap);
    }
}