/**
* Collect address discount amount
*
* @param Address $address
* @return $this
*/
public function collect(Address $address)
{
parent::collect($address);
$quote = $address->getQuote();
$store = $this->_storeManager->getStore($quote->getStoreId());
$this->_calculator->reset($address);
$items = $this->_getAddressItems($address);
if (!count($items)) {
return $this;
}
$eventArgs = array('website_id' => $store->getWebsiteId(), 'customer_group_id' => $quote->getCustomerGroupId(), 'coupon_code' => $quote->getCouponCode());
$this->_calculator->init($store->getWebsiteId(), $quote->getCustomerGroupId(), $quote->getCouponCode());
$this->_calculator->initTotals($items, $address);
$address->setDiscountDescription(array());
$items = $this->_calculator->sortItemsByPriority($items);
foreach ($items as $item) {
if ($item->getNoDiscount()) {
$item->setDiscountAmount(0);
$item->setBaseDiscountAmount(0);
continue;
}
/**
* Child item discount we calculate for parent
*/
if ($item->getParentItemId()) {
continue;
}
$eventArgs['item'] = $item;
$this->_eventManager->dispatch('sales_quote_address_discount_item', $eventArgs);
if ($item->getHasChildren() && $item->isChildrenCalculated()) {
$isMatchedParent = $this->_calculator->canApplyRules($item);
$this->_calculator->setSkipActionsValidation($isMatchedParent);
foreach ($item->getChildren() as $child) {
$this->_calculator->process($child);
if ($isMatchedParent) {
$this->_recalculateChildDiscount($child);
}
$eventArgs['item'] = $child;
$this->_eventManager->dispatch('sales_quote_address_discount_item', $eventArgs);
$this->_aggregateItemDiscount($child);
}
$this->_calculator->setSkipActionsValidation(false);
} else {
$this->_calculator->process($item);
$this->_aggregateItemDiscount($item);
}
}
/**
* Process shipping amount discount
*/
$address->setShippingDiscountAmount(0);
$address->setBaseShippingDiscountAmount(0);
if ($address->getShippingAmount()) {
$this->_calculator->processShippingAmount($address);
$this->_addAmount(-$address->getShippingDiscountAmount());
$this->_addBaseAmount(-$address->getBaseShippingDiscountAmount());
}
$this->_calculator->prepareDescription($address);
return $this;
}
public function testApplyRulesThatAppliedRuleIdsAreCollected()
{
$positivePrice = 1;
$ruleId1 = 123;
$ruleId2 = 234;
$expectedRuleIds = [$ruleId1 => $ruleId1, $ruleId2 => $ruleId2];
$this->model->init($this->model->getWebsiteId(), $this->model->getCustomerGroupId(), $this->model->getCouponCode());
$this->item->setDiscountCalculationPrice($positivePrice);
$this->item->setData('calculation_price', $positivePrice);
$this->model->setSkipActionsValidation(true);
$this->rulesApplier->expects($this->once())->method('applyRules')->with($this->equalTo($this->item), $this->equalTo($this->ruleCollection), $this->anything(), $this->anything())->will($this->returnValue($expectedRuleIds));
$this->rulesApplier->expects($this->once())->method('setAppliedRuleIds')->with($this->anything(), $expectedRuleIds);
$this->model->process($this->item);
}
