public function testBasketElementsVatAmounts()
{
$basket = new BasketTest();
$pool = $this->getMockBuilder('Sonata\\Component\\Product\\Pool')->disableOriginalConstructor()->getMock();
$pool->expects($this->any())->method('getProvider')->will($this->returnValue($this->getMock('Sonata\\Component\\Product\\ProductProviderInterface')));
$basket->setProductPool($pool);
$element1 = $this->getMockBuilder('Sonata\\BasketBundle\\Entity\\BaseBasketElement')->getMock();
$element1->expects($this->any())->method('getProduct')->will($this->returnValue($this->getMockBuilder('Sonata\\ProductBundle\\Entity\\BaseProduct')->getMock()));
$element1->expects($this->any())->method('getVatRate')->will($this->returnValue(20));
$element1->expects($this->any())->method('getVatAmount')->will($this->returnValue(3));
$element2 = $this->getMockBuilder('Sonata\\BasketBundle\\Entity\\BaseBasketElement')->getMock();
$element2->expects($this->any())->method('getProduct')->will($this->returnValue($this->getMockBuilder('Sonata\\ProductBundle\\Entity\\BaseProduct')->getMock()));
$element2->expects($this->any())->method('getVatRate')->will($this->returnValue(10));
$element2->expects($this->any())->method('getVatAmount')->will($this->returnValue(2));
$element3 = $this->getMockBuilder('Sonata\\BasketBundle\\Entity\\BaseBasketElement')->getMock();
$element3->expects($this->any())->method('getProduct')->will($this->returnValue($this->getMockBuilder('Sonata\\ProductBundle\\Entity\\BaseProduct')->getMock()));
$element3->expects($this->any())->method('getVatRate')->will($this->returnValue(10));
$element3->expects($this->any())->method('getVatAmount')->will($this->returnValue(5));
$basket->setBasketElements(array($element1, $element2, $element3));
$items = $basket->getVatAmounts();
$this->assertTrue(is_array($items), 'Should return an array');
foreach ($items as $item) {
$this->assertArrayHasKey('rate', $item, 'Array items should contains a "rate" key');
$this->assertArrayHasKey('amount', $item, 'Array items should contains a "amount" key');
$this->assertTrue(in_array($item['rate'], array(10, 20)));
$this->assertTrue(in_array($item['amount'], array(7, 3)));
}
}