主页 / 问题 / 808122
Accepted
Narnik Gamarnik
Asked:2020-04-02 04:43:56 +0000 UTC2020-04-02 04:43:56 +0000 UTC

测试未涵盖锁定功能

  • 772

服务.py

import blockcypher
from django.contrib.auth import get_user_model
from .utils import get_wallet_model


def generate_new_address(user, symbol):
    if isinstance(user, get_user_model()) and get_wallet_model(symbol):
        try:
            r = blockcypher.generate_new_address(coin_symbol=self.coin_symbol, api_key=api_key)
            obj = get_wallet_model(symbol).objects.create(
                user = user,
                private = r['private'],
                public = r['public'],
                address = r['address'],
                wif = r['wif']
            )
            return obj
        except Exception:
            return None

尝试锁定时，以免每次都拉取外部API，我尝试以下操作：

def fake_generate_new_address(self, user, symbol):
    if isinstance(user, get_user_model()) and utils.get_wallet_model(symbol):
        try:
            r = {
                'address': '1HL7fjCRGKC4EfPRjHVgmLmi7Bmpu8hGps',
                'private': '2a9a2a50252bf2eb24553e70861e1774aa8507af1b9497e1da01fa0086a3dfb7',
                'public': '02053a352366b3bee2e3d0d29b346822706bdf7af3cd00f7ce9d3516010d4a37c4',
                'wif': 'KxeXM1gzy5PMJ47orJiZrBK89DycjPxbq7GVS1KcV7UKAFgFTQJx'
            }
            print(r)
            obj = utils.get_wallet_model(symbol).objects.create(
                user = user,
                private = r['private'],
                public = r['public'],
                address = r['address'],
                wif = r['wif']
                )
            return obj
        except Exception:
            return None


@mock.patch('wallets.services.generate_new_address')
def test_new_address_with_valid_data(self):
    obj = services.generate_new_address(self.user, 'dash')
    assertTrue(isinstance(obj, models.Dash)

但结果是：

在此处输入图像描述

我究竟做错了什么？

这个例子只是一长串列表中的一个。以完全相同的方式，例如，不可能涵盖@property 和@cached_property 模型。

python

1 个回答

  1. Best Answer

    弄清楚模拟是如何工作的。

    部分，设法覆盖如下：

    def test_new_address_with_valid_data(self):
    import blockcypher
    blockcypher.generate_new_address = mock.MagicMock(return_value={
        'address': '1HL7fjCRGKC4EfPRjHVgmLmi7Bmpu8hGps',
        'private': '2a9a2a50252bf2eb24553e70861e1774aa8507af1b9497e1da01fa0086a3dfb7',
        'public': '02053a352366b3bee2e3d0d29b346822706bdf7af3cd00f7ce9d3516010d4a37c4',
        'wif': 'KxeXM1gzy5PMJ47orJiZrBK89DycjPxbq7GVS1KcV7UKAFgFTQJx'
    })
    obj = services.generate_new_address(self.user, 'btc')
    self.assertTrue(
        isinstance(
            obj,
            models.Btc
        )
    )

    也就是说，使用MagicMock，您可以替换任何函数或类的返回值，甚至是那些从外部导入的。

    • 1

相关问题