编写单元测试类的__init__方法

时间:2012-02-12 02:08:20

标签: python unit-testing nose

我对单元测试和编写/使用异常非常新。我目前正在努力学习最佳实践并将它们集成到我的项目中。作为对我一直在阅读的一些事情的考验,我写了一个简单的合同模块。下面是契约类的初始化,它有几个相互依赖的参数。

我将如何/应该根据其参数依赖性为init方法编写测试。

提前致谢!

def __init__(self, code, description ,contract_type,
             start_date ,end_date ,reminder_date, 
             customer=None, isgroup=False, vendor=None, 
             discount_perc=None):

    contract_types = ['item','vendor']
    self.code = code
    self.description = description
    self.contract_type = contract_type
    self.start_date = start_date
    self.end_date = end_date
    self.reminder_date = reminder_date
    if contract_type not in contract_types:
        raise AttributeError("Valid contract types are 'item' & 'vendor'")
    if isgroup:
        if customer:
            raise AttributeError("Group contracts should not have 'customer' passed in")
        self.is_group_contract = True
    else:
        if customer:
            self.add_customer(customer)
        else:
            raise AttributeError('Customer required for non group contracts.')
    if contract_type == 'vendor':
        if vendor and discount_perc:
            self.vendor = vendor
            self.discount_perc = discount_perc
        else:
            if not vendor:
                raise AttributeError('Vendor contracts require vendor to be passed in')
            if not discount_perc:
                raise AttributeError('Vendor contracts require discount_perc(Decimal)')

如果这类问题不适合SO,我可能会更好地去哪儿?

1 个答案:

答案 0 :(得分:3)

我将__init__视为与任何其他(非类或静态)方法类似 - 根据各种输入组合测试预期输出。但除此之外,我还会测试它返回(或不返回,取决于你的要求)单例对象。
但是,有人可能更喜欢将单例测试作为__new__相关的测试用例进行提取。

最终您将获得以下测试:

  1. 无效的参数类型处理(空/非空字符串,整数,元组,字符串等)。
  2. 无效的参数组合处理(在您的情况下,它是上升的例外)。
  3. 可选参数存在/不存在处理(默认值有效,自定义也可以等)。
  4. 有效的参数组合处理(正流量工作)。
  5. 结果对象的属性存在/不存在及其值(大多数情况下,您在其他方法中依赖它们)。
  6. 结果对象是单身(或不是)。
  7. ???
  8. 另一个提示:将contract_types = ['item','vendor']提取到class属性将有助于业务逻辑测试组织。