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());
}