manyi/testcase/unit/test_common.py

# -*- coding: utf-8 -*-
# !/usr/bin/env python

from apilib.monetary import RMB
from apps.web.agent.define import AGENT_INCOME_TYPE
from apps.web.common.models import WithdrawRecord
from apps.web.common.transaction import WITHDRAW_PAY_TYPE
from apps.web.dealer.define import DEALER_INCOME_TYPE


def test_new_withdraw_record(bankcard, dealer, agent_no_customized, gateway):
    withdraw_record = WithdrawRecord.create(dealer, gateway, bankcard, DEALER_INCOME_TYPE.DEVICE_INCOME,
                                            WITHDRAW_PAY_TYPE.BANK, {
                                                'amount': RMB(10),
                                                'serviceFee': RMB('0.08'),
                                                'actualPay': RMB('9.92')
                                            }, False, False)
    assert withdraw_record

    withdraw_record = WithdrawRecord.create(dealer, gateway, bankcard, DEALER_INCOME_TYPE.AD_INCOME,
                                            WITHDRAW_PAY_TYPE.WECHAT, {
                                                'amount': RMB(10),
                                                'serviceFee': RMB('0.08'),
                                                'actualPay': RMB('9.92')
                                            }, False, False)
    assert withdraw_record

    withdraw_record = WithdrawRecord.create(agent_no_customized, gateway, bankcard,
                                            AGENT_INCOME_TYPE.DEALER_WITHDRAW_FEE,
                                            WITHDRAW_PAY_TYPE.BANK, {
                                                'amount': RMB(10),
                                                'serviceFee': RMB('0.08'),
                                                'actualPay': RMB('9.92')
                                            }, True, False)
    assert withdraw_record

    withdraw_record = WithdrawRecord.create(agent_no_customized, gateway, bankcard, AGENT_INCOME_TYPE.DEALER_DEVICE_FEE,
                                            WITHDRAW_PAY_TYPE.BANK, {
                                                'amount': RMB(10),
                                                'serviceFee': RMB('0.08'),
                                                'actualPay': RMB('9.92')
                                            }, True, False)
    assert withdraw_record