py.test:模拟一个不断变化的日期时间对象

时间:2017-10-20 11:15:05

标签: python unit-testing mocking pytest

如果该对象不断变化,您如何模拟返回的日期时间对象?

例如,我有一些查询api的代码,获取一些时间序列数据并将其作为一系列日期时间对象返回。

所以来自查询的api的返回数据看起来像是:

{
    'from_date': 1497881400000.0,
                'group_by': [],
                'message': '',
                'query': 'avg:system.load.norm.5{role:lb}',
                'res_type': 'time_series',
                'resp_version': 1,
                'series': [{'aggr': 'avg',
                                    'attributes': {},
                                    'display_name': 'system.load.norm.5',
                                    'end': 1497709799000,
                                    'expression': 'avg:system.load.norm.5{role:lb}',
                                    'interval': 1,
                                    'length': 60,
                                    'metric': 
                                    'avg:system.load.norm.5{role:lb}',
                                    'pointlist': [[1497881400000.0,
                                     0.06923198629355053], 
                                    [.. 60 more points]],
                                    'query_index': 0,
                                    'scope': 'role:lb',
                                    'start': 1497709742000,
                                    'unit': [None, None]}],
                'status': 'ok',
                'to_date': 1497709800000
}

我的代码执行此操作:

from typing import NamedTuple

class SampleMetric(NamedTuple):
    query: str
    time: datetime.datetime
    value: int

pointlist = api_query['series'][0]['pointlist']
return [SampleMetric(query=metric,
                             time=datetime.datetime.fromtimestamp(
                                 point[0] / 1000),
                             value=point[1]) for point in pointlist]

在上述api数据的情况下看起来像这样:

[SampleMetric(query='avg:system.load.norm.5{role:lb}', time=datetime.datetime(2017, 10, 20, 11, 2), value=231.8605833053589),
 SampleMetric(query='avg:system.load.norm.5{role:lb}', time=datetime.datetime(2017, 10, 20, 11, 3), value=243.17432022094727),...]

所以我可以轻松地创建一个模拟api对象,它返回一个字典,如上面的示例所示,但我真的很难模拟返回的更改日期时间对象。

即。 time=datetime.datetime(2017, 10, 20, 11, 2)

有人可以给我任何关于如何做到这一点的想法吗?目前我已经解决了这个问题,只有一个时间点指向我的返回数据并硬编码时间值,但我觉得这不是一个真正的世界测试,因为它应该返回60个结果。

忽略此示例中使用的实际值,我知道它们是错的,但这只是为了说明目的,因为我正在努力解释我的意思。

1 个答案:

答案 0 :(得分:1)

我不确定你的意思是什么"经常改变",所以我会采取两个猜测并回答它们。

我的第一个猜测是,当结果取决于当前时间时,您想知道如何编写单元测试。这使得输出不断变化,因此测试很难写。

要解决这个问题,我要将当前时间作为显式输入参数传递给我测试的代码,或者我模拟datetime.now()方法。这样,单元测试总是根据同一时间计算,并且应该总是得到相同的结果。这也使得测试有趣的边缘情况变得更加容易,例如夏令时的变化。您可以为每个有趣的场景编写一个单独的测试用例,其中包含不同的当前时间"在每个。

我的第二个猜测是,您正在询问如何为输入数据创建一组60个不同的日期时间值。我可能会为前几个条目计算几个有趣的值,然后用for循环或range()表达式生成其余的值。