public function testOrderElementsVatAmounts()
{
$order = new OrderTest();
$element1 = $this->getMockBuilder('Sonata\\OrderBundle\\Entity\\BaseOrderElement')->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\\OrderBundle\\Entity\\BaseOrderElement')->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));
$order->setOrderElements(array($element1, $element2, $element3));
$items = $order->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)));
}
}
