Server : Apache System : Linux server2.corals.io 4.18.0-348.2.1.el8_5.x86_64 #1 SMP Mon Nov 15 09:17:08 EST 2021 x86_64 User : corals ( 1002) PHP Version : 7.4.33 Disable Function : exec,passthru,shell_exec,system Directory :  /home/corals/old/vendor/magento/module-tax/Test/Unit/Model/TaxClass/

Upload File : current_dir [ Writeable ] document_root [ Writeable ]

<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
declare(strict_types=1);

namespace Magento\Tax\Test\Unit\Model\TaxClass;

use Magento\Framework\Exception\LocalizedException;
use Magento\Framework\ObjectManagerInterface;
use Magento\Tax\Model\ClassModel;
use Magento\Tax\Model\TaxClass\Factory;
use Magento\Tax\Model\TaxClass\Type\Customer;
use Magento\Tax\Model\TaxClass\Type\Product;
use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;

class FactoryTest extends TestCase
{
    /**
     * @dataProvider createDataProvider
     *
     * @param string $classType
     * @param string $className
     * @param MockObject $classTypeMock
     */
    public function testCreate($classType, $className, $classTypeMock)
    {
        $classMock = $this->createPartialMock(
            ClassModel::class,
            ['getClassType', 'getId', '__wakeup']
        );
        $classMock->expects($this->once())->method('getClassType')->willReturn($classType);
        $classMock->expects($this->once())->method('getId')->willReturn(1);

        $objectManager = $this->getMockForAbstractClass(ObjectManagerInterface::class);
        $objectManager->expects(
            $this->once()
        )->method(
            'create'
        )->with(
            $className,
            ['data' => ['id' => 1]]
        )->willReturn(
            $classTypeMock
        );

        $taxClassFactory = new Factory($objectManager);
        $this->assertEquals($classTypeMock, $taxClassFactory->create($classMock));
    }

    /**
     * @return array
     */
    public function createDataProvider()
    {
        $customerClassMock = $this->createMock(Customer::class);
        $productClassMock = $this->createMock(Product::class);
        return [
            [
                ClassModel::TAX_CLASS_TYPE_CUSTOMER,
                Customer::class,
                $customerClassMock,
            ],
            [
                ClassModel::TAX_CLASS_TYPE_PRODUCT,
                Product::class,
                $productClassMock
            ]
        ];
    }

    public function testCreateWithWrongClassType()
    {
        $wrongClassType = 'TYPE';
        $classMock = $this->createPartialMock(
            ClassModel::class,
            ['getClassType', 'getId', '__wakeup']
        );
        $classMock->expects($this->once())->method('getClassType')->willReturn($wrongClassType);

        $objectManager = $this->getMockForAbstractClass(ObjectManagerInterface::class);

        $taxClassFactory = new Factory($objectManager);

        $this->expectException(LocalizedException::class);
        $this->expectExceptionMessage(sprintf('Invalid type of tax class "%s"', $wrongClassType));
        $taxClassFactory->create($classMock);
    }
}