"added another aggregation test"

This reverts commit 4c31193b82ca665fe27b1ddce3cbacbb2b7fcd32.
This commit is contained in:
Bastien Gérard 2019-06-14 21:04:02 +02:00
parent 1fcd706e11
commit 6e8ea50c19

View File

@ -7,7 +7,6 @@ from decimal import Decimal
from bson import DBRef, ObjectId
import pymongo
from pymongo.errors import ConfigurationError
from pymongo.read_preferences import ReadPreference
from pymongo.results import UpdateResult
import six
@ -17,7 +16,7 @@ from mongoengine import *
from mongoengine.connection import get_connection, get_db
from mongoengine.context_managers import query_counter, switch_db
from mongoengine.errors import InvalidQueryError
from mongoengine.mongodb_support import get_mongodb_version, MONGODB_36
from mongoengine.mongodb_support import MONGODB_36, get_mongodb_version
from mongoengine.queryset import (DoesNotExist, MultipleObjectsReturned,
QuerySet, QuerySetManager, queryset_manager)
@ -832,8 +831,6 @@ class QuerySetTest(unittest.TestCase):
def test_bulk_insert(self):
"""Ensure that bulk insert works"""
MONGO_VER = self.mongodb_version
class Comment(EmbeddedDocument):
name = StringField()
@ -847,10 +844,6 @@ class QuerySetTest(unittest.TestCase):
Blog.drop_collection()
# get MongoDB version info
connection = get_connection()
info = connection.test.command('buildInfo')
# Recreates the collection
self.assertEqual(0, Blog.objects.count())
@ -5386,6 +5379,13 @@ class QuerySetTest(unittest.TestCase):
{'_id': None, 'avg': 29, 'total': 2}
])
data = Person.objects().aggregate({'$match': {'name': 'Isabella Luanna'}})
self.assertEqual(list(data), [
{u'_id': p1.pk,
u'age': 16,
u'name': u'Isabella Luanna'}]
)
def test_queryset_aggregation_with_skip(self):
class Person(Document):
name = StringField()