Your IP : 216.73.216.97
<?php
/**
* Copyright © Magento, Inc. All rights reserved.
* See COPYING.txt for license details.
*/
declare(strict_types=1);
namespace Magento\Weee\Test\Unit\Model;
use Magento\Catalog\Model\Product;
use Magento\Customer\Api\AccountManagementInterface;
use Magento\Customer\Model\Session;
use Magento\Eav\Model\Entity\Attribute;
use Magento\Eav\Model\Entity\AttributeFactory;
use Magento\Framework\Data\Collection\AbstractDb;
use Magento\Framework\DataObject;
use Magento\Framework\Model\Context;
use Magento\Framework\Pricing\PriceCurrencyInterface;
use Magento\Framework\Registry;
use Magento\Framework\TestFramework\Unit\Helper\ObjectManager;
use Magento\Store\Model\Group;
use Magento\Store\Model\Store;
use Magento\Store\Model\StoreManagerInterface;
use Magento\Store\Model\Website;
use Magento\Tax\Helper\Data;
use Magento\Tax\Model\Calculation;
use Magento\Tax\Model\CalculationFactory;
use Magento\Weee\Model\Config;
use Magento\Weee\Model\Tax;
use PHPUnit\Framework\MockObject\MockObject;
use PHPUnit\Framework\TestCase;
/**
* @SuppressWarnings(PHPMD.CouplingBetweenObjects)
*/
class TaxTest extends TestCase
{
/**
* @var Tax
*/
protected $model;
/**
* @var MockObject
*/
protected $context;
/**
* @var MockObject
*/
protected $registry;
/**
* @var MockObject
*/
protected $attributeFactory;
/**
* @var MockObject
*/
protected $storeManager;
/**
* @var MockObject
*/
protected $calculationFactory;
/**
* @var MockObject
*/
protected $customerSession;
/**
* @var MockObject
*/
protected $accountManagement;
/**
* @var MockObject
*/
protected $taxData;
/**
* @var MockObject
*/
protected $resource;
/**
* @var MockObject
*/
protected $weeeConfig;
/**
* @var MockObject
*/
protected $priceCurrency;
/**
* @var MockObject
*/
protected $resourceCollection;
/**
* @var MockObject
*/
protected $data;
/**
* @var ObjectManager
*/
protected $objectManager;
/**
* Setup the test
*/
protected function setUp(): void
{
$this->objectManager = new ObjectManager($this);
$this->context = $this->createMock(Context::class);
$this->registry = $this->createMock(Registry::class);
$this->attributeFactory = $this->getMockBuilder(AttributeFactory::class)
->disableOriginalConstructor()
->onlyMethods(['create'])
->getMock();
$this->storeManager = $this->getMockForAbstractClass(StoreManagerInterface::class);
$this->calculationFactory = $this->getMockBuilder(CalculationFactory::class)
->disableOriginalConstructor()
->onlyMethods(['create'])
->getMock();
$this->customerSession = $this->getMockBuilder(Session::class)
->disableOriginalConstructor()
->onlyMethods(
[
'getCustomerId',
]
)
->addMethods(
[
'getDefaultTaxShippingAddress',
'getDefaultTaxBillingAddress',
'getCustomerTaxClassId'
]
)
->getMock();
$this->customerSession->expects($this->any())->method('getCustomerId')->willReturn(null);
$this->customerSession->expects($this->any())->method('getDefaultTaxShippingAddress')->willReturn(null);
$this->customerSession->expects($this->any())->method('getDefaultTaxBillingAddress')->willReturn(null);
$this->customerSession->expects($this->any())->method('getCustomerTaxClassId')->willReturn(null);
$className = AccountManagementInterface::class;
$this->accountManagement = $this->createMock($className);
$className = Data::class;
$this->taxData = $this->createMock($className);
$className = \Magento\Weee\Model\ResourceModel\Tax::class;
$this->resource = $this->createMock($className);
$className = Config::class;
$this->weeeConfig = $this->createMock($className);
$className = PriceCurrencyInterface::class;
$this->priceCurrency = $this->createMock($className);
$className = AbstractDb::class;
$this->resourceCollection = $this->createMock($className);
$this->model = $this->objectManager->getObject(
Tax::class,
[
'context' => $this->context,
'registry' => $this->registry,
'attributeFactory' => $this->attributeFactory,
'storeManager' => $this->storeManager,
'calculationFactory' => $this->calculationFactory,
'customerSession' => $this->customerSession,
'accountManagement' => $this->accountManagement,
'taxData' => $this->taxData,
'resource' => $this->resource,
'weeeConfig' => $this->weeeConfig,
'priceCurrency' => $this->priceCurrency,
'resourceCollection' => $this->resourceCollection,
]
);
}
/**
* @dataProvider getProductWeeeAttributesDataProvider
* @param array $weeeTaxCalculationsByEntity
* @param mixed $websitePassed
* @param string $expectedFptLabel
* @return void
*/
public function testGetProductWeeeAttributes(
array $weeeTaxCalculationsByEntity,
$websitePassed,
string $expectedFptLabel
): void {
$product = $this->createMock(Product::class);
$website = $this->createMock(Website::class);
$store = $this->createMock(Store::class);
$group = $this->createMock(Group::class);
$attribute = $this->createMock(Attribute::class);
$calculation = $this->createMock(Calculation::class);
$obj = new DataObject(['country' => 'US', 'region' => 'TX']);
$calculation->expects($this->once())
->method('getRateRequest')
->willReturn($obj);
$calculation->expects($this->once())
->method('getDefaultRateRequest')
->willReturn($obj);
$calculation->expects($this->any())
->method('getRate')
->willReturn('10');
$attribute->expects($this->once())
->method('getAttributeCodesByFrontendType')
->willReturn(['0'=>'fpt']);
$this->storeManager->expects($this->any())
->method('getWebsite')
->willReturn($website);
$website->expects($this->any())
->method('getId')
->willReturn($websitePassed);
$website->expects($this->any())
->method('getDefaultGroup')
->willReturn($group);
$group->expects($this->any())
->method('getDefaultStore')
->willReturn($store);
$store->expects($this->any())
->method('getId')
->willReturn(1);
if ($websitePassed) {
$product->expects($this->never())
->method('getStore')
->willReturn($store);
} else {
$product->expects($this->once())
->method('getStore')
->willReturn($store);
$store->expects($this->once())
->method('getWebsiteId')
->willReturn(1);
}
$product->expects($this->any())
->method('getId')
->willReturn(1);
$this->weeeConfig->expects($this->any())
->method('isEnabled')
->willReturn(true);
$this->weeeConfig->expects($this->any())
->method('isTaxable')
->willReturn(true);
$this->attributeFactory->expects($this->any())
->method('create')
->willReturn($attribute);
$this->calculationFactory->expects($this->any())
->method('create')
->willReturn($calculation);
$this->priceCurrency->expects($this->any())
->method('round')
->with(0.1)
->willReturn(0.25);
$this->resource->expects($this->any())
->method('fetchWeeeTaxCalculationsByEntity')
->willReturn([
0 => $weeeTaxCalculationsByEntity
]);
$result = $this->model->getProductWeeeAttributes($product, null, null, $websitePassed, true);
$this->assertIsArray($result);
$this->assertArrayHasKey(0, $result);
$obj = $result[0];
$this->assertEquals(1, $obj->getAmount());
$this->assertEquals(0.25, $obj->getTaxAmount());
$this->assertEquals($weeeTaxCalculationsByEntity['attribute_code'], $obj->getCode());
$this->assertEquals(__($expectedFptLabel), $obj->getName());
}
/**
* Test getWeeeAmountExclTax method
*
* @param string $productTypeId
* @param string $productPriceType
* @dataProvider getWeeeAmountExclTaxDataProvider
*/
public function testGetWeeeAmountExclTax($productTypeId, $productPriceType)
{
$product = $this->getMockBuilder(Product::class)
->disableOriginalConstructor()
->setMethods(['getTypeId', 'getPriceType'])
->getMock();
$product->expects($this->any())->method('getTypeId')->willReturn($productTypeId);
$product->expects($this->any())->method('getPriceType')->willReturn($productPriceType);
$weeeDataHelper = $this->getMockBuilder(DataObject::class)
->disableOriginalConstructor()
->setMethods(['getAmountExclTax'])
->getMock();
$weeeDataHelper->expects($this->at(0))->method('getAmountExclTax')->willReturn(10);
$weeeDataHelper->expects($this->at(1))->method('getAmountExclTax')->willReturn(30);
$tax = $this->getMockBuilder(Tax::class)
->disableOriginalConstructor()
->setMethods(['getProductWeeeAttributes'])
->getMock();
$tax->expects($this->once())->method('getProductWeeeAttributes')
->willReturn([$weeeDataHelper, $weeeDataHelper]);
$this->assertEquals(40, $tax->getWeeeAmountExclTax($product));
}
/**
* Test getWeeeAmountExclTax method for dynamic bundle product
*/
public function testGetWeeeAmountExclTaxForDynamicBundleProduct()
{
$product = $this->getMockBuilder(Product::class)
->disableOriginalConstructor()
->setMethods(['getTypeId', 'getPriceType'])
->getMock();
$product->expects($this->once())->method('getTypeId')->willReturn('bundle');
$product->expects($this->once())->method('getPriceType')->willReturn(0);
$weeeDataHelper = $this->getMockBuilder(DataObject::class)
->disableOriginalConstructor()
->getMock();
$tax = $this->getMockBuilder(Tax::class)
->disableOriginalConstructor()
->setMethods(['getProductWeeeAttributes'])
->getMock();
$tax->expects($this->once())->method('getProductWeeeAttributes')->willReturn([$weeeDataHelper]);
$this->assertEquals(0, $tax->getWeeeAmountExclTax($product));
}
/**
* @return array
*/
public function getProductWeeeAttributesDataProvider()
{
return [
'store_label_defined' => [
'weeeTaxCalculationsByEntity' => [
'weee_value' => 1,
'label_value' => 'fpt_label',
'frontend_label' => 'fpt_label_frontend',
'attribute_code' => 'fpt_code',
],
'websitePassed' => 1,
'expectedFptLabel' => 'fpt_label',
],
'store_label_not_defined' => [
'weeeTaxCalculationsByEntity' => [
'weee_value' => 1,
'label_value' => '',
'frontend_label' => 'fpt_label_frontend',
'attribute_code' => 'fpt_code',
],
'websitePassed' => 1,
'expectedFptLabel' => 'fpt_label_frontend',
],
'website_not_passed' => [
'weeeTaxCalculationsByEntity' => [
'weee_value' => 1,
'label_value' => '',
'frontend_label' => 'fpt_label_frontend',
'attribute_code' => 'fpt_code',
],
'websitePassed' => null,
'expectedFptLabel' => 'fpt_label_frontend',
],
];
}
/**
* @return array
*/
public function getWeeeAmountExclTaxDataProvider()
{
return [
[
'bundle', 1
],
[
'simple', 0
],
[
'simple', 1
]
];
}
}