我正在对Expense模型上的方法进行单元测试。模型的相关部分如下所示:

class Expense(ndb.Model):

    user = ndb.KeyProperty(User, required=True)
    timestamp = ndb.DateTimeProperty(required=True)
    ...

    @staticmethod
    def new(user, amount, comment, description, timestamp):
        """creates a new expense and stores it. It returns the Expense instance"""
        ... validation ...
        exp = Expense(user=user, timestamp=timestamp, ...)
        exp.put()
        return exp

    @staticmethod
    def list(user, from_date=None, to_date=None):
        ...
        query = Expense.query(Expense.user == user)
        if from_date is not None:
            if not isinstance(from_date, datetime.datetime):
                raise ValueError("from_date must be a datetime.datetime instance")
            query.filter(Expense.timestamp >= from_date)
        if to_date is not None:
            if not isinstance(to_date, datetime.datetime):
                raise ValueError("to_date must be a datetime.datetime instance")
            query.filter(Expense.timestamp <= to_date)
        query.order(- Expense.timestamp)
        return query


我有以下单元测试来测试Expense.list

测试的前4行创建Expense模型实例的实例。


一于2013年1月3日
一于2013年11月1日
一于2013年11月2日
一于2014年1月1日


主页,为您节省一些水平滚动。

def test_list_expenses_filter_by_date(user):
    expense1 = Expense.new(user=user, amount=200.12, description="a netbook", comment="", timestamp=datetime.datetime(year=2013, month=1, day=3, hour=12, minute=30))
    expense2 = Expense.new(user=user, amount=1.0, description="A can of coke", comment="Really cheap!",timestamp=datetime.datetime(year=2013, month=11, day=1, hour=12, minute=30))
    expense3 = Expense.new(user=user, amount=3.05, description="A burger", comment="Tasty!", timestamp=datetime.datetime(year=2013, month=11, day=2, hour=12))
    expense4 = Expense.new(user=user, amount=0.30, description="Toilet visit", comment="I normally don't pay for these things, but it was urgent.", timestamp=datetime.datetime(year=2014, month=1, day=1, hour=1, minute=15))

    from_date = datetime.datetime(year=2013, month=5, day=1)
    to_date = datetime.datetime(year=2013, month=12, day=31)
    exps  = Expense.list(user=user, from_date=from_date, to_date=to_date)
    for expense in exps:
        assert from_date <= expense.timestamp <= to_date


但是,当我从pytest运行此测试时,在下面的行中出现了AssertionError,如pytest所示:

assert datetime.datetime(2013, 5, 1, 0, 0) <= datetime.datetime(2013, 1, 3, 12, 30)


E +其中datetime.datetime(2013,1,3,12,30)= Expense(key = Key('Expense',5066549580791808),amount = 200.12,comment = u'',desc ... tamp = datetime.datetime (2013,1,3,12,30),user = Key('User',5629499534213120))。timestamp

那么,为什么我的Expense.list方法返回的数据超出范围的实例呢?

我在应用程序引擎测试平台和请求处理程序中都尝试过,在浏览器中查看输出,因此这不是数据存储区存根中的错误。

我也尝试过以不同的方式构建查询。我没有重复调用Query.filter,而是将过滤器表达式添加到列表中,将该列表作为*args传递给ndb.AND,然后将其传递给单个Query.filter,但这也不起作用。

我排除了NDB中存在错误的可能性。以下测试通过:

def test_ndb_date():
    from google.appengine.ext import ndb

    class Time(ndb.Model):
        timestamp = ndb.DateTimeProperty(required=True)

    t1 = Time(timestamp=datetime.datetime.now() - datetime.timedelta(days=365))
    t2 = Time(timestamp=datetime.datetime.now())
    t3 = Time(timestamp=datetime.datetime.now() + datetime.timedelta(days=365))
    ndb.put_multi([t1, t2, t3])
    from_date = datetime.datetime.now() - datetime.timedelta(days=1)
    to_date = datetime.datetime.now() + datetime.timedelta(days = 1)
    result = Time.query(Time.timestamp <= from_date, Time.timestamp >= to_date).fetch()
    for time in result:
        assert from_date <= time.timestamp >= to_date

最佳答案

我误解了Query.filter的工作方式。它不会更改当前的Query实例,但是会返回一个新实例。

关于python - NDB不能按日期正确查询,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/20955523/

10-12 16:51