在哪里模拟测试调用服务的Django视图?

时间:2015-05-20 19:12:07

标签: python django python-mock

这是Django 1.6.8,Python 2.7和mock库。

我有一个视图,使用suds调用销售税信息的远程服务(这是一个简化版本):

def sales_tax(self, bundle_slug):
    bundle = get_object_or_404(Bundle, slug=bundle_slug,
                                active=True)

    cloud = TaxCloud(TAXCLOUD_API_ID, TAXCLOUD_API_KEY)
    origin = Address('origin address')
    destination = Address('customer address')
    cart_item = CartItem(bundle.sku, TAXCLOUD_TIC_ONLINE_GAMES,
                        bundle.price, 1)

    try:
        rate_info = cloud.get_rate(origin, destination,
                                    [cart_item],
                                    str(customer.id))

        sales_tax = Decimal(rate_info['sales_tax'])

        response = {'salesTax': locale.currency(sales_tax),
                    'total':  locale.currency(bundle.price + sales_tax)}
    except TaxCloudException as tce:
        response = {
            'error': str(tce)
        }

以下是TaxCloud类的相关代码:

from suds.client import Client
from suds import WebFault

class TaxCloud(object):

    def __init__(self, api_login_id, api_key):
        self.api_login_id = api_login_id
        self.api_key = api_key
        self.soap_url = 'https://api.taxcloud.net/1.0/TaxCloud.asmx'
        self.wsdl_url = 'https://api.taxcloud.net/1.0/?wsdl' 

        self.client = Client(url=self.wsdl_url, location=self.soap_url, faults=False)

    def get_rate(self, billing_address, origin_address, raw_cart_items, customer_id):
        address = self.convert_to_address(billing_address)
        origin = self.convert_to_address(origin_address)
        cart_items = self.convert_to_cart_list(raw_cart_items)

        response = self.client.service.Lookup(self.api_login_id,
                                              self.api_key, customer_id,
                                              self.cart_id(customer_id), cart_items,
                                              address, origin, True, None)

        if( response[1].ResponseType == 'Error' ):
            raise TaxCloudException(response[1].Messages[0][0].Message)

        return {
            'sales_tax': str(response[1].CartItemsResponse[0][0].TaxAmount),
            'cart_id': response[1].CartID
        }

在我的视图测试中,我不想调用远程服务。使用this example of a mocked client,我构建了一个愚蠢的模拟类(我的ClientMock完全匹配该答案中的示例):

class TaxCloudServiceClientMock(ClientMock):
    """
    Mock object that implements remote side services.
    """

    def Lookup(cls, api_id, api, customer_id, cart_id, cart_items,
               address, origin, flag, setting):
        """
        Stub for remote service.
        """
        return """(200, (LookupRsp){
               ResponseType = "OK"
               Messages = ""
               CartID = "82cabf35faf66d8b197c7040a9f7382b3f61573fc043d73717"
               CartItemsResponse =
                  (ArrayOfCartItemResponse){
                     CartItemResponse[] =
                        (CartItemResponse){
                           CartItemIndex = 1
                           TaxAmount = 0.10875
                        },
                  }
                })"""

在我的测试中,我正在尝试@patch Client类中使用的TaxCloud

@patch('sales.TaxCloud.Client', new=TaxCloudServiceClientMock)
def test_can_retrieve_sales_tax(self):
    from django.core.urlresolvers import reverse

    tax_url = reverse('sales_tax', kwargs={'bundle_slug': self.bundle.slug})
    self.client.login(username=self.user.username, password='testpassword')

    response = self.client.get(tax_url, {'address1': '1234 Blah St',
                                        'city': 'Some City',
                                        'state': 'OH',
                                        'zipcode': '12345',
                                        'country': 'US'},
        HTTP_X_REQUESTED_WITH='XMLHttpRequest')

    self.assertEqual(response.status_code, 200)

然而,远程呼叫仍在进行中。根据“Where to mock”文档,我正确定位sales.TaxCloud.Client而不是suds.client.Client

什么可能导致补丁被忽略/绕过?

0 个答案:

没有答案