From 6e8ea50c199f7821db56c1a1397c3b90cd96c8db Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Bastien=20G=C3=A9rard?= Date: Fri, 14 Jun 2019 21:04:02 +0200 Subject: [PATCH] "added another aggregation test" This reverts commit 4c31193b82ca665fe27b1ddce3cbacbb2b7fcd32. --- tests/queryset/queryset.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/tests/queryset/queryset.py b/tests/queryset/queryset.py index 1f09ca59..55f256d9 100644 --- a/tests/queryset/queryset.py +++ b/tests/queryset/queryset.py @@ -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()