public function testExtract()
{
$customerData = ['firstname' => 'firstname', 'lastname' => 'firstname', 'email' => 'email.example.com'];
$this->formFactory->expects($this->once())->method('create')->with('customer', 'form-code')->willReturn($this->customerForm);
$this->customerForm->expects($this->once())->method('extractData')->with($this->request)->willReturn($customerData);
$this->customerForm->expects($this->once())->method('getAllowedAttributes')->willReturn(['group_id' => 'attribute object']);
$this->customerFactory->expects($this->once())->method('create')->willReturn($this->customerData);
$this->dataObjectHelper->expects($this->once())->method('populateWithArray')->with($this->customerData, $customerData, '\\Magento\\Customer\\Api\\Data\\CustomerInterface')->willReturn($this->customerData);
$this->storeManager->expects($this->once())->method('getStore')->willReturn($this->store);
$this->store->expects($this->exactly(2))->method('getId')->willReturn(1);
$this->customerGroupManagement->expects($this->once())->method('getDefaultGroup')->with(1)->willReturn($this->customerGroup);
$this->customerGroup->expects($this->once())->method('getId')->willReturn(1);
$this->customerData->expects($this->once())->method('setGroupId')->with(1);
$this->store->expects($this->once())->method('getWebsiteId')->willReturn(1);
$this->customerData->expects($this->once())->method('setWebsiteId')->with(1);
$this->customerData->expects($this->once())->method('setStoreId')->with(1);
$this->assertSame($this->customerData, $this->customerExtractor->extract('form-code', $this->request));
}
public function testDelete()
{
$groupId = 6;
$this->group->expects($this->once())->method('getId')->willReturn($groupId);
$this->groupRegistry->expects($this->once())->method('retrieve')->with($groupId)->willReturn($this->groupModel);
$this->groupModel->expects($this->once())->method('usesAsDefault')->willReturn(false);
$this->groupModel->expects($this->once())->method('delete');
$this->groupRegistry->expects($this->once())->method('remove')->with($groupId);
$this->assertTrue($this->model->delete($this->group));
}
public function testExecuteWithTaxClassAndException()
{
$taxClass = '3';
$groupId = 0;
$code = 'NOT LOGGED IN';
$this->request->expects($this->exactly(3))->method('getParam')->withConsecutive(['tax_class'], ['id'], ['code'])->willReturnOnConsecutiveCalls($taxClass, $groupId, null);
$this->resultRedirectFactory->expects($this->once())->method('create')->willReturn($this->resultRedirect);
$this->groupRepositoryMock->expects($this->once())->method('getById')->with($groupId)->willReturn($this->group);
$this->group->expects($this->once())->method('getCode')->willReturn($code);
$this->group->expects($this->once())->method('setCode')->with($code);
$this->group->expects($this->once())->method('setTaxClassId')->with($taxClass);
$this->groupRepositoryMock->expects($this->once())->method('save')->with($this->group);
$this->messageManager->expects($this->once())->method('addSuccess')->with(__('You saved the customer group.'));
$exception = new \Exception('Exception');
$this->resultRedirect->expects($this->at(0))->method('setPath')->with('customer/group')->willThrowException($exception);
$this->messageManager->expects($this->once())->method('addError')->with('Exception');
$this->dataObjectProcessorMock->expects($this->once())->method('buildOutputDataArray')->with($this->group, '\\Magento\\Customer\\Api\\Data\\GroupInterface')->willReturn(['code' => $code]);
$this->session->expects($this->once())->method('setCustomerGroupData')->with(['customer_group_code' => $code]);
$this->resultRedirect->expects($this->at(1))->method('setPath')->with('customer/group/edit', ['id' => $groupId]);
$this->assertSame($this->resultRedirect, $this->controller->execute());
}
/**
* Verifies that the tax class model exists and is a customer tax class type.
*
* @param int $taxClassId The id of the tax class model to check
* @param \Magento\Customer\Api\Data\GroupInterface $group The original group parameters
* @return void
* @throws InputException Thrown if the tax class model is invalid
*/
protected function _verifyTaxClassModel($taxClassId, $group)
{
try {
/* @var TaxClassInterface $taxClassData */
$taxClassData = $this->taxClassRepository->get($taxClassId);
} catch (\Magento\Framework\Exception\NoSuchEntityException $e) {
throw InputException::invalidFieldValue('taxClassId', $group->getTaxClassId());
}
if ($taxClassData->getClassType() !== TaxClassManagementInterface::TYPE_CUSTOMER) {
throw InputException::invalidFieldValue('taxClassId', $group->getTaxClassId());
}
}