/**
* @param bool $isWebsiteScope
* @param array $websites
* @param int $quoteId
* @dataProvider dispatchDataProvider
*/
public function testDispatch($isWebsiteScope, $websites, $quoteId)
{
$this->configMock->expects($this->once())->method('isWebsiteScope')->will($this->returnValue($isWebsiteScope));
$customerDataObjectMock = $this->getMockBuilder('Magento\\Customer\\Api\\Data\\CustomerInterface')->disableOriginalConstructor()->getMock();
$customerDataObjectMock->expects($this->any())->method('getGroupId')->will($this->returnValue(1));
$customerDataObjectMock->expects($this->any())->method('getWebsiteId')->will($this->returnValue(2));
if ($isWebsiteScope) {
$websites = $websites[0];
$this->storeManagerMock->expects($this->once())->method('getWebsite')->with(2)->will($this->returnValue($websites));
} else {
$this->storeManagerMock->expects($this->once())->method('getWebsites')->will($this->returnValue($websites));
}
$origCustomerDataObjectMock = $this->getMockBuilder('Magento\\Customer\\Api\\Data\\CustomerInterface')->disableOriginalConstructor()->getMock();
$origCustomerDataObjectMock->expects($this->any())->method('getGroupId')->will($this->returnValue(2));
$this->eventMock->expects($this->any())->method('getCustomerDataObject')->will($this->returnValue($customerDataObjectMock));
$this->eventMock->expects($this->any())->method('getOrigCustomerDataObject')->will($this->returnValue($origCustomerDataObjectMock));
/** @var \PHPUnit_Framework_MockObject_MockObject|\Magento\Quote\Model\Quote $quoteMock */
$quoteMock = $this->getMockBuilder('Magento\\Quote\\Model\\Quote')->setMethods(['setWebsite', 'setCustomerGroupId', 'collectTotals', '__wakeup'])->disableOriginalConstructor()->getMock();
$websiteCount = count($websites);
if ($quoteId) {
$this->quoteRepositoryMock->expects($this->exactly($websiteCount))->method('getForCustomer')->will($this->returnValue($quoteMock));
$quoteMock->expects($this->exactly($websiteCount))->method('setWebsite');
$quoteMock->expects($this->exactly($websiteCount))->method('setCustomerGroupId');
$quoteMock->expects($this->exactly($websiteCount))->method('collectTotals');
$this->quoteRepositoryMock->expects($this->exactly($websiteCount))->method('save')->with($quoteMock);
} else {
$this->quoteRepositoryMock->expects($this->exactly($websiteCount))->method('getForCustomer')->willThrowException(new \Magento\Framework\Exception\NoSuchEntityException());
$quoteMock->expects($this->never())->method('setCustomerGroupId');
$quoteMock->expects($this->never())->method('collectTotals');
$this->quoteRepositoryMock->expects($this->never())->method('save');
}
$this->customerQuote->dispatch($this->observerMock);
}
/**
* @param bool $isWebsiteScope
* @param array $websites
* @param int $quoteId
* @dataProvider dispatchDataProvider
*/
public function testDispatch($isWebsiteScope, $websites, $quoteId)
{
$this->configMock->expects($this->once())->method('isWebsiteScope')->will($this->returnValue($isWebsiteScope));
$customerDataObjectMock = $this->getMockBuilder('Magento\\Customer\\Service\\V1\\Data\\Customer')->disableOriginalConstructor()->getMock();
$customerDataObjectMock->expects($this->any())->method('getGroupId')->will($this->returnValue(1));
$customerDataObjectMock->expects($this->any())->method('getWebsiteId')->will($this->returnValue(2));
if ($isWebsiteScope) {
$websites = $websites[0];
$this->storeManagerMock->expects($this->once())->method('getWebsite')->with(2)->will($this->returnValue($websites));
} else {
$this->storeManagerMock->expects($this->once())->method('getWebsites')->will($this->returnValue($websites));
}
$origCustomerDataObjectMock = $this->getMockBuilder('Magento\\Customer\\Service\\V1\\Data\\Customer')->disableOriginalConstructor()->getMock();
$origCustomerDataObjectMock->expects($this->any())->method('getGroupId')->will($this->returnValue(2));
$this->eventMock->expects($this->any())->method('getCustomerDataObject')->will($this->returnValue($customerDataObjectMock));
$this->eventMock->expects($this->any())->method('getOrigCustomerDataObject')->will($this->returnValue($origCustomerDataObjectMock));
/** @var \PHPUnit_Framework_MockObject_MockObject|\Magento\Sales\Model\Quote $quoteMock */
$quoteMock = $this->getMockBuilder('Magento\\Sales\\Model\\Quote')->setMethods(array('setWebsite', 'loadByCustomer', 'getId', 'setCustomerGroupId', 'collectTotals', 'save', '__wakeup'))->disableOriginalConstructor()->getMock();
$websiteCount = count($websites);
$this->quoteFactoryMock->expects($this->exactly($websiteCount))->method('create')->will($this->returnValue($quoteMock));
$quoteMock->expects($this->exactly($websiteCount))->method('setWebsite');
$quoteMock->expects($this->exactly($websiteCount))->method('loadByCustomer');
$quoteMock->expects($this->exactly($websiteCount))->method('getId')->will($this->returnValue($quoteId));
if ($quoteId) {
$quoteMock->expects($this->exactly($websiteCount))->method('setCustomerGroupId');
$quoteMock->expects($this->exactly($websiteCount))->method('collectTotals');
$quoteMock->expects($this->exactly($websiteCount))->method('save');
} else {
$quoteMock->expects($this->never())->method('setCustomerGroupId');
$quoteMock->expects($this->never())->method('collectTotals');
$quoteMock->expects($this->never())->method('save');
}
$this->customerQuote->dispatch($this->observerMock);
}
