如何在App Engine NDB中测试DateTimeProperty?

时间:2012-10-23 06:55:11

标签: python google-app-engine app-engine-ndb

我正在尝试使用App Engine的NDB测试DateTimeProperty的过滤器,但是我将它设置为auto_now?

有没有办法绕过这个进行单元测试?

示例:

class MyModel(ndb.Model)
  timestamp = ndb.DateTimeProperty(auto_now)
  name = ndb.StringProperty()


def testMyModelFilter(self):
  test1 = MyModel()
  test1.timestamp = datetime.datetime.now() - datetime.timedelta(hours=2)
  test1.put()
  test2 = MyModel()
  test2.timestamp = datetime.datetime.now() - datetime.timedelta(hours=1)
  test2.put()

  hour_ago = datetime.datetime.now() - datetime.timedelta(hours=1)
  fetched = MyModel.query().filter(MyModel.timestamp < hour_ago).fetch(
      None, keys_only=True)

不幸的是,当我使用test.put()将其提交到数据存储区时,它会使用put()的时间。

1 个答案:

答案 0 :(得分:10)

因此,您可以尝试的一件事(正如@mjibson所暗示的那样)在测试期间覆盖您的模型。由于MyModel本身是一个对象,因此您可以将_auto_now的{​​{1}}属性修改为False以进行测试。例如:

timestamp

我记得在其他地方看过这种技术,所以如果我能找到它,我会链接(如果它可以工作,当然:)。