mongomock的内部总金额似乎不起作用

Question

我有一个聚合Mongo查询，它投影一些字段，并使用$sum计算另外两个字段。查询按预期工作，所以我为它创建了一个单元测试，但令我惊讶的是，测试失败了。

我创建了一个最小的、完整的、可验证的例子来检验我的假设，即这是MongoMock的问题，而且似乎是！

这是代码：

import mongoengine as mongo
from mongoengine import connect
from mongoengine.queryset import QuerySet
class ValuesList(mongo.EmbeddedDocument):
updated_value = mongo.DecimalField()

class ValuesHistory(mongo.Document):
name = mongo.StringField()
base_value = mongo.DecimalField()
values_list = mongo.EmbeddedDocumentListField(ValuesList, required=False)
meta = {
'collection' : 'values_history'
}
def __str__(self):
return 'name: {}nbase_value: {}n'.format(self.name, self.base_value)

def migrate_data(new_collection):
ValuesHistory.objects.aggregate(
{'$project': {'name': 1,
'base_value': {'$sum': ['$base_value', {'$arrayElemAt': ['$values_list.updated_value', -1]}]}
}
},
{'$out': "{}".format(new_collection)}
)

def clear_tables_and_insert_test_data(db):
db.test.values_history.drop()
db.test.updated_values.drop()
ValuesHistory(name='first',
base_value=100,
values_list=[ValuesList(updated_value=5),
ValuesList(updated_value=15)]).save()
def run_aggregate_query_with_db(db):
new_collection_name = 'updated_values'
migrate_data(new_collection_name)
new_group = ValuesHistory.switch_collection(ValuesHistory(), new_collection_name)
aggregated_values = QuerySet(ValuesHistory, new_group._get_collection()).all()
for value in aggregated_values:
print(value)
db.close()

关于上面代码的快速解释。

ValuesHistory是一个类，包含字符串名称、数字base_value和值列表(ValuesList类(。

方法clear_tables_and_insert_test_data清除此测试中使用的两个表并插入一些测试数据。

migrate_data方法中的查询创建一个新集合(通过$out运算符(，并且新创建集合的base_value应该是values_list列表中当前值和最后一个值的总和。在我的情况下，它应该是115(100是当前值，15是列表上的最后一个值(。

如果我使用到本地MongoDB的连接运行代码，如下所示：

if __name__ == '__main__':
db = connect('test') # connect to real instance of Mongo
clear_tables_and_insert_test_data(db)
run_aggregate_query_with_db(db)

结果我得了115分，这正是人们所期望的。

如果我改为使用到MongoMock:的连接

if __name__ == '__main__':
db = connect('test', host='mongomock://localhost') # Connect to MongoMock instance
clear_tables_and_insert_test_data(db)
run_aggregate_query_with_db(db)

结果我得了100，这很奇怪！看起来$sum运算符没有正确完成它的工作，因为100和15的总和得到了100！

EDIT：我还尝试使用$add运算符，但问题仍然存在，当它应该是115时，会产生100。

TL；DR

问题：我应该如何在MongoMock上的聚合管道中使用$sum(或$add(，以便产生正确的值？

Answer 1

所描述的问题实际上是Mongomock上的一个错误，该错误一直存在到3.14.0版本。

在最初的问题发布后，我在Mongomock的github上打开了一个问题，描述了这个问题。在它被修复后不久，3.15.0版本已经在几天前发布。我运行了这个问题的代码，现在$add和$sum运算符的问题都解决了！

TL；DR

更新到Mongomock 3.15.0就足以解决这个问题。