def test_cancel_order_without_coupon(self):
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
def cancel(self, payment_serial_number):
assert len(payment_serial_number) == 32
return 'cancel_order_success'
self.test_place_order_without_coupon()
new_payment = Payment.objects.all().order_by('-datetime')[0]
AlipayUseCases.api_cls = AlipayApisMock
alipay_usecases = AlipayUseCases()
new_coupon = alipay_usecases.cancel_order(new_payment)
# Check the payment data
payment = Payment.objects.get(serial_number=new_payment.serial_number)
assert payment.serial_number == new_payment.serial_number
assert len(payment.transactions.all()) == 0
assert payment.status == PAYMENT_STATUS.CANCELLED
assert payment.coupon is None
assert new_coupon is None
def test_payment_unfreeze_without_coupon(self):
self.test_on_payment_success_without_coupon()
new_payment = Payment.objects.all().order_by('-datetime')[0]
alipay_usecases = AlipayUseCases()
alipay_usecases.on_payment_unfreeze(new_payment)
new_payment.refresh_from_db()
assert new_payment.status == PAYMENT_STATUS.FINISHED
self.platform_account.refresh_from_db()
assert self.platform_account.alipay_balance == 0
assert self.platform_account.alipay_withdrawable_balance == 0
self.merchant_account.refresh_from_db()
assert self.merchant_account.alipay_balance == 2000
assert self.merchant_account.alipay_withdrawable_balance == 2000
self.inviter_account.refresh_from_db()
assert self.inviter_account.alipay_balance == 0
assert self.inviter_account.alipay_withdrawable_balance == 0
self.originator_account.refresh_from_db()
assert self.originator_account.alipay_balance == 0
assert self.originator_account.alipay_withdrawable_balance == 0
def test_marketer_withdraw(self):
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
def pay_to_alipay(self, serial_number, receiver_alipay_id, amount,
desc, payee_type, payee_real_name):
assert len(serial_number) == 32
assert receiver_alipay_id == self.test_case.inviter.alipay_id
assert amount == 1000
assert desc == '付款联盟提现'
return 'refund_request_success'
self.inviter_account.alipay_balance = 10000
self.inviter_account.alipay_withdrawable_balance = 5000
self.inviter_account.save()
AlipayUseCases.api_cls = AlipayApisMock
alipay_balance_usecases = AlipayUseCases()
alipay_balance_usecases.withdraw(self.inviter_account, 1000)
withdraw = Withdraw.objects.all().order_by('-datetime')[0]
# Check withdraw
assert len(withdraw.serial_number) == 32
assert len(withdraw.transactions.all()) == 1
assert (timezone.now() - withdraw.datetime) < timedelta(seconds=30)
assert withdraw.account_id == self.inviter_account.id
assert withdraw.amount == 1000
assert withdraw.status == WITHDRAW_STATUS.FINISHED
# Check the transaction
transactions = withdraw.transactions.all().order_by('id')
assert transactions[0].object_id == withdraw.serial_number
assert transactions[
0].content_object.serial_number == withdraw.serial_number
assert transactions[
0].transaction_type == TRANSACTION_TYPE.MARKETER_WITHDRAW
assert (timezone.now() -
transactions[0].datetime) < timedelta(seconds=30)
assert transactions[0].account_id == self.inviter_account.id
assert transactions[0].amount == -1000
assert transactions[0].balance_after_transaction == 9000
# Check the balance
self.inviter_account.refresh_from_db()
assert self.inviter_account.alipay_balance == 9000
assert self.inviter_account.alipay_withdrawable_balance == 4000
def test_place_order(self):
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
def place_order(self, notify_url, order_title,
payment_serial_number, total_amount, buyer_id):
assert notify_url == 'http://test.com'
assert order_title == self.test_case.merchant.name
assert len(payment_serial_number) == 32
assert total_amount == 1900
assert buyer_id == '1234567890'
return {
'code': '10000',
'msg': 'Success',
'out_trade_no': payment_serial_number,
'trade_no': '2018110522001483870500447579'
}
AlipayUseCases.api_cls = AlipayApisMock
alipay_usecases = AlipayUseCases()
result = alipay_usecases.place_order(self.client, self.merchant,
self.coupon, 2000,
'http://test.com')
assert result['trade_no'] == '2018110522001483870500447579'
new_payment = Payment.objects.all().order_by('-datetime')[0]
assert new_payment
assert len(new_payment.serial_number) == 32
assert len(new_payment.transactions.all()) == 0
assert (timezone.now() - new_payment.datetime) < timedelta(seconds=30)
assert new_payment.pay_channel == PAY_CHANNELS.ALIPAY
assert new_payment.merchant_id == self.merchant.id
assert new_payment.status == PAYMENT_STATUS.UNPAID
assert new_payment.client_id == self.client.id
assert new_payment.order_price == 2000
assert new_payment.coupon_id == self.coupon.id
assert new_payment.platform_share == round(1900 * PLATFORM_SHARE_RATE)
assert new_payment.inviter_share == round(1900 * INVITER_SHARE_RATE)
assert new_payment.originator_share == round(1900 *
ORIGINATOR_SHARE_RATE)
assert new_payment.note is None
def test_marketer_transfer_insufficient(self):
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
def pay_to_alipay(self, serial_number, receiver_alipay_id, amount,
desc, payee_type, payee_real_name):
assert len(serial_number) == 32
assert receiver_alipay_id == self.test_case.inviter.alipay_id
assert amount == 10
assert desc == '付款联盟邀请人支付宝账号验证'
return 'refund_request_success'
AlipayUseCases.api_cls = AlipayApisMock
alipay_balance_usecases = AlipayUseCases()
self.assertRaises(BalanceInsufficient,
alipay_balance_usecases.transfer,
self.inviter.wechat_unionid, self.inviter.alipay_id,
self.inviter.name, 10)
transfer = TransferRecord.objects.all().order_by('-datetime')[0]
# Check transfer
assert len(transfer.serial_number) == 32
assert (timezone.now() - transfer.datetime) < timedelta(seconds=30)
assert transfer.account_number == self.inviter.alipay_id
assert transfer.account_name == self.inviter.name
assert transfer.amount == 10
assert transfer.status == WITHDRAW_STATUS.FAILED
def test_place_order_without_coupon(self):
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
def place_order(self, notify_url, order_title,
payment_serial_number, total_amount, buyer_id):
assert notify_url == 'http://test.com'
assert order_title == self.test_case.merchant.name
assert len(payment_serial_number) == 32
assert total_amount == 2000
assert buyer_id == '1234567890'
return 'place_order_success'
AlipayUseCases.api_cls = AlipayApisMock
alipay_usecases = AlipayUseCases()
result = alipay_usecases.place_order(self.client, self.merchant, None,
2000, 'http://test.com')
assert result == 'place_order_success'
new_payment = Payment.objects.all().order_by('-datetime')[0]
assert new_payment
assert len(new_payment.serial_number) == 32
assert len(new_payment.transactions.all()) == 0
assert (timezone.now() - new_payment.datetime) < timedelta(seconds=30)
assert new_payment.pay_channel == PAY_CHANNELS.ALIPAY
assert new_payment.merchant_id == self.merchant.id
assert new_payment.status == PAYMENT_STATUS.UNPAID
assert new_payment.client_id == self.client.id
assert new_payment.order_price == 2000
assert new_payment.coupon is None
assert new_payment.platform_share == 0
assert new_payment.inviter_share == 0
assert new_payment.originator_share == 0
assert new_payment.note is None
def test_cancel_order(self):
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
def cancel(self, payment_serial_number):
assert len(payment_serial_number) == 32
return 'cancel_order_success'
self.test_place_order()
new_payment = Payment.objects.all().order_by('-datetime')[0]
AlipayUseCases.api_cls = AlipayApisMock
alipay_usecases = AlipayUseCases()
new_coupon = alipay_usecases.cancel_order(new_payment)
# Check the payment data
payment = Payment.objects.get(serial_number=new_payment.serial_number)
assert payment.serial_number == new_payment.serial_number
assert len(payment.transactions.all()) == 0
assert payment.status == PAYMENT_STATUS.CANCELLED
assert payment.coupon.status == COUPON_STATUS.DESTROYED
assert new_coupon.rule == payment.coupon.rule
assert new_coupon.client == payment.coupon.client
assert new_coupon.discount == payment.coupon.discount
assert new_coupon.min_charge == payment.coupon.min_charge
assert new_coupon.originator_merchant == payment.coupon.originator_merchant
assert new_coupon.status == COUPON_STATUS.NOT_USED
assert new_coupon.obtain_datetime == payment.coupon.obtain_datetime
assert new_coupon.use_datetime is None
def test_marketer_withdraw_insufficient(self):
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
def pay_to_alipay(self, serial_number, receiver_alipay_id, amount,
desc, payee_type, payee_real_name):
assert len(serial_number) == 32
assert receiver_alipay_id == self.test_case.inviter.alipay_id
assert amount == 1000
assert desc == '付款联盟提现'
return 'refund_request_success'
self.inviter_account.alipay_balance = 10000
self.inviter_account.alipay_withdrawable_balance = 500
self.inviter_account.save()
AlipayUseCases.api_cls = AlipayApisMock
alipay_balance_usecases = AlipayUseCases()
self.assertRaises(BalanceInsufficient,
alipay_balance_usecases.withdraw,
self.inviter_account, 1000)
withdraw = Withdraw.objects.all().order_by('-datetime')[0]
# Check withdraw
assert len(withdraw.serial_number) == 32
assert len(withdraw.transactions.all()) == 0
assert (timezone.now() - withdraw.datetime) < timedelta(seconds=30)
assert withdraw.account_id == self.inviter_account.id
assert withdraw.amount == 1000
assert withdraw.status == WITHDRAW_STATUS.FAILED
# Check the balance
self.inviter_account.refresh_from_db()
assert self.inviter_account.alipay_balance == 10000
assert self.inviter_account.alipay_withdrawable_balance == 500
#!/usr/bin/python3
#
# Project: payunion
# Author: Tian Xu
#
# Copyright (c) 2018 麦禾互动. All rights reserved.
from common.doclink.exceptions import ApiRequestError, ApiReturnedError
from common.error_handler import MarketerError
from common.models import TransferRecord
from common.payment.alipay import AlipayUseCases
from common.payment.exceptions import BalanceInsufficient
from config import TRANSFER_STATUS
alipay_payment = AlipayUseCases()
class TransferRecordManager:
""" 支付宝转账记录 """
@staticmethod
def check_transfer_success(unionid,
account_number=None,
account_name=None):
"""
查询针对该用户名和账号转账成功记录
:param unionid:
:param account_number:
:param account_name:
:return:
"""
filter_params = dict(wechat_unionid=unionid,
status=TRANSFER_STATUS.FINISHED)
def test_refund_without_coupon(self):
self.test_on_payment_success_without_coupon()
payment = Payment.objects.all().order_by('-datetime')[0]
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
self.payment = payment
def refund(self, payment_serial_number, refund_serial_number,
refund_amount):
assert payment_serial_number == self.payment.serial_number
assert len(refund_serial_number) == 32
assert refund_amount == 2000
AlipayUseCases.api_cls = AlipayApisMock
alipay_usecases = AlipayUseCases()
alipay_usecases.request_refund(payment)
# Check the refund data
refund = Refund.objects.all().order_by('-datetime')[0]
assert len(refund.serial_number) == 32
assert len(refund.transactions.all()) == 3
assert (timezone.now() - refund.datetime) < timedelta(seconds=30)
assert refund.status == REFUND_STATUS.FINISHED
assert refund.payment.serial_number == payment.serial_number
# Check the payment data
payment.refresh_from_db()
assert payment.status == PAYMENT_STATUS.REFUND
# Check the transactions.
transactions = refund.transactions.all().order_by('id')
assert transactions[0].object_id == refund.serial_number
assert transactions[
0].content_object.serial_number == refund.serial_number
assert transactions[
0].transaction_type == TRANSACTION_TYPE.MERCHANT_REFUND
assert (timezone.now() -
transactions[0].datetime) < timedelta(seconds=30)
assert transactions[0].account_id == self.merchant_account.id
assert transactions[0].amount == -2000
assert transactions[0].balance_after_transaction == 0
assert transactions[1].object_id == refund.serial_number
assert transactions[
1].content_object.serial_number == refund.serial_number
assert transactions[
1].transaction_type == TRANSACTION_TYPE.PLATFORM_EARNING_MERCHANT_REFUND
assert (timezone.now() -
transactions[1].datetime) < timedelta(seconds=30)
assert transactions[1].account_id == self.platform_account.id
assert transactions[1].amount == 2000
assert transactions[1].balance_after_transaction == 2000
assert transactions[2].object_id == refund.serial_number
assert transactions[
2].content_object.serial_number == refund.serial_number
assert transactions[
2].transaction_type == TRANSACTION_TYPE.PLATFORM_REFUND
assert (timezone.now() -
transactions[2].datetime) < timedelta(seconds=30)
assert transactions[2].account_id == self.platform_account.id
assert transactions[2].amount == -2000
assert transactions[2].balance_after_transaction == 0
# Check the account
self.platform_account.refresh_from_db()
self.merchant_account.refresh_from_db()
assert self.platform_account.alipay_balance == 0
assert self.platform_account.alipay_withdrawable_balance == 0
assert self.merchant_account.alipay_balance == 0
assert self.merchant_account.alipay_withdrawable_balance == 0
def test_refund(self):
self.test_on_payment_success()
payment = Payment.objects.all().order_by('-datetime')[0]
class AlipayApisMock(object):
test_case = self
def __init__(self, app_id, private_key, alipay_public_key):
assert app_id == dynasettings.ALIPAY_APP_ID
assert private_key == dynasettings.ALIPAY_APP_PRIVATE_KEY
assert alipay_public_key == dynasettings.ALIPAY_PUBLIC_KEY
self.payment = payment
def refund(self, payment_serial_number, refund_serial_number,
refund_amount):
assert payment_serial_number == self.payment.serial_number
assert len(refund_serial_number) == 32
assert refund_amount == 1900
AlipayUseCases.api_cls = AlipayApisMock
alipay_usecases = AlipayUseCases()
alipay_usecases.request_refund(payment)
# Check the refund data
refund = Refund.objects.all().order_by('-datetime')[0]
assert len(refund.serial_number) == 32
assert len(refund.transactions.all()) == 3
assert (timezone.now() - refund.datetime) < timedelta(seconds=30)
assert refund.status == REFUND_STATUS.FINISHED
assert refund.payment.serial_number == payment.serial_number
# Check the payment data
payment.refresh_from_db()
assert payment.status == PAYMENT_STATUS.REFUND
# Check the transactions.
transactions = refund.transactions.all().order_by('id')
assert transactions[0].object_id == refund.serial_number
assert transactions[
0].content_object.serial_number == refund.serial_number
assert transactions[
0].transaction_type == TRANSACTION_TYPE.MERCHANT_REFUND
assert (timezone.now() -
transactions[0].datetime) < timedelta(seconds=30)
assert transactions[0].account_id == self.merchant_account.id
assert transactions[0].amount == -1900 + (payment.platform_share +
payment.inviter_share +
payment.originator_share)
assert transactions[0].balance_after_transaction == 0
assert transactions[1].object_id == refund.serial_number
assert transactions[
1].content_object.serial_number == refund.serial_number
assert transactions[
1].transaction_type == TRANSACTION_TYPE.PLATFORM_EARNING_MERCHANT_REFUND
assert (timezone.now() -
transactions[1].datetime) < timedelta(seconds=30)
assert transactions[1].account_id == self.platform_account.id
assert transactions[1].amount == 1900 - (payment.platform_share +
payment.inviter_share +
payment.originator_share)
assert transactions[1].balance_after_transaction == 1900
assert transactions[2].object_id == refund.serial_number
assert transactions[
2].content_object.serial_number == refund.serial_number
assert transactions[
2].transaction_type == TRANSACTION_TYPE.PLATFORM_REFUND
assert (timezone.now() -
transactions[2].datetime) < timedelta(seconds=30)
assert transactions[2].account_id == self.platform_account.id
assert transactions[2].amount == -1900
assert transactions[2].balance_after_transaction == 0
# Check the account
self.platform_account.refresh_from_db()
self.merchant_account.refresh_from_db()
assert self.platform_account.alipay_balance == 0
assert self.platform_account.alipay_withdrawable_balance == 0
assert self.merchant_account.alipay_balance == 0
assert self.merchant_account.alipay_withdrawable_balance == 0
# check the coupon
self.assertEqual(payment.coupon.status, COUPON_STATUS.DESTROYED)
coupon = payment.coupon
new_coupon = payment.coupon.client.coupon_set.exclude(
status=COUPON_STATUS.DESTROYED).first()
self.assertEqual(new_coupon.rule, coupon.rule)
self.assertEqual(new_coupon.discount, coupon.discount)
self.assertEqual(new_coupon.min_charge, coupon.min_charge)
self.assertEqual(new_coupon.originator_merchant,
coupon.originator_merchant)
self.assertEqual(new_coupon.status, COUPON_STATUS.NOT_USED)
self.assertEqual(new_coupon.obtain_datetime, coupon.obtain_datetime)
def test_on_payment_success_without_coupon(self):
self.test_place_order_without_coupon()
new_payment = Payment.objects.all().order_by('-datetime')[0]
alipay_usecases = AlipayUseCases()
alipay_usecases.on_payment_success(new_payment)
# Check the payment data
payment = Payment.objects.get(serial_number=new_payment.serial_number)
assert payment.serial_number == new_payment.serial_number
assert len(payment.transactions.all()) == 3
assert payment.datetime == new_payment.datetime
assert payment.pay_channel == PAY_CHANNELS.ALIPAY
assert payment.status == PAYMENT_STATUS.FROZEN
assert payment.merchant_id == self.merchant.id
assert payment.client_id == self.client.id
assert payment.order_price == 2000
assert payment.coupon is None
assert payment.platform_share == 0
assert payment.inviter_share == 0
assert payment.originator_share == 0
assert payment.note is None
# Check the transactions.
transactions = payment.transactions.all().order_by('id')
assert transactions[0].object_id == payment.serial_number
assert transactions[
0].content_object.serial_number == payment.serial_number
assert transactions[
0].transaction_type == TRANSACTION_TYPE.PLATFORM_RECEIVE
assert (timezone.now() -
transactions[0].datetime) < timedelta(seconds=30)
assert transactions[0].account_id == self.platform_account.id
assert transactions[0].amount == 2000
assert transactions[0].balance_after_transaction == 2000
assert transactions[1].object_id == payment.serial_number
assert transactions[
1].content_object.serial_number == payment.serial_number
assert transactions[
1].transaction_type == TRANSACTION_TYPE.PLATFORM_EXPEND_MERCHANT_RECEIVE
assert (timezone.now() -
transactions[1].datetime) < timedelta(seconds=30)
assert transactions[1].account_id == self.platform_account.id
assert transactions[1].amount == -2000
assert transactions[1].balance_after_transaction == 0
assert transactions[2].object_id == payment.serial_number
assert transactions[
2].content_object.serial_number == payment.serial_number
assert transactions[
2].transaction_type == TRANSACTION_TYPE.MERCHANT_RECEIVE
assert (timezone.now() -
transactions[2].datetime) < timedelta(seconds=30)
assert transactions[2].account_id == self.merchant_account.id
assert transactions[2].amount == 2000
assert transactions[2].balance_after_transaction == 2000
# Check the account
self.platform_account.refresh_from_db()
self.merchant_account.refresh_from_db()
assert self.platform_account.alipay_balance == 0
assert self.platform_account.alipay_withdrawable_balance == 0
assert self.merchant_account.alipay_balance == 2000
assert self.merchant_account.alipay_withdrawable_balance == 0