27 $this->customerAddressHelperMock = $this->createMock(\
Magento\Customer\Helper\Address::class);
37 $eventMock = $this->createPartialMock(\
Magento\Framework\Event::class, [
'getShippingAssignment',
'getQuote']);
38 $observer = $this->createPartialMock(\
Magento\Framework\Event\Observer::class, [
'getEvent']);
39 $observer->expects($this->exactly(2))->method(
'getEvent')->willReturn($eventMock);
41 $shippingAssignmentMock = $this->createMock(\
Magento\Quote\Api\Data\ShippingAssignmentInterface::class);
42 $quoteMock = $this->createMock(\
Magento\Quote\Model\Quote::class);
44 $eventMock->expects($this->once())->method(
'getShippingAssignment')->willReturn($shippingAssignmentMock);
45 $eventMock->expects($this->once())->method(
'getQuote')->willReturn($quoteMock);
47 $shippingMock = $this->createMock(\
Magento\Quote\Api\Data\ShippingInterface::class);
48 $shippingAssignmentMock->expects($this->once())->method(
'getShipping')->willReturn($shippingMock);
50 $quoteAddress = $this->createPartialMock(
51 \
Magento\Quote\Model\Quote\Address::class,
53 'getPrevQuoteCustomerGroupId',
54 'unsPrevQuoteCustomerGroupId',
55 'hasPrevQuoteCustomerGroupId',
60 $shippingMock->expects($this->once())->method(
'getAddress')->willReturn($quoteAddress);
62 $this->customerAddressHelperMock->expects($this->once())
63 ->method(
'getTaxCalculationAddressType')
64 ->will($this->returnValue($configAddressType));
66 $quoteAddress->expects($this->once())->method(
'hasPrevQuoteCustomerGroupId');
67 $id = $quoteAddress->expects($this->any())->method(
'getPrevQuoteCustomerGroupId');
68 $quoteAddress->expects($this->any())->method(
'setCustomerGroupId')->with(
$id);
69 $quoteAddress->expects($this->any())->method(
'getQuote');
70 $quoteAddress->expects($this->any())->method(
'unsPrevQuoteCustomerGroupId');
81 [\Magento\Customer\Model\Address\AbstractAddress::TYPE_SHIPPING],
83 [\Magento\Customer\Model\Address\AbstractAddress::TYPE_BILLING],
testExecute($configAddressType)
restoreCustomerGroupIdDataProvider()
$customerAddressHelperMock