public function testDeleteById()
{
$addressId = 12;
$customerId = 43;
$this->address->expects($this->once())->method('getCustomerId')->willReturn($customerId);
$addressCollection = $this->getMock('Magento\\Customer\\Model\\Resource\\Address\\Collection', [], [], '', false);
$this->addressRegistry->expects($this->once())->method('retrieve')->with($addressId)->willReturn($this->address);
$this->customerRegistry->expects($this->once())->method('retrieve')->with($customerId)->willReturn($this->customer);
$this->customer->expects($this->once())->method('getAddressesCollection')->willReturn($addressCollection);
$addressCollection->expects($this->once())->method('clear');
$this->addressResourceModel->expects($this->once())->method('delete')->with($this->address);
$this->addressRegistry->expects($this->once())->method('remove')->with($addressId);
$this->assertTrue($this->repository->deleteById($addressId));
}
/**
* @SuppressWarnings(PHPMD.ExcessiveMethodLength)
*/
public function testSaveWithPasswordHash()
{
$customerId = 1;
$storeId = 2;
$passwordHash = 'ukfa4sdfa56s5df02asdf4rt';
$this->prepareMocksForValidation(true);
$region = $this->getMockForAbstractClass('Magento\\Customer\\Api\\Data\\RegionInterface', [], '', false);
$address = $this->getMockForAbstractClass('Magento\\Customer\\Api\\Data\\AddressInterface', [], '', false, false, true, ['setCustomerId', 'setRegion', 'getRegion', 'getId']);
$address2 = $this->getMockForAbstractClass('Magento\\Customer\\Api\\Data\\AddressInterface', [], '', false, false, true, ['setCustomerId', 'setRegion', 'getRegion', 'getId']);
$customerModel = $this->getMock('Magento\\Customer\\Model\\Customer', ['getId', 'setId', 'setStoreId', 'getStoreId', 'getAttributeSetId', 'setAttributeSetId', 'setRpToken', 'setRpTokenCreatedAt', 'getDataModel', 'setPasswordHash'], [], '', false);
$customerAttributesMetaData = $this->getMockForAbstractClass('Magento\\Framework\\Api\\CustomAttributesDataInterface', [], '', false, false, true, ['getId', 'getEmail', 'getWebsiteId', 'getAddresses', 'setAddresses']);
$this->customer->expects($this->atLeastOnce())->method('getId')->willReturn($customerId);
$this->customerRegistry->expects($this->atLeastOnce())->method('retrieve')->with($customerId)->willReturn($customerModel);
$customerModel->expects($this->atLeastOnce())->method('getDataModel')->willReturn($this->customer);
$this->imageProcessor->expects($this->once())->method('save')->with($this->customer, CustomerMetadataInterface::ENTITY_TYPE_CUSTOMER, $this->customer)->willReturn($customerAttributesMetaData);
$address->expects($this->once())->method('setCustomerId')->with($customerId)->willReturnSelf();
$address->expects($this->once())->method('getRegion')->willReturn($region);
$address->expects($this->atLeastOnce())->method('getId')->willReturn(7);
$address->expects($this->once())->method('setRegion')->with($region);
$customerAttributesMetaData->expects($this->any())->method('getAddresses')->willReturn([$address]);
$customerAttributesMetaData->expects($this->at(1))->method('setAddresses')->with([]);
$customerAttributesMetaData->expects($this->at(2))->method('setAddresses')->with([$address]);
$this->extensibleDataObjectConverter->expects($this->once())->method('toNestedArray')->with($customerAttributesMetaData, [], '\\Magento\\Customer\\Api\\Data\\CustomerInterface')->willReturn(['customerData']);
$this->customerFactory->expects($this->once())->method('create')->with(['data' => ['customerData']])->willReturn($customerModel);
$customerModel->expects($this->once())->method('getStoreId')->willReturn(null);
$store = $this->getMock('Magento\\Store\\Model\\Store', [], [], '', false);
$store->expects($this->once())->method('getId')->willReturn($storeId);
$this->storeManager->expects($this->once())->method('getStore')->willReturn($store);
$customerModel->expects($this->once())->method('setStoreId')->with($storeId);
$customerModel->expects($this->once())->method('setId')->with(null);
$customerModel->expects($this->once())->method('getAttributeSetId')->willReturn(null);
$customerModel->expects($this->once())->method('setAttributeSetId')->with(\Magento\Customer\Api\CustomerMetadataInterface::ATTRIBUTE_SET_ID_CUSTOMER);
$customerModel->expects($this->once())->method('setPasswordHash')->with($passwordHash);
$customerModel->expects($this->atLeastOnce())->method('getId')->willReturn($customerId);
$this->customerResourceModel->expects($this->once())->method('save')->with($customerModel);
$this->customerRegistry->expects($this->once())->method('push')->with($customerModel);
$this->customer->expects($this->any())->method('getAddresses')->willReturn([$address, $address2]);
$this->addressRepository->expects($this->once())->method('save')->with($address);
$customerAttributesMetaData->expects($this->once())->method('getEmail')->willReturn('*****@*****.**');
$customerAttributesMetaData->expects($this->once())->method('getWebsiteId')->willReturn(2);
$this->customerRegistry->expects($this->once())->method('retrieveByEmail')->with('*****@*****.**', 2)->willReturn($customerModel);
$this->eventManager->expects($this->once())->method('dispatch')->with('customer_save_after_data_object', ['customer_data_object' => $this->customer, 'orig_customer_data_object' => $customerAttributesMetaData]);
$this->model->save($this->customer, $passwordHash);
}
/**
* {@inheritdoc}
* @SuppressWarnings(PHPMD.CyclomaticComplexity)
*/
public function save(\Magento\Customer\Api\Data\CustomerInterface $customer, $passwordHash = null)
{
$this->validate($customer);
$origAddresses = $customer->getAddresses();
$customer->setAddresses([]);
$customerData = $this->extensibleDataObjectConverter->toNestedArray($customer, [], '\\Magento\\Customer\\Api\\Data\\CustomerInterface');
$customer->setAddresses($origAddresses);
$customerModel = $this->customerFactory->create(['data' => $customerData]);
$storeId = $customerModel->getStoreId();
if ($storeId === null) {
$customerModel->setStoreId($this->storeManager->getStore()->getId());
}
$customerModel->setId($customer->getId());
// Need to use attribute set or future updates can cause data loss
if (!$customerModel->getAttributeSetId()) {
$customerModel->setAttributeSetId(\Magento\Customer\Api\CustomerMetadataInterface::ATTRIBUTE_SET_ID_CUSTOMER);
}
// Populate model with secure data
if ($customer->getId()) {
/*
* TODO: Check \Magento\Customer\Model\Resource\Customer::changeResetPasswordLinkToken setAttribute
* and make sure its consistent
*/
$customerSecure = $this->customerRegistry->retrieveSecureData($customer->getId());
$customerModel->setRpToken($customerSecure->getRpToken());
$customerModel->setRpTokenCreatedAt($customerSecure->getRpTokenCreatedAt());
$customerModel->setPasswordHash($customerSecure->getPasswordHash());
} else {
if ($passwordHash) {
$customerModel->setPasswordHash($passwordHash);
}
}
$this->customerResourceModel->save($customerModel);
$this->customerRegistry->push($customerModel);
$customerId = $customerModel->getId();
if ($customer->getAddresses() !== null) {
if ($customer->getId()) {
$existingAddresses = $this->getById($customer->getId())->getAddresses();
$getIdFunc = function ($address) {
return $address->getId();
};
$existingAddressIds = array_map($getIdFunc, $existingAddresses);
} else {
$existingAddressIds = [];
}
$savedAddressIds = [];
foreach ($customer->getAddresses() as $address) {
$address->setCustomerId($customerId)->setRegion($address->getRegion());
$this->addressRepository->save($address);
if ($address->getId()) {
$savedAddressIds[] = $address->getId();
}
}
$addressIdsToDelete = array_diff($existingAddressIds, $savedAddressIds);
foreach ($addressIdsToDelete as $addressId) {
$this->addressRepository->deleteById($addressId);
}
}
$savedCustomer = $this->get($customer->getEmail(), $customer->getWebsiteId());
$this->eventManager->dispatch('customer_save_after_data_object', ['customer_data_object' => $savedCustomer, 'orig_customer_data_object' => $customer]);
return $savedCustomer;
}
