public function testAssignData()
 {
     $baId = '1678235';
     $customerId = 67;
     $referenceId = '1234124';
     $data = new DataObject([PaymentInterface::KEY_ADDITIONAL_DATA => [AbstractAgreement::TRANSPORT_BILLING_AGREEMENT_ID => $baId]]);
     $paymentInfo = $this->getMockBuilder(Payment::class)->disableOriginalConstructor()->getMock();
     $quote = $this->getMockBuilder(Quote::class)->disableOriginalConstructor()->setMethods(['__wakeup', 'getCustomerId'])->getMock();
     $this->payment->setInfoInstance($paymentInfo);
     $agreementModel = $this->getMockBuilder(Agreement::class)->disableOriginalConstructor()->setMethods(['__wakeup', 'load', 'getCustomerId', 'getId', 'getReferenceId'])->getMock();
     $this->agreementFactory->expects(static::once())->method('create')->willReturn($agreementModel);
     $paymentInfo->expects(static::once())->method('getQuote')->willReturn($quote);
     $agreementModel->expects(static::once())->method('load')->with($baId);
     $agreementModel->expects(static::once())->method('getId')->willReturn($baId);
     $agreementModel->expects(static::atLeastOnce())->method('getCustomerId')->willReturn($customerId);
     $agreementModel->expects(static::atLeastOnce())->method('getReferenceId')->willReturn($referenceId);
     $quote->expects(static::once())->method('getCustomerId')->willReturn($customerId);
     $paymentInfo->expects(static::exactly(2))->method('setAdditionalInformation')->willReturnMap([AbstractAgreement::TRANSPORT_BILLING_AGREEMENT_ID, $baId, AbstractAgreement::PAYMENT_INFO_REFERENCE_ID, $referenceId]);
     $this->payment->assignData($data);
 }
 /**
  * @param DataObject $data
  * @throws \Magento\Framework\Exception\LocalizedException
  */
 private function parentAssignDataExpectation(DataObject $data)
 {
     $eventData = [AbstractDataAssignObserver::METHOD_CODE => $this, AbstractDataAssignObserver::MODEL_CODE => $this->payment->getInfoInstance(), AbstractDataAssignObserver::DATA_CODE => $data];
     $this->eventManagerMock->expects(static::exactly(2))->method('dispatch')->willReturnMap([['payment_method_assign_data_' . AbstractAgreementStub::STUB_CODE, $eventData], ['payment_method_assign_data', $eventData]]);
 }