LinkPriceTest.php

Go to the documentation of this file.

<?php
namespace Magento\Downloadable\Test\Unit\Pricing\Price;

class LinkPriceTest extends \PHPUnit\Framework\TestCase
{
    protected $linkPrice;

    protected $amountMock;

    protected $saleableItemMock;

    protected $calculatorMock;

    protected $linkMock;

    protected $priceCurrencyMock;

    protected function setUp()
    {
        $this->saleableItemMock = $this->createMock(\Magento\Catalog\Model\Product::class);
        $this->amountMock = $this->createMock(\Magento\Framework\Pricing\Amount\Base::class);
        $this->calculatorMock = $this->createMock(\Magento\Framework\Pricing\Adjustment\Calculator::class);
        $this->linkMock = $this->createPartialMock(
            \Magento\Downloadable\Model\Link::class,
            ['getPrice', 'getProduct', '__wakeup']
        );

        $this->priceCurrencyMock = $this->createMock(\Magento\Framework\Pricing\PriceCurrencyInterface::class);

        $this->linkPrice = new \Magento\Downloadable\Pricing\Price\LinkPrice(
            $this->saleableItemMock,
            1,
            $this->calculatorMock,
            $this->priceCurrencyMock
        );
    }

    public function testGetLinkAmount()
    {
        $amount = 100;
        $convertedAmount = 50;

        $this->linkMock->expects($this->once())
            ->method('getPrice')
            ->will($this->returnValue($amount));
        $this->linkMock->expects($this->once())
            ->method('getProduct')
            ->will($this->returnValue($this->saleableItemMock));
        $this->priceCurrencyMock->expects($this->once())
            ->method('convertAndRound')
            ->with($amount)
            ->will($this->returnValue($convertedAmount));
        $this->calculatorMock->expects($this->once())
            ->method('getAmount')
            ->with($convertedAmount, $this->equalTo($this->saleableItemMock))
            ->will($this->returnValue($convertedAmount));

        $result = $this->linkPrice->getLinkAmount($this->linkMock);
        $this->assertEquals($convertedAmount, $result);
    }
}