public function testSetCustomerData() { /** @var \Magento\Customer\Service\V1\Data\Customer $customerDataMock */ $customerDataMock = $this->getMock('Magento\\Customer\\Service\\V1\\Data\\Customer', [], [], '', false); $this->_quoteMock->expects($this->once())->method('assignCustomer')->with($customerDataMock); $customerDataMock->expects($this->once())->method('getId'); $this->_checkoutModel->setCustomerData($customerDataMock); }