def test_all_payment_search(self):
created_payments = [
PagSegPayment(status=STATUS_CREATED) for i in xrange(3)
]
ndb.put_multi(created_payments)
# reversing because search is desc
created_payments.reverse()
sent_payments = [
PagSegPayment(status=STATUS_SENT_TO_PAGSEGURO) for i in xrange(3)
]
ndb.put_multi(sent_payments)
sent_payments.reverse()
# Testing search for all
cmd = pagseguro_facade.search_all_payments(page_size=3).execute()
self.assertListEqual(sent_payments, cmd.result)
cmd2 = pagseguro_facade.search_all_payments(
page_size=3, start_cursor=cmd.cursor).execute()
self.assertListEqual(created_payments, cmd2.result)
cmd3 = pagseguro_facade.search_all_payments(
page_size=3, start_cursor=cmd2.cursor).execute()
self.assertListEqual([], cmd3.result)
# Test search based on status
cmd = pagseguro_facade.search_all_payments(STATUS_SENT_TO_PAGSEGURO,
page_size=4).execute()
self.assertListEqual(sent_payments, cmd.result)
cmd = pagseguro_facade.search_all_payments(STATUS_CREATED,
page_size=4).execute()
self.assertListEqual(created_payments, cmd.result)
def do_business(self):
super(ProcessExternalPaymentCmd, self).do_business()
dct = self.__transaction_dct
if self.result:
self.result.status = XML_STATUS_TO_MODEL_STATUS.get(
dct['transaction']['status'])
cmd = UpdatePaymentAndSaveLog(self.result)
cmd.execute()
else:
self.result = PagSegPayment(
code=dct['transaction']['code'],
status=XML_STATUS_TO_MODEL_STATUS.get(
dct['transaction']['status']),
total=dct['transaction']['grossAmount'],
net_amount=dct['transaction']['netAmount'])
payment_key = self.result.put()
def create_item_cmd(item):
return _SimpleSave(
PagSegItem(description=item['description'],
price=Decimal(item['amount']),
quantity=int(item['quantity'])))
items = dct['transaction']['items']['item']
if isinstance(items, dict):
items = [items]
items_cmd = [
CreatePaymentToItem(payment_key, create_item_cmd(item))
for item in items
]
cmd = CommandParallel(
CreatePagSegPaymentToLog(
payment_key,
_SimpleSave(PagSegLog(status=self.result.status))),
*items_cmd)
sender = dct['transaction'].get('sender')
if sender:
user = facade.get_user_by_email(sender['email'])()
if user is None:
user = facade.save_user_cmd(sender['email'],
sender['name'])()
cmd.append(SaveToPayment(user, payment_key))
cmd.execute()
def test_success(self, UrlFetchClassMock):
# Mocking pagseguro Fetch
UrlFetchClassMock.return_value = _build_mock()
# Mocking previous command
data_cmd_mock = Mock()
data_cmd_mock.access_data.email = '*****@*****.**'
data_cmd_mock.access_data.token = '4567890oiuytfgh'
reference0 = ItemReferenceMock()
reference1 = ItemReferenceMock()
items = [
PagSegItem(description='Python Course',
price=Decimal('120'),
quantity=1,
reference=reference0.key),
PagSegItem(description='Another Python Course',
price=Decimal('240'),
quantity=2,
reference=reference1.key)
]
ndb.put_multi(items)
data_cmd_mock.items = items
data_cmd_mock.client_form = ValidateClientCmd(email='*****@*****.**',
name='Jhon Doe').form
data_cmd_mock.address_form = pagseguro_facade.validate_address_cmd(
'Rua 1', '2', 'meu bairro', '12345678', 'São Paulo', 'SP',
'apto 4').form
payment = PagSegPayment()
payment.put()
data_cmd_mock.result = payment
contact_pagseguro_cmd = ContactPagseguro('https://store.com/pagseguro')
CommandSequential(data_cmd_mock, contact_pagseguro_cmd).execute()
self.assertEqual(payment, contact_pagseguro_cmd.result)
self.assertEqual(payment.status, STATUS_SENT_TO_PAGSEGURO)
self.assertEqual(
contact_pagseguro_cmd.checkout_code, _SUCCESS_PAGSEGURO_CODE,
'Should have the code extracted from xml _PAGSEGURO_DETAIL_XML')
self.assertIsNone(payment.code)
def test_get_payment_search(self):
created_payment = PagSegPayment(status=STATUS_CREATED)
created_payment.put()
comand = GetPayment(created_payment.key.id())
payment = comand()
self.assertEqual(created_payment, payment)
def __init__(self):
super(SavePagseguroDataCmd, self).__init__()
self._create_attributes()
self.result = PagSegPayment()
