# -*- coding: utf-8 -*- from collections import OrderedDict import copy import datetime import os from packaging import version import re import sys import time from unittest import TestCase, skipIf, skipUnless import uuid import mongomock from mongomock import ConfigurationError from mongomock import Database from mongomock import helpers from mongomock import InvalidURI from mongomock import OperationFailure try: from bson import DBRef, decimal128 from bson.objectid import ObjectId import pymongo from pymongo import MongoClient as PymongoClient from pymongo import read_concern from pymongo.read_preferences import ReadPreference except ImportError: from mongomock.object_id import ObjectId from mongomock import read_concern from tests.utils import DBRef try: from bson.code import Code from bson.regex import Regex from bson.son import SON import execjs # noqa pylint: disable=unused-import _HAVE_MAP_REDUCE = any(r.is_available() for r in execjs.runtimes().values()) except ImportError: _HAVE_MAP_REDUCE = False Code = str from tests.multicollection import MultiCollection SERVER_VERSION = version.parse(mongomock.SERVER_VERSION) class InterfaceTest(TestCase): def test__can_create_db_without_path(self): self.assertIsNotNone(mongomock.MongoClient()) def test__can_create_db_with_path(self): self.assertIsNotNone(mongomock.MongoClient('mongodb://localhost')) def test__can_create_db_with_multiple_pathes(self): hostnames = ['mongodb://localhost:27017', 'mongodb://localhost:27018'] self.assertIsNotNone(mongomock.MongoClient(hostnames)) def test__repr(self): self.assertEqual(repr(mongomock.MongoClient()), "mongomock.MongoClient('localhost', 27017)") def test__bad_uri_raises(self): with self.assertRaises(InvalidURI): mongomock.MongoClient('http://host1') with self.assertRaises(InvalidURI): mongomock.MongoClient('://host1') with self.assertRaises(InvalidURI): mongomock.MongoClient('mongodb://') with self.assertRaises(InvalidURI): mongomock.MongoClient('mongodb://localhost/path/mongodb.sock') with self.assertRaises(InvalidURI): mongomock.MongoClient('mongodb://localhost?option') with self.assertRaises(ValueError): mongomock.MongoClient('mongodb:host2') def test__none_uri_host(self): self.assertIsNotNone(mongomock.MongoClient('host1')) self.assertIsNotNone(mongomock.MongoClient('//host2')) self.assertIsNotNone(mongomock.MongoClient('mongodb:12')) class DatabaseGettingTest(TestCase): def setUp(self): super(DatabaseGettingTest, self).setUp() self.client = mongomock.MongoClient() @skipIf(not helpers.HAVE_PYMONGO, 'pymongo not installed') def test__get_database_read_concern(self): db = self.client.get_database('a', read_concern=read_concern.ReadConcern('majority')) self.assertEqual('majority', db.read_concern.level) def test__getting_database_via_getattr(self): db1 = self.client.some_database_here db2 = self.client.some_database_here self.assertIs(db1, db2) self.assertIs(db1, self.client['some_database_here']) self.assertIsInstance(db1, Database) self.assertIs(db1.client, self.client) self.assertIs(db2.client, self.client) def test__getting_database_via_getitem(self): db1 = self.client['some_database_here'] db2 = self.client['some_database_here'] self.assertIs(db1, db2) self.assertIs(db1, self.client.some_database_here) self.assertIsInstance(db1, Database) def test__drop_database(self): db = self.client.a collection = db.a doc_id = collection.insert_one({'aa': 'bb'}).inserted_id self.assertEqual(collection.count_documents({'_id': doc_id}), 1) self.client.drop_database('a') self.assertEqual(collection.count_documents({'_id': doc_id}), 0) db = self.client.a collection = db.a doc_id = collection.insert_one({'aa': 'bb'}).inserted_id self.assertEqual(collection.count_documents({'_id': doc_id}), 1) self.client.drop_database(db) self.assertEqual(collection.count_documents({'_id': doc_id}), 0) def test__drop_database_system_collection(self): db = self.client.a collection = db['system.foo'] doc_id = collection.insert_one({'aa': 'bb'}).inserted_id self.assertEqual(collection.count_documents({'_id': doc_id}), 1) self.client.drop_database('a') self.assertEqual(collection.count_documents({'_id': doc_id}), 0) def test__drop_database_indexes(self): db = self.client.somedb collection = db.a collection.create_index('simple') collection.create_index([('value', 1)], unique=True) collection.create_index([('sparsed', 1)], unique=True, sparse=True) self.client.drop_database('somedb') # Make sure indexes' rules no longer apply collection.insert_one({'value': 'not_unique_but_ok', 'sparsed': 'not_unique_but_ok'}) collection.insert_one({'value': 'not_unique_but_ok'}) collection.insert_one({'sparsed': 'not_unique_but_ok'}) self.assertEqual(collection.count_documents({}), 3) def test__sparse_unique_index(self): db = self.client.somedb collection = db.a collection.create_index([('value', 1)], unique=True, sparse=True) collection.insert_one({'value': 'should_be_unique'}) collection.insert_one({'simple': 'simple_without_value'}) collection.insert_one({'simple': 'simple_without_value2'}) collection.create_index([('value', 1)], unique=True, sparse=True) def test__alive(self): self.assertTrue(self.client.alive()) def test__dereference(self): db = self.client.a collection = db.a to_insert = {'_id': 'a', 'aa': 'bb'} collection.insert_one(to_insert) a = db.dereference(DBRef('a', 'a', db.name)) self.assertEqual(to_insert, a) def test__getting_default_database_valid(self): def gddb(uri): client = mongomock.MongoClient(uri) return client, client.get_default_database() c, db = gddb('mongodb://host1/foo') self.assertIsNotNone(db) self.assertIsInstance(db, Database) self.assertIs(db.client, c) self.assertIs(db, c['foo']) c, db = gddb('mongodb://host1/bar') self.assertIs(db, c['bar']) c, db = gddb(r'mongodb://a%00lice:f%00oo@127.0.0.1/t%00est') self.assertIs(db, c['t\x00est']) c, db = gddb('mongodb://bob:bar@[::1]:27018/admin') self.assertIs(db, c['admin']) c, db = gddb('mongodb://%24am:f%3Azzb%40zz@127.0.0.1/' 'admin%3F?authMechanism=MONGODB-CR') self.assertIs(db, c['admin?']) c, db = gddb(['mongodb://localhost:27017/foo', 'mongodb://localhost:27018/foo']) self.assertIs(db, c['foo']) # As of pymongo 3.5, get_database() is equivalent to # the old behavior of get_default_database() client = mongomock.MongoClient('mongodb://host1/foo') self.assertIs(client.get_database(), client['foo']) def test__getting_default_database_invalid(self): def client(uri): return mongomock.MongoClient(uri) c = client('mongodb://host1') with self.assertRaises(ConfigurationError): c.get_default_database() c = client('host1') with self.assertRaises(ConfigurationError): c.get_default_database() c = client('') with self.assertRaises(ConfigurationError): c.get_default_database() c = client('mongodb://host1/') with self.assertRaises(ConfigurationError): c.get_default_database() def test__getting_default_database_with_default_parameter(self): c = mongomock.MongoClient('mongodb://host1/') self.assertIs(c.get_default_database('foo'), c['foo']) self.assertIs(c.get_default_database(default='foo'), c['foo']) def test__getting_default_database_ignoring_default_parameter(self): c = mongomock.MongoClient('mongodb://host1/bar') self.assertIs(c.get_default_database('foo'), c['bar']) self.assertIs(c.get_default_database(default='foo'), c['bar']) @skipIf(not helpers.HAVE_PYMONGO, 'pymongo not installed') def test__getting_default_database_preserves_options(self): client = mongomock.MongoClient('mongodb://host1/foo') db = client.get_database(read_preference=ReadPreference.NEAREST) self.assertEqual(db.name, 'foo') self.assertEqual(ReadPreference.NEAREST, db.read_preference) self.assertEqual(ReadPreference.PRIMARY, client.read_preference) class UTCPlus2(datetime.tzinfo): def fromutc(self, dt): return dt + self.utcoffset(dt) def tzname(self, dt): return '' def utcoffset(self, dt): return datetime.timedelta(hours=2) def dst(self, dt): return datetime.timedelta() @skipIf(not helpers.HAVE_PYMONGO, 'pymongo not installed') @skipIf(os.getenv('NO_LOCAL_MONGO'), 'No local Mongo server running') class _CollectionComparisonTest(TestCase): """Compares a fake collection with the real mongo collection implementation This is done via cross-comparison of the results. """ def setUp(self): super(_CollectionComparisonTest, self).setUp() self.fake_conn = mongomock.MongoClient() self.mongo_conn = self._connect_to_local_mongodb() self.db_name = 'mongomock___testing_db' self.collection_name = 'mongomock___testing_collection' self.mongo_conn.drop_database(self.db_name) self.mongo_collection = self.mongo_conn[self.db_name][self.collection_name] self.fake_collection = self.fake_conn[self.db_name][self.collection_name] self.cmp = MultiCollection({ 'fake': self.fake_collection, 'real': self.mongo_collection, }) def _create_compare_for_collection(self, collection_name, db_name=None): if not db_name: db_name = self.db_name mongo_collection = self.mongo_conn[db_name][collection_name] fake_collection = self.fake_conn[db_name][collection_name] return MultiCollection({ 'fake': fake_collection, 'real': mongo_collection, }) def _connect_to_local_mongodb(self, num_retries=60): """Performs retries on connection refused errors (for travis-ci builds)""" for retry in range(num_retries): if retry > 0: time.sleep(0.5) try: return PymongoClient( host=os.environ.get('TEST_MONGO_HOST', 'localhost'), maxPoolSize=1 ) except pymongo.errors.ConnectionFailure as e: if retry == num_retries - 1: raise if 'connection refused' not in e.message.lower(): raise def tearDown(self): super(_CollectionComparisonTest, self).tearDown() self.mongo_conn.close() class EqualityCollectionTest(_CollectionComparisonTest): def test__database_equality(self): self.assertEqual(self.mongo_conn[self.db_name], self.mongo_conn[self.db_name]) self.assertEqual(self.fake_conn[self.db_name], self.fake_conn[self.db_name]) @skipIf(sys.version_info < (3,), 'Older versions of Python do not handle hashing the same way') @skipIf( helpers.PYMONGO_VERSION and helpers.PYMONGO_VERSION < version.parse('3.12'), "older versions of pymongo didn't have proper hashing") def test__database_hashable(self): {self.mongo_conn[self.db_name]} # pylint: disable=pointless-statement {self.fake_conn[self.db_name]} # pylint: disable=pointless-statement @skipIf(sys.version_info < (3,), 'Older versions of Python do not handle hashing the same way') @skipUnless( helpers.PYMONGO_VERSION and helpers.PYMONGO_VERSION < version.parse('3.12'), "older versions of pymongo didn't have proper hashing") def test__database_not_hashable(self): with self.assertRaises(TypeError): {self.mongo_conn[self.db_name]} # pylint: disable=pointless-statement with self.assertRaises(TypeError): {self.fake_conn[self.db_name]} # pylint: disable=pointless-statement class MongoClientCollectionTest(_CollectionComparisonTest): def test__find_is_empty(self): self.cmp.do.delete_many({}) self.cmp.compare.find() def test__inserting(self): self.cmp.do.delete_many({}) data = {'a': 1, 'b': 2, 'c': 'data'} self.cmp.do.insert_one(data) self.cmp.compare.find() # single document, no need to ignore order def test__bulk_insert(self): objs = [{'a': 2, 'b': {'c': 3}}, {'c': 5}, {'d': 7}] results_dict = self.cmp.do.insert_many(objs) for results in results_dict.values(): self.assertEqual(len(results.inserted_ids), len(objs)) self.assertEqual( len(set(results.inserted_ids)), len(results.inserted_ids), 'Returned object ids not unique!') self.cmp.compare_ignore_order.find() def test__insert(self): if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.insert({'a': 1}) return self.cmp.do.insert({'a': 1}) self.cmp.compare.find() def test__insert_one(self): self.cmp.do.insert_one({'a': 1}) self.cmp.compare.find() def test__insert_many(self): self.cmp.do.insert_many([{'a': 1}, {'a': 2}]) self.cmp.compare.find() def test__save(self): # add an item with a non ObjectId _id first. self.cmp.do.insert_one({'_id': 'b'}) if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.save({'_id': ObjectId(), 'someProp': 1}) return self.cmp.do.save({'_id': ObjectId(), 'someProp': 1}) self.cmp.compare_ignore_order.find() def test__insert_object_id_as_dict(self): self.cmp.do.delete_many({}) doc_ids = [ # simple top-level dictionary {'A': 1}, # dict with value as list {'A': [1, 2, 3]}, # dict with value as dict {'A': {'sub': {'subsub': 3}}} ] for doc_id in doc_ids: _id = { key: value.inserted_id for key, value in self.cmp.do.insert_one({'_id': doc_id, 'a': 1}).items() } self.assertEqual(_id['fake'], _id['real']) self.assertEqual(_id['fake'], doc_id) self.assertEqual(_id['real'], doc_id) self.assertEqual(type(_id['fake']), type(_id['real'])) self.cmp.compare.find({'_id': doc_id}) docs = self.cmp.compare.find_one({'_id': doc_id}) self.assertEqual(docs['fake']['_id'], doc_id) self.assertEqual(docs['real']['_id'], doc_id) self.cmp.do.delete_one({'_id': doc_id}) def test__count(self): if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.count() return self.cmp.compare.count() self.cmp.do.insert_one({'a': 1}) self.cmp.compare.count() self.cmp.do.insert_one({'a': 0}) self.cmp.compare.count() self.cmp.compare.count({'a': 1}) @skipIf( helpers.PYMONGO_VERSION and helpers.PYMONGO_VERSION < version.parse('3.8'), 'older version of pymongo does not have count_documents') def test__count_documents(self): self.cmp.compare.count_documents({}) self.cmp.do.insert_one({'a': 1}) self.cmp.compare.count_documents({}) self.cmp.do.insert_one({'a': 0}) self.cmp.compare.count_documents({}) self.cmp.compare.count_documents({'a': 1}) self.cmp.compare.count_documents({}, skip=10) self.cmp.compare.count_documents({}, skip=0) self.cmp.compare.count_documents({}, skip=10, limit=100) self.cmp.compare.count_documents({}, skip=10, limit=3) self.cmp.compare_exceptions.count_documents({}, limit='one') self.cmp.compare_exceptions.count_documents({}, limit='1') @skipIf( helpers.PYMONGO_VERSION and helpers.PYMONGO_VERSION < version.parse('3.8'), 'older version of pymongo does not have estimated_document_count') def test__estimated_document_count(self): self.cmp.compare.estimated_document_count() self.cmp.do.insert_one({'a': 1}) self.cmp.compare.estimated_document_count() self.cmp.do.insert_one({'a': 0}) self.cmp.compare.estimated_document_count() if SERVER_VERSION < version.parse('5'): self.cmp.compare.estimated_document_count(skip=2) else: self.cmp.compare_exceptions.estimated_document_count(skip=2) self.cmp.compare_exceptions.estimated_document_count(filter={'a': 1}) def test__reindex(self): self.cmp.compare.create_index('a') self.cmp.do.insert_one({'a': 1}) if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.reindex() return self.cmp.do.reindex() def test__find_one(self): self.cmp.do.insert_one({'_id': 'id1', 'name': 'new'}) self.cmp.compare.find_one({'_id': 'id1'}) self.cmp.do.insert_one({'_id': 'id2', 'name': 'another new'}) self.cmp.compare.find_one({'_id': 'id2'}, {'_id': 1}) self.cmp.compare.find_one('id2', {'_id': 1}) def test__find_one_no_args(self): self.cmp.do.insert_one({'_id': 'new_obj', 'field': 'value'}) self.cmp.compare.find_one() def test__find_by_attributes(self): id1 = ObjectId() self.cmp.do.insert_one({'_id': id1, 'name': 'new'}) self.cmp.do.insert_one({'name': 'another new'}) self.cmp.compare_ignore_order.sort_by(lambda doc: str(doc.get('name', str(doc)))).find() self.cmp.compare.find({'_id': id1}) def test__find_by_document(self): self.cmp.do.insert_one({'name': 'new', 'doc': {'key': 'val'}}) self.cmp.do.insert_one({'name': 'another new'}) self.cmp.do.insert_one({'name': 'new', 'doc': {'key': ['val']}}) self.cmp.do.insert_one({'name': 'new', 'doc': {'key': ['val', 'other val']}}) self.cmp.compare_ignore_order.find() self.cmp.compare.find({'doc': {'key': 'val'}}) self.cmp.compare.find({'doc': {'key': {'$eq': 'val'}}}) def test__find_by_empty_document(self): self.cmp.do.insert_one({'doc': {'data': 'val'}}) self.cmp.do.insert_one({'doc': {}}) self.cmp.do.insert_one({'doc': None}) self.cmp.compare.find({'doc': {}}) def test__find_by_attributes_return_fields(self): id1 = ObjectId() id2 = ObjectId() self.cmp.do.insert_one( {'_id': id1, 'name': 'new', 'someOtherProp': 2, 'nestedProp': {'a': 1}}) self.cmp.do.insert_one({'_id': id2, 'name': 'another new'}) self.cmp.compare_ignore_order.find({}, {'_id': 0}) # test exclusion of _id self.cmp.compare_ignore_order.find({}, {'_id': 1, 'someOtherProp': 1}) # test inclusion self.cmp.compare_ignore_order.find({}, {'_id': 0, 'someOtherProp': 0}) # test exclusion self.cmp.compare_ignore_order.find({}, {'_id': 0, 'someOtherProp': 1}) # test mixed _id:0 self.cmp.compare_ignore_order.find({}, {'someOtherProp': 0}) # test no _id, otherProp:0 self.cmp.compare_ignore_order.find({}, {'someOtherProp': 1}) # test no _id, otherProp:1 self.cmp.compare.find({'_id': id1}, {'_id': 0}) # test exclusion of _id self.cmp.compare.find({'_id': id1}, {'_id': 1, 'someOtherProp': 1}) # test inclusion self.cmp.compare.find({'_id': id1}, {'_id': 0, 'someOtherProp': 0}) # test exclusion # test mixed _id:0 self.cmp.compare.find({'_id': id1}, {'_id': 0, 'someOtherProp': 1}) # test no _id, otherProp:0 self.cmp.compare.find({'_id': id1}, {'someOtherProp': 0}) # test no _id, otherProp:1 self.cmp.compare.find({'_id': id1}, {'someOtherProp': 1}) def test__find_by_attributes_return_fields_elemMatch(self): id = ObjectId() self.cmp.do.insert_one({ '_id': id, 'owns': [ {'type': 'hat', 'color': 'black'}, {'type': 'hat', 'color': 'green'}, {'type': 't-shirt', 'color': 'black', 'size': 'small'}, {'type': 't-shirt', 'color': 'black'}, {'type': 't-shirt', 'color': 'white'} ], 'hat': 'red' }) elem = {'$elemMatch': {'type': 't-shirt', 'color': 'black'}} # test filtering on array field only self.cmp.compare.find({'_id': id}, {'owns': elem}) # test filtering on array field with inclusion self.cmp.compare.find({'_id': id}, {'owns': elem, 'hat': 1}) # test filtering on array field with exclusion self.cmp.compare.find({'_id': id}, {'owns': elem, 'hat': 0}) # test filtering on non array field self.cmp.compare.find({'_id': id}, {'hat': elem}) # test no match self.cmp.compare.find({'_id': id}, {'owns': {'$elemMatch': {'type': 'cap'}}}) def test__find_with_expr(self): self.cmp.do.insert_many([ {'_id': 1, 'a': [5]}, {'_id': 2, 'a': [1, 2, 3]}, {'_id': 3, 'a': []}, ]) self.cmp.compare.find({'$expr': {'$eq': [{'$size': ['$a']}, 1]}}) self.cmp.do.insert_one({'_id': 4}) self.cmp.compare_exceptions.find({'$expr': {'$eq': [{'$size': ['$a']}, 1]}}) def test_double_negation(self): self.cmp.do.insert_many([ {'_id': 1, 'a': 'some str'}, {'_id': 2, 'a': 'another str'}, {'_id': 3, 'a': []}, ]) self.cmp.compare.find({'a': {'$not': {'$not': {'$regex': '^some'}}}}) def test__size(self): id = ObjectId() self.cmp.do.insert_one({ '_id': id, 'l_string': 1, 'l_tuple': ['a', 'b'], 'null_field': None }) self.cmp.compare.find({'_id': id}) self.cmp.compare.find({'_id': id, 'l_string': {'$not': {'$size': 0}}}) self.cmp.compare.find({'_id': id, 'l_tuple': {'$size': 2}}) self.cmp.compare.find({'_id': id, 'missing_field': {'$size': 1}}) self.cmp.compare.find({'_id': id, 'null_field': {'$size': 1}}) def test__all_with_other_operators(self): objs = [{'list': ['a']}, {'list': ['a', 123]}, {'list': ['a', 123, 'xyz']}] self.cmp.do.insert_many(objs) self.cmp.compare.find({'list': {'$all': ['a'], '$size': 1}}) self.cmp.compare.find({'list': {'$all': ['a', 123], '$size': 2}}) self.cmp.compare.find({'list': {'$all': ['a', 123, 'xyz'], '$size': 3}}) self.cmp.compare.find({'list': {'$all': ['a'], '$size': 3}}) self.cmp.compare.find({'list': {'$all': ['a', 123], '$in': ['xyz']}}) self.cmp.compare.find({'list': {'$all': ['a', 123, 'xyz'], '$in': ['abcdef']}}) self.cmp.compare.find({'list': {'$all': ['a'], '$eq': ['a']}}) def test__regex_match_non_string(self): id = ObjectId() self.cmp.do.insert_one({ '_id': id, 'test': 1 }) self.cmp.compare.find({'_id': id, 'test': {'$regex': '1'}}) def test__regex_match_non_string_in_list(self): id = ObjectId() self.cmp.do.insert_one({ '_id': id, 'test': [3, 2, 1] }) self.cmp.compare.find({'_id': id, 'test': {'$regex': '1'}}) def test__find_by_dotted_attributes(self): """Test seaching with dot notation.""" green_bowler = { 'name': 'bob', 'hat': {'color': 'green', 'type': 'bowler'}} red_bowler = { 'name': 'sam', 'hat': {'color': 'red', 'type': 'bowler'}} self.cmp.do.insert_one(green_bowler) self.cmp.do.insert_one(red_bowler) self.cmp.compare_ignore_order.find() self.cmp.compare_ignore_order.find({'name': 'sam'}) self.cmp.compare_ignore_order.find({'hat.color': 'green'}) self.cmp.compare_ignore_order.find({'hat.type': 'bowler'}) self.cmp.compare.find({ 'hat.color': 'red', 'hat.type': 'bowler' }) self.cmp.compare.find({ 'name': 'bob', 'hat.color': 'red', 'hat.type': 'bowler' }) self.cmp.compare.find({'hat': 'a hat'}) self.cmp.compare.find({'hat.color.cat': 'red'}) def test__find_empty_array_field(self): # See #90 self.cmp.do.insert_one({'array_field': []}) self.cmp.compare.find({'array_field': []}) def test__find_non_empty_array_field(self): # See #90 self.cmp.do.insert_one({'array_field': [['abc']]}) self.cmp.do.insert_one({'array_field': ['def']}) self.cmp.compare.find({'array_field': ['abc']}) self.cmp.compare.find({'array_field': [['abc']]}) self.cmp.compare.find({'array_field': 'def'}) self.cmp.compare.find({'array_field': ['def']}) def test__find_by_objectid_in_list(self): # See #79 self.cmp.do.insert_one( {'_id': 'x', 'rel_id': [ObjectId('52d669dcad547f059424f783')]}) self.cmp.compare.find({'rel_id': ObjectId('52d669dcad547f059424f783')}) def test__find_subselect_in_list(self): # See #78 self.cmp.do.insert_one({'_id': 'some_id', 'a': [{'b': 1, 'c': 2}]}) self.cmp.compare.find_one({'a.b': 1}) def test__find_dict_in_nested_list(self): # See #539 self.cmp.do.insert_one({'a': {'b': [{'c': 1}]}}) self.cmp.compare.find({'a.b': {'c': 1}}) def test__find_by_regex_object(self): """Test searching with regular expression objects.""" bob = {'name': 'bob'} sam = {'name': 'sam'} self.cmp.do.insert_one(bob) self.cmp.do.insert_one(sam) self.cmp.compare_ignore_order.find() regex = re.compile('bob|sam') self.cmp.compare_ignore_order.find({'name': regex}) regex = re.compile('bob|notsam') self.cmp.compare_ignore_order.find({'name': regex}) self.cmp.compare_ignore_order.find({'name': {'$regex': regex}}) upper_regex = Regex('Bob') self.cmp.compare_ignore_order.find({'name': {'$regex': upper_regex}}) self.cmp.compare_ignore_order.find({'name': { '$regex': upper_regex, '$options': 'i', }}) self.cmp.compare_ignore_order.find({'name': { '$regex': upper_regex, '$options': 'I', }}) self.cmp.compare_ignore_order.find({'name': { '$regex': upper_regex, '$options': 'z', }}) def test__find_by_regex_string(self): """Test searching with regular expression string.""" bob = {'name': 'bob'} sam = {'name': 'sam'} self.cmp.do.insert_one(bob) self.cmp.do.insert_one(sam) self.cmp.compare_ignore_order.find() self.cmp.compare_ignore_order.find({'name': {'$regex': 'bob|sam'}}) self.cmp.compare_ignore_order.find({'name': {'$regex': 'bob|notsam'}}) self.cmp.compare_ignore_order.find({'name': {'$regex': 'Bob', '$options': 'i'}}) self.cmp.compare_ignore_order.find({'name': {'$regex': 'Bob', '$options': 'I'}}) self.cmp.compare_ignore_order.find({'name': {'$regex': 'Bob', '$options': 'z'}}) def test__find_in_array_by_regex_object(self): """Test searching inside array with regular expression object.""" bob = {'name': 'bob', 'text': ['abcd', 'cde']} sam = {'name': 'sam', 'text': ['bde']} self.cmp.do.insert_one(bob) self.cmp.do.insert_one(sam) regex = re.compile('^a') self.cmp.compare_ignore_order.find({'text': regex}) regex = re.compile('e$') self.cmp.compare_ignore_order.find({'text': regex}) regex = re.compile('bde|cde') self.cmp.compare_ignore_order.find({'text': regex}) def test__find_in_array_by_regex_string(self): """Test searching inside array with regular expression string""" bob = {'name': 'bob', 'text': ['abcd', 'cde']} sam = {'name': 'sam', 'text': ['bde']} self.cmp.do.insert_one(bob) self.cmp.do.insert_one(sam) self.cmp.compare_ignore_order.find({'text': {'$regex': '^a'}}) self.cmp.compare_ignore_order.find({'text': {'$regex': 'e$'}}) self.cmp.compare_ignore_order.find({'text': {'$regex': 'bcd|cde'}}) def test__find_by_regex_string_on_absent_field_dont_break(self): """Test searching on absent field with regular expression string dont break""" bob = {'name': 'bob'} sam = {'name': 'sam'} self.cmp.do.insert_one(bob) self.cmp.do.insert_one(sam) self.cmp.compare_ignore_order.find({'text': {'$regex': 'bob|sam'}}) def test__find_by_elemMatch(self): self.cmp.do.insert_one({'field': [{'a': 1, 'b': 2}, {'c': 3, 'd': 4}]}) self.cmp.do.insert_one({'field': [{'a': 1, 'b': 4}, {'c': 3, 'd': 8}]}) self.cmp.do.insert_one({'field': 'nonlist'}) self.cmp.do.insert_one({'field': 2}) self.cmp.compare.find({'field': {'$elemMatch': {'b': 1}}}) self.cmp.compare_ignore_order.find({'field': {'$elemMatch': {'a': 1}}}) self.cmp.compare.find({'field': {'$elemMatch': {'b': {'$gt': 3}}}}) def test__find_by_elemMatchDirectQuery(self): self.cmp.do.insert_many([ {'_id': 0, 'arr': [0, 1, 2, 3, 10]}, {'_id': 1, 'arr': [0, 2, 4, 6]}, {'_id': 2, 'arr': [1, 3, 5, 7]} ]) self.cmp.compare_ignore_order.find({'arr': {'$elemMatch': {'$lt': 10, '$gt': 4}}}) def test__find_in_array(self): self.cmp.do.insert_one({'field': [{'a': 1, 'b': 2}, {'c': 3, 'd': 4}]}) self.cmp.compare.find({'field.0.a': 1}) self.cmp.compare.find({'field.0.b': 2}) self.cmp.compare.find({'field.1.c': 3}) self.cmp.compare.find({'field.1.d': 4}) self.cmp.compare.find({'field.0': {'$exists': True}}) self.cmp.compare.find({'field.0': {'$exists': False}}) self.cmp.compare.find({'field.0.a': {'$exists': True}}) self.cmp.compare.find({'field.0.a': {'$exists': False}}) self.cmp.compare.find({'field.1.a': {'$exists': True}}) self.cmp.compare.find({'field.1.a': {'$exists': False}}) self.cmp.compare.find( {'field.0.a': {'$exists': True}, 'field.1.a': {'$exists': False}}) def test__find_in_array_equal_null(self): self.cmp.do.insert_many([ {'_id': 1, 'shape': [{'color': 'red'}]}, {'_id': 2, 'shape': [{'color': 'yellow'}]}, {'_id': 3, 'shape': [{'color': 'red'}, {'color': 'yellow'}]}, {'_id': 4, 'shape': [{'size': 3}]}, {'_id': 5}, {'_id': 6, 'shape': {'color': ['red', 'yellow']}}, {'_id': 7, 'shape': [{'color': 'red'}, {'color': None}]}, ]) self.cmp.compare_ignore_order.find({'shape.color': {'$eq': None}}) self.cmp.compare_ignore_order.find({'shape.color': None}) def test__find_notequal(self): """Test searching with operators other than equality.""" bob = {'_id': 1, 'name': 'bob'} sam = {'_id': 2, 'name': 'sam'} a_goat = {'_id': 3, 'goatness': 'very'} self.cmp.do.insert_many([bob, sam, a_goat]) self.cmp.compare_ignore_order.find() self.cmp.compare_ignore_order.find({'name': {'$ne': 'bob'}}) self.cmp.compare_ignore_order.find({'goatness': {'$ne': 'very'}}) self.cmp.compare_ignore_order.find({'goatness': {'$ne': 'not very'}}) self.cmp.compare_ignore_order.find({'snakeness': {'$ne': 'very'}}) def test__find_notequal_by_value(self): """Test searching for None.""" bob = {'_id': 1, 'name': 'bob', 'sheepness': {'sometimes': True}} sam = {'_id': 2, 'name': 'sam', 'sheepness': {'sometimes': True}} a_goat = {'_id': 3, 'goatness': 'very', 'sheepness': {}} self.cmp.do.insert_many([bob, sam, a_goat]) self.cmp.compare_ignore_order.find({'goatness': None}) self.cmp.compare_ignore_order.find({'sheepness.sometimes': None}) def test__find_not(self): bob = {'_id': 1, 'name': 'bob'} sam = {'_id': 2, 'name': 'sam'} self.cmp.do.insert_many([bob, sam]) self.cmp.compare_ignore_order.find() self.cmp.compare_ignore_order.find({'name': {'$not': {'$ne': 'bob'}}}) self.cmp.compare_ignore_order.find({'name': {'$not': {'$ne': 'sam'}}}) self.cmp.compare_ignore_order.find({'name': {'$not': {'$ne': 'dan'}}}) self.cmp.compare_ignore_order.find({'name': {'$not': {'$eq': 'bob'}}}) self.cmp.compare_ignore_order.find({'name': {'$not': {'$eq': 'sam'}}}) self.cmp.compare_ignore_order.find({'name': {'$not': {'$eq': 'dan'}}}) self.cmp.compare_ignore_order.find({'name': {'$not': re.compile('dan')}}) self.cmp.compare_ignore_order.find({'name': {'$not': Regex('dan')}}) def test__find_not_exceptions(self): # pylint: disable=expression-not-assigned self.cmp.do.insert_one(dict(noise='longhorn')) with self.assertRaises(OperationFailure): self.mongo_collection.find({'name': {'$not': True}})[0] with self.assertRaises(OperationFailure): self.fake_collection.find({'name': {'$not': True}})[0] with self.assertRaises(OperationFailure): self.mongo_collection.find({'name': {'$not': []}})[0] with self.assertRaises(OperationFailure): self.fake_collection.find({'name': {'$not': []}})[0] with self.assertRaises(OperationFailure): self.mongo_collection.find({'name': {'$not': ''}})[0] with self.assertRaises(OperationFailure): self.fake_collection.find({'name': {'$not': ''}})[0] def test__find_compare(self): self.cmp.do.insert_one(dict(noise='longhorn', sqrd='non numeric')) for x in range(10): self.cmp.do.insert_one(dict(num=x, sqrd=x * x)) self.cmp.compare_ignore_order.find({'sqrd': {'$lte': 4}}) self.cmp.compare_ignore_order.find({'sqrd': {'$lt': 4}}) self.cmp.compare_ignore_order.find({'sqrd': {'$gte': 64}}) self.cmp.compare_ignore_order.find({'sqrd': {'$gte': 25, '$lte': 36}}) def test__find_compare_objects(self): self.cmp.do.insert_many([ {'_id': 1, 'counts': {'circles': 3}}, {'_id': 2, 'counts': {'squares': 0}}, {'_id': 3, 'counts': {'arrows': 15}}, {'_id': 4, 'counts': {'circles': 1}}, {'_id': 5, 'counts': OrderedDict([ ('circles', 1), ('arrows', 15), ])}, {'_id': 6, 'counts': OrderedDict([ ('arrows', 15), ('circles', 1), ])}, {'_id': 7}, {'_id': 8, 'counts': {}}, {'_id': 9, 'counts': {'circles': 'three'}}, {'_id': 10, 'counts': {'circles': None}}, {'_id': 11, 'counts': {'circles': b'bytes'}}, ]) self.cmp.compare_ignore_order.find({'counts': {'$gt': {'circles': 1}}}) def test__find_compare_nested_objects(self): self.cmp.do.insert_many([ {'_id': 1, 'counts': {'circles': {'blue': 3}}}, {'_id': 2, 'counts': {'squares': 0}}, {'_id': 3, 'counts': {'arrows': {'blue': 2}}}, {'_id': 4, 'counts': {'circles': {}}}, {'_id': 5, 'counts': {'arrows': True}}, ]) self.cmp.compare_ignore_order.find( {'counts': {'$gt': {'circles': {'blue': 1}}}}) def test__find_sets(self): single = 4 even = [2, 4, 6, 8] prime = [2, 3, 5, 7] self.cmp.do.insert_many([ dict(x=single), dict(x=even), dict(x=prime), dict()]) self.cmp.compare_ignore_order.find({'x': {'$in': [7, 8]}}) self.cmp.compare_ignore_order.find({'x': {'$in': [4, 5]}}) self.cmp.compare_ignore_order.find({'x': {'$in': [4, None]}}) self.cmp.compare_ignore_order.find({'x': {'$nin': [2, 5]}}) self.cmp.compare_ignore_order.find({'x': {'$all': [2, 5]}}) self.cmp.compare_ignore_order.find({'x': {'$all': [7, 8]}}) self.cmp.compare_ignore_order.find({'x': 2}) self.cmp.compare_ignore_order.find({'x': 4}) self.cmp.compare_ignore_order.find({'$or': [{'x': 4}, {'x': 2}]}) self.cmp.compare_ignore_order.find({'$or': [{'x': 4}, {'x': 7}]}) self.cmp.compare_ignore_order.find({'$and': [{'x': 2}, {'x': 7}]}) self.cmp.compare_ignore_order.find({'$nor': [{'x': 3}]}) self.cmp.compare_ignore_order.find({'$nor': [{'x': 4}, {'x': 2}]}) def test__find_operators_in_list(self): self.cmp.do.insert_many([ dict(x=4), dict(x=[300, 500, 4]), dict(x=[1200, 300, 1400])]) self.cmp.compare_ignore_order.find({'x': {'$gte': 1100, '$lte': 1250}}) self.cmp.compare_ignore_order.find({'x': {'$gt': 300, '$lt': 400}}) def test__find_sets_regex(self): self.cmp.do.insert_many([ {'x': '123'}, {'x': ['abc', 'abd']}, ]) digits_pat = re.compile(r'^\d+') str_pat = re.compile(r'^ab[cd]') non_existing_pat = re.compile(r'^lll') self.cmp.compare_ignore_order.find({'x': {'$in': [digits_pat]}}) self.cmp.compare_ignore_order.find({'x': {'$in': [str_pat]}}) self.cmp.compare_ignore_order.find({'x': {'$in': [non_existing_pat]}}) self.cmp.compare_ignore_order.find({'x': {'$in': [non_existing_pat, '123']}}) self.cmp.compare_ignore_order.find({'x': {'$nin': [str_pat]}}) self.cmp.compare_ignore_order.find({'x': {'$nin': [non_existing_pat]}}) def test__find_negative_matches(self): self.cmp.do.insert_many([ {'_id': 1, 'shape': [{'color': 'red'}]}, {'_id': 2, 'shape': [{'color': 'yellow'}]}, {'_id': 3, 'shape': [{'color': 'red'}, {'color': 'yellow'}]}, {'_id': 4, 'shape': [{'size': 3}]}, {'_id': 5}, {'_id': 6, 'shape': {'color': ['red', 'yellow']}}, {'_id': 7, 'shape': {'color': 'red'}}, {'_id': 8, 'shape': {'color': ['blue', 'yellow']}}, {'_id': 9, 'shape': {'color': ['red']}}, ]) self.cmp.compare_ignore_order.find({'shape.color': {'$ne': 'red'}}) self.cmp.compare_ignore_order.find({'shape.color': {'$ne': ['red']}}) self.cmp.compare_ignore_order.find({'shape.color': {'$nin': ['blue', 'red']}}) def test__find_ne_multiple_keys(self): self.cmp.do.insert_many([ {'_id': 1, 'cases': [{'total': 1}]}, {'_id': 2, 'cases': [{'total': 2}]}, {'_id': 3, 'cases': [{'total': 3}]}, {'_id': 4, 'cases': []}, {'_id': 5}, ]) self.cmp.compare_ignore_order.find({'cases.total': {'$gt': 1, '$ne': 3}}) self.cmp.compare_ignore_order.find({'cases.total': {'$gt': 1, '$nin': [1, 3]}}) def test__find_and_modify_remove(self): self.cmp.do.insert_many([{'a': x, 'junk': True} for x in range(10)]) if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.find_and_modify( {'a': 2}, remove=True, fields={'_id': False, 'a': True}) return self.cmp.compare.find_and_modify({'a': 2}, remove=True, fields={'_id': False, 'a': True}) self.cmp.compare_ignore_order.find() def test__find_one_and_delete(self): self.cmp.do.insert_many([{'a': i} for i in range(10)]) self.cmp.compare.find_one_and_delete({'a': 5}, {'_id': False}) self.cmp.compare.find() def test__find_one_and_replace(self): self.cmp.do.insert_many([{'a': i} for i in range(10)]) self.cmp.compare.find_one_and_replace( {'a': 5}, {'a': 11}, projection={'_id': False}) self.cmp.compare.find() def test__find_one_and_update(self): self.cmp.do.insert_many([{'a': i} for i in range(10)]) self.cmp.compare.find_one_and_update( {'a': 5}, {'$set': {'a': 11}}, projection={'_id': False}) self.cmp.compare.find() def test__find_sort_list(self): self.cmp.do.delete_many({}) for data in ({'a': 1, 'b': 3, 'c': 'data1'}, {'a': 2, 'b': 2, 'c': 'data3'}, {'a': 3, 'b': 1, 'c': 'data2'}): self.cmp.do.insert_one(data) self.cmp.compare.find(sort=[('a', 1), ('b', -1)]) self.cmp.compare.find(sort=[('b', 1), ('a', -1)]) self.cmp.compare.find(sort=[('b', 1), ('a', -1), ('c', 1)]) def test__find_sort_list_empty_order(self): self.cmp.do.delete_many({}) for data in ({'a': 1}, {'a': 2, 'b': -2}, {'a': 3, 'b': 4}, {'a': 4, 'b': b'bin1'}, {'a': 4, 'b': b'bin2'}, {'a': 4, 'b': b'alongbin1'}, {'a': 4, 'b': b'alongbin2'}, {'a': 4, 'b': b'zlongbin1'}, {'a': 4, 'b': b'zlongbin2'}): self.cmp.do.insert_one(data) self.cmp.compare.find(sort=[('b', 1)]) self.cmp.compare.find(sort=[('b', -1)]) def test__find_sort_list_nested_doc(self): self.cmp.do.delete_many({}) for data in ({'root': {'a': 1, 'b': 3, 'c': 'data1'}}, {'root': {'a': 2, 'b': 2, 'c': 'data3'}}, {'root': {'a': 3, 'b': 1, 'c': 'data2'}}): self.cmp.do.insert_one(data) self.cmp.compare.find(sort=[('root.a', 1), ('root.b', -1)]) self.cmp.compare.find(sort=[('root.b', 1), ('root.a', -1)]) self.cmp.compare.find( sort=[ ('root.b', 1), ('root.a', -1), ('root.c', 1)]) def test__find_sort_list_nested_list(self): self.cmp.do.delete_many({}) for data in ({'root': [{'a': 1, 'b': 3, 'c': 'data1'}]}, {'root': [{'a': 2, 'b': 2, 'c': 'data3'}]}, {'root': [{'a': 3, 'b': 1, 'c': 'data2'}]}): self.cmp.do.insert_one(data) self.cmp.compare.find(sort=[('root.0.a', 1), ('root.0.b', -1)]) self.cmp.compare.find(sort=[('root.0.b', 1), ('root.0.a', -1)]) self.cmp.compare.find( sort=[ ('root.0.b', 1), ('root.0.a', -1), ('root.0.c', 1)]) def test__find_limit(self): self.cmp.do.delete_many({}) for data in ({'a': 1, 'b': 3, 'c': 'data1'}, {'a': 2, 'b': 2, 'c': 'data3'}, {'a': 3, 'b': 1, 'c': 'data2'}): self.cmp.do.insert_one(data) self.cmp.compare.find(limit=2, sort=[('a', 1), ('b', -1)]) # pymongo limit defaults to 0, returning everything self.cmp.compare.find(limit=0, sort=[('a', 1), ('b', -1)]) def test__find_projection_subdocument_lists(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'a': 1, 'b': [{'c': 3, 'd': 4}, {'c': 5, 'd': 6}]}) self.cmp.compare.find_one({'a': 1}, {'_id': 0, 'a': 1, 'b': 1}) self.cmp.compare_exceptions.find_one( {'a': 1}, OrderedDict([('_id', 0), ('a', 1), ('b', 1), ('b.c', 1)])) self.cmp.compare_exceptions.find_one( {'a': 1}, OrderedDict([('_id', 0), ('a', 1), ('b.c', 1), ('b', 1)])) self.cmp.compare.find_one({'a': 1}, {'_id': 0, 'a': 1, 'b.c': 1}) self.cmp.compare.find_one({'a': 1}, {'_id': 0, 'a': 0, 'b.c': 0}) self.cmp.compare.find_one({'a': 1}, {'_id': 0, 'a': 1, 'b.c.e': 1}) self.cmp.compare_exceptions.find_one( {'a': 1}, OrderedDict([('_id', 0), ('a', 0), ('b.c', 0), ('b.c.e', 0)])) # This one is not implemented in mongmock yet. # self.cmp.compare.find_one( # {'a': 1}, OrderedDict([('_id', 0), ('a', 0), ('b.c.e', 0), ('b.c', 0)])) def test__find_type(self): supported_types = ( 'double', 'string', 'object', 'array', 'binData', 'objectId', 'bool', 'date', 'int', 'long', 'decimal', 'number', ) self.cmp.do.insert_many([ {'a': 1.2}, # double {'a': 'a string value'}, # string {'a': {'b': 1}}, # object {'a': [1, 2, 3]}, # array or int {'a': b'hello'}, # binData {'a': ObjectId()}, # objectId {'a': True}, # bool {'a': datetime.datetime.now()}, # date {'a': 1}, # int {'a': 1 << 32}, # long {'a': decimal128.Decimal128('1.1')}, # decimal ]) for type_name in supported_types: self.cmp.compare.find({'a': {'$type': type_name}}) @skipIf(sys.version_info < (3, 7), 'Older versions of Python cannot copy regex partterns') @skipIf( helpers.PYMONGO_VERSION >= version.parse('4.0'), 'pymongo v4 or above do not specify uuid encoding') def test__sort_mixed_types(self): self.cmp.do.insert_many([ {'type': 'bool', 'a': True}, {'type': 'datetime', 'a': datetime.datetime.now()}, {'type': 'dict', 'a': {'a': 1}}, {'type': 'emptyList', 'a': []}, {'type': 'int', 'a': 1}, {'type': 'listOfList', 'a': [[1, 2], [3, 4]]}, {'type': 'missing'}, {'type': 'None', 'a': None}, {'type': 'ObjectId', 'a': ObjectId()}, {'type': 'regex', 'a': re.compile('a')}, {'type': 'repeatedInt', 'a': [1, 2]}, {'type': 'string', 'a': 'a'}, {'type': 'tupleOfTuple', 'a': ((1, 2), (3, 4))}, {'type': 'uuid', 'a': uuid.UUID(int=3)}, {'type': 'DBRef', 'a': DBRef('a', 'a', 'db_name')} ]) self.cmp.compare.find({}, sort=[('a', 1), ('type', 1)]) @skipIf( helpers.PYMONGO_VERSION >= version.parse('4.0'), 'pymongo v4 or above do not specify uuid encoding') def test__find_sort_uuid(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'_id': uuid.UUID(int=3), 'timestamp': 99, 'a': 1}, {'_id': uuid.UUID(int=1), 'timestamp': 100, 'a': 3}, {'_id': uuid.UUID(int=2), 'timestamp': 100, 'a': 2}, ]) self.cmp.compare.find({}, sort=[('timestamp', 1), ('_id', 1)]) @skipIf( helpers.PYMONGO_VERSION < version.parse('4.0'), 'old version of pymongo accepts to encode uuid') def test__fail_at_uuid_encoding(self): self.cmp.compare_exceptions.insert_one({'_id': uuid.UUID(int=2)}) def test__find_all(self): self.cmp.do.insert_many([ { 'code': 'ijk', 'tags': ['electronics', 'school'], 'qty': [{'size': 'M', 'num': 100, 'color': 'green'}], }, { 'code': 'efg', 'tags': ['school', 'book'], 'qty': [ {'size': 'S', 'num': 10, 'color': 'blue'}, {'size': 'M', 'num': 100, 'color': 'blue'}, {'size': 'L', 'num': 100, 'color': 'green'}, ], }, ]) self.cmp.compare.find({'qty.size': {'$all': ['M', 'L']}}) # def test__as_class(self): # class MyDict(dict): # pass # # self.cmp.do.delete_many({}) # self.cmp.do.insert_one( # {'a': 1, 'b': {'ba': 3, 'bb': 4, 'bc': [{'bca': 5}]}}) # self.cmp.compare.find({}, as_class=MyDict) # self.cmp.compare.find({'a': 1}, as_class=MyDict) def test__return_only_selected_fields(self): self.cmp.do.insert_one({'name': 'Chucky', 'type': 'doll', 'model': 'v6'}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection=['type']) def test__return_only_selected_fields_no_id(self): self.cmp.do.insert_one({'name': 'Chucky', 'type': 'doll', 'model': 'v6'}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection={'type': 1, '_id': 0}) def test__return_only_selected_fields_nested_field_found(self): self.cmp.do.insert_one( {'name': 'Chucky', 'properties': {'type': 'doll', 'model': 'v6'}}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection=['properties.type']) def test__return_only_selected_fields_nested_field_not_found(self): self.cmp.do.insert_one( {'name': 'Chucky', 'properties': {'type': 'doll', 'model': 'v6'}}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection=['properties.color']) def test__return_only_selected_fields_nested_field_found_no_id(self): self.cmp.do.insert_one( {'name': 'Chucky', 'properties': {'type': 'doll', 'model': 'v6'}}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection={'properties.type': 1, '_id': 0}) def test__return_only_selected_fields_nested_field_not_found_no_id(self): self.cmp.do.insert_one( {'name': 'Chucky', 'properties': {'type': 'doll', 'model': 'v6'}}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection={'properties.color': 1, '_id': 0}) def test__exclude_selected_fields(self): self.cmp.do.insert_one({'name': 'Chucky', 'type': 'doll', 'model': 'v6'}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection={'type': 0}) def test__exclude_selected_fields_including_id(self): self.cmp.do.insert_one({'name': 'Chucky', 'type': 'doll', 'model': 'v6'}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection={'type': 0, '_id': 0}) def test__exclude_all_fields_including_id(self): self.cmp.do.insert_one({'name': 'Chucky', 'type': 'doll'}) self.cmp.compare.find( {'name': 'Chucky'}, projection={'type': 0, '_id': 0, 'name': 0}) def test__exclude_selected_nested_fields(self): self.cmp.do.insert_one( {'name': 'Chucky', 'properties': {'type': 'doll', 'model': 'v6'}}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection={'properties.type': 0}) def test__exclude_all_selected_nested_fields(self): self.cmp.do.insert_one( {'name': 'Chucky', 'properties': {'type': 'doll', 'model': 'v6'}}) self.cmp.compare_ignore_order.find( {'name': 'Chucky'}, projection={'properties.type': 0, 'properties.model': 0}) def test__default_fields_if_projection_empty(self): self.cmp.do.insert_one({'name': 'Chucky', 'type': 'doll', 'model': 'v6'}) self.cmp.compare_ignore_order.find({'name': 'Chucky'}, projection=[]) def test__projection_slice_int_first(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare.find({'name': 'Array'}, projection={'name': 1, 'values': {'$slice': 1}}) def test__projection_slice_int_last(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare.find({'name': 'Array'}, projection={'name': 1, 'values': {'$slice': -1}}) def test__projection_slice_list_pos(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare.find({'name': 'Array'}, projection={ 'name': 1, 'values': {'$slice': [3, 1]}}) def test__projection_slice_list_neg(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare.find({'name': 'Array'}, projection={ 'name': 1, 'values': {'$slice': [-3, 1]}}) def test__projection_slice_list_pos_to_end(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare.find({'name': 'Array'}, projection={ 'name': 1, 'values': {'$slice': [3, 10]}}) def test__projection_slice_list_neg_to_end(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare.find({'name': 'Array'}, projection={ 'name': 1, 'values': {'$slice': [-3, 10]}}) def test__projection_slice_list_select_subfield(self): self.cmp.do.insert_one({'name': 'Array', 'values': [ {'num': 0, 'val': 1}, {'num': 1, 'val': 2}]}) self.cmp.compare_exceptions.find({'name': 'Array'}, projection={ 'values.num': 1, 'values': {'$slice': 1}}) def test__projection_slice_list_wrong_num_slice(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare_exceptions.find({'name': 'Array'}, projection={ 'name': 1, 'values': {'$slice': [-3, 10, 1]}}) def test__projection_slice_list_wrong_slice_type(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare_exceptions.find({'name': 'Array'}, projection={ 'name': 1, 'values': {'$slice': [1.0]}}) def test__projection_slice_list_wrong_slice_value_type(self): self.cmp.do.insert_one({'name': 'Array', 'values': [0, 1, 2, 3, 4, 5, 6, 7]}) self.cmp.compare_exceptions.find({'name': 'Array'}, projection={ 'name': 1, 'values': {'$slice': '3'}}) def test__projection_slice_list_wrong_value_type(self): self.cmp.do.insert_one({'name': 'Array', 'values': 0}) self.cmp.compare_exceptions.find({'name': 'Array'}, projection={ 'name': 1, 'values': {'$slice': 1}}) def test__remove(self): """Test the remove method.""" self.cmp.do.insert_one({'value': 1}) self.cmp.compare_ignore_order.find() if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.remove() return self.cmp.do.remove() self.cmp.compare.find() self.cmp.do.insert_many([ {'name': 'bob'}, {'name': 'sam'}, ]) self.cmp.compare_ignore_order.find() self.cmp.do.remove({'name': 'bob'}) self.cmp.compare_ignore_order.find() self.cmp.do.remove({'name': 'notsam'}) self.cmp.compare.find() self.cmp.do.remove({'name': 'sam'}) self.cmp.compare.find() def test__delete_one(self): self.cmp.do.insert_many([{'a': i} for i in range(10)]) self.cmp.compare.find() self.cmp.do.delete_one({'a': 5}) self.cmp.compare.find() def test__delete_many(self): self.cmp.do.insert_many([{'a': i} for i in range(10)]) self.cmp.compare.find() self.cmp.do.delete_many({'a': {'$gt': 5}}) self.cmp.compare.find() def test__update(self): doc = {'a': 1} self.cmp.do.insert_one(doc) new_document = {'new_attr': 2} if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.update({'a': 1}, new_document) return self.cmp.do.update({'a': 1}, new_document) self.cmp.compare_ignore_order.find() @skipIf(helpers.PYMONGO_VERSION >= version.parse('4.0'), 'pymongo v4 or above dropped update') def test__update_upsert_with_id(self): self.cmp.do.update( {'a': 1}, {'_id': ObjectId('52d669dcad547f059424f783'), 'a': 1}, upsert=True) self.cmp.compare.find() def test__update_with_zero_id(self): self.cmp.do.insert_one({'_id': 0}) self.cmp.do.replace_one({'_id': 0}, {'a': 1}) self.cmp.compare.find() def test__update_upsert_with_dots(self): self.cmp.do.update_one( {'a.b': 1}, {'$set': {'c': 2}}, upsert=True) self.cmp.compare.find() def test__update_upsert_with_operators(self): self.cmp.do.update_one( {'$or': [{'name': 'billy'}, {'name': 'Billy'}]}, {'$set': {'name': 'Billy', 'age': 5}}, upsert=True) self.cmp.compare.find() self.cmp.do.update_one({'a.b': {'$eq': 1}, 'd': {}}, {'$set': {'c': 2}}, upsert=True) self.cmp.compare.find() def test__update_upsert_with_matched_subdocuments(self): self.cmp.do.update_one( {'b.c.': 1, 'b.d': 3}, {'$set': {'a': 1}}, upsert=True) self.cmp.compare.find() @skipIf(helpers.PYMONGO_VERSION >= version.parse('4.0'), 'pymongo v4 or above dropped update') def test__update_with_empty_document_comes(self): """Tests calling update_one with just '{}' for replacing whole document""" self.cmp.do.insert_one({'name': 'bob', 'hat': 'wide'}) self.cmp.do.update({'name': 'bob'}, {}) self.cmp.compare.find() def test__update_one(self): self.cmp.do.insert_many([{'a': 1, 'b': 0}, {'a': 2, 'b': 0}]) self.cmp.compare.find() self.cmp.do.update_one({'a': 2}, {'$set': {'b': 1}}) self.cmp.compare.find() self.cmp.do.update_one({'a': 3}, {'$set': {'a': 3, 'b': 0}}) self.cmp.compare.find() self.cmp.do.update_one({'a': 3}, {'$set': {'a': 3, 'b': 0}}, upsert=True) self.cmp.compare.find() self.cmp.compare_exceptions.update_one({}, {'$set': {}}) self.cmp.compare_exceptions.update_one({'a': 'does-not-exist'}, {'$set': {}}) self.cmp.compare_exceptions.update_one({'a': 'does-not-exist'}, {'$set': {}}, upsert=True) def test__update_many(self): self.cmp.do.insert_many([{'a': 1, 'b': 0}, {'a': 2, 'b': 0}]) self.cmp.compare.find() self.cmp.do.update_many({'b': 1}, {'$set': {'b': 1}}) self.cmp.compare.find() self.cmp.do.update_many({'b': 0}, {'$set': {'b': 1}}) self.cmp.compare.find() def test__replace_one(self): self.cmp.do.insert_many([{'a': 1, 'b': 0}, {'a': 2, 'b': 0}]) self.cmp.compare.find() self.cmp.do.replace_one({'a': 2}, {'a': 3, 'b': 0}) self.cmp.compare.find() self.cmp.do.replace_one({'a': 4}, {'a': 4, 'b': 0}) self.cmp.compare.find() self.cmp.do.replace_one({'a': 4}, {'a': 4, 'b': 0}, upsert=True) self.cmp.compare.find() def test__set(self): """Tests calling update with $set members.""" self.cmp.do.update_one( {'_id': 42}, {'$set': {'some': 'thing'}}, upsert=True) self.cmp.compare.find({'_id': 42}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_one({'name': 'bob'}, {'$set': {'hat': 'green'}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_one({'name': 'bob'}, {'$set': {'hat': 'red'}}) self.cmp.compare.find({'name': 'bob'}) def test__unset(self): """Tests calling update with $unset members.""" self.cmp.do.update_many({'name': 'bob'}, {'$set': {'a': 'aaa'}}, upsert=True) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$unset': {'a': 0}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$set': {'a': 'aaa'}}, upsert=True) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$unset': {'a': 1}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$set': {'a': 'aaa'}}, upsert=True) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$unset': {'a': ''}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$set': {'a': 'aaa'}}, upsert=True) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$unset': {'a': True}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$set': {'a': 'aaa'}}, upsert=True) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$unset': {'a': False}}) self.cmp.compare.find({'name': 'bob'}) def test__unset_nested(self): self.cmp.do.update_many({'_id': 1}, {'$set': {'a': {'b': 1, 'c': 2}}}, upsert=True) self.cmp.do.update_many({'_id': 1}, {'$unset': {'a.b': True}}) self.cmp.compare.find() self.cmp.do.update_many({'_id': 1}, {'$set': {'a': {'b': 1, 'c': 2}}}, upsert=True) self.cmp.do.update_many({'_id': 1}, {'$unset': {'a.b': False}}) self.cmp.compare.find() self.cmp.do.update_many({'_id': 1}, {'$set': {'a': {'b': 1}}}, upsert=True) self.cmp.do.update_many({'_id': 1}, {'$unset': {'a.b': True}}) self.cmp.compare.find() self.cmp.do.update_many({'_id': 1}, {'$set': {'a': {'b': 1}}}, upsert=True) self.cmp.do.update_many({'_id': 1}, {'$unset': {'a.b': False}}) self.cmp.compare.find() def test__unset_positional(self): self.cmp.do.insert_one({'a': 1, 'b': [{'c': 2, 'd': 3}]}) self.cmp.do.update_many( {'a': 1, 'b': {'$elemMatch': {'c': 2, 'd': 3}}}, {'$unset': {'b.$.c': ''}} ) self.cmp.compare.find() def test__set_upsert(self): self.cmp.do.delete_many({}) self.cmp.do.update_many({'name': 'bob'}, {'$set': {'age': 1}}, True) self.cmp.compare.find() self.cmp.do.update_many({'name': 'alice'}, {'$set': {'age': 1}}, True) self.cmp.compare_ignore_order.find() def test__set_subdocument_array(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': [0, 0]}) self.cmp.do.insert_one({'name': 'bob', 'some_field': 'B', 'data': [0, 0]}) self.cmp.do.update_many({'name': 'bob'}, {'$set': {'some_field': 'A', 'data.1': 3}}) self.cmp.compare.find() def test__set_subdocument_array_bad_index_after_dot(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'some_field': 'B', 'data': [0, 0]}) self.cmp.do.update_many({'name': 'bob'}, {'$set': {'some_field': 'A', 'data.3': 1}}) self.cmp.compare.find() def test__set_subdocument_array_bad_neg_index_after_dot(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'some_field': 'B', 'data': [0, 0]}) self.cmp.compare_exceptions.update_many({'name': 'bob'}, {'$set': {'data.-3': 1}}) def test__set_subdocuments_positional(self): self.cmp.do.insert_one({'name': 'bob', 'subdocs': [ {'id': 1, 'name': 'foo'}, {'id': 2, 'name': 'bar'} ]}) self.cmp.do.update_many( {'name': 'bob', 'subdocs.id': 2}, {'$set': {'subdocs.$': {'id': 3, 'name': 'baz'}}}) self.cmp.compare.find() def test__inc(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) for _ in range(3): self.cmp.do.update_many({'name': 'bob'}, {'$inc': {'count': 1}}) self.cmp.compare.find({'name': 'bob'}) def test__max(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) for i in range(3): self.cmp.do.update_many({'name': 'bob'}, {'$max': {'count': i}}) self.cmp.compare.find({'name': 'bob'}) def test__min(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) for i in range(3): self.cmp.do.update_many({'name': 'bob'}, {'$min': {'count': i}}) self.cmp.compare.find({'name': 'bob'}) def test__inc_upsert(self): self.cmp.do.delete_many({}) for _ in range(3): self.cmp.do.update_many({'name': 'bob'}, {'$inc': {'count': 1}}, True) self.cmp.compare.find({'name': 'bob'}) def test__inc_subdocument(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': {'age': 0}}) self.cmp.do.update_many({'name': 'bob'}, {'$inc': {'data.age': 1}}) self.cmp.compare.find() self.cmp.do.update_many({'name': 'bob'}, {'$inc': {'data.age2': 1}}) self.cmp.compare.find() def test__inc_subdocument_array(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': [0, 0]}) self.cmp.do.update_many({'name': 'bob'}, {'$inc': {'data.1': 1}}) self.cmp.compare.find() self.cmp.do.update_many({'name': 'bob'}, {'$inc': {'data.1': 1}}) self.cmp.compare.find() def test__inc_subdocument_array_bad_index_after_dot(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': [0, 0]}) self.cmp.do.update_many({'name': 'bob'}, {'$inc': {'data.3': 1}}) self.cmp.compare.find() def test__inc_subdocument_array_bad_neg_index_after_dot(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': [0, 0]}) self.cmp.compare_exceptions.update_many({'name': 'bob'}, {'$inc': {'data.-3': 1}}) def test__inc_subdocument_positional(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': [{'age': 0}, {'age': 1}]}) self.cmp.do.update_many( {'name': 'bob', 'data': {'$elemMatch': {'age': 0}}}, {'$inc': {'data.$.age': 1}}) self.cmp.compare.find() def test__setOnInsert(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$setOnInsert': {'age': 1}}) self.cmp.compare.find() self.cmp.do.update_many({'name': 'ann'}, {'$setOnInsert': {'age': 1}}) self.cmp.compare.find() def test__setOnInsert_upsert(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$setOnInsert': {'age': 1}}, True) self.cmp.compare.find() self.cmp.do.update_many({'name': 'ann'}, {'$setOnInsert': {'age': 1}}, True) self.cmp.compare.find() def test__setOnInsert_subdocument(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': {'age': 0}}) self.cmp.do.update_many({'name': 'bob'}, {'$setOnInsert': {'data.age': 1}}) self.cmp.compare.find() self.cmp.do.update_many({'name': 'bob'}, {'$setOnInsert': {'data.age1': 1}}) self.cmp.compare.find() self.cmp.do.update_many({'name': 'ann'}, {'$setOnInsert': {'data.age': 1}}) self.cmp.compare.find() def test__setOnInsert_subdocument_upsert(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': {'age': 0}}) self.cmp.do.update_many( {'name': 'bob'}, {'$setOnInsert': {'data.age': 1}}, True) self.cmp.compare.find() self.cmp.do.update_many( {'name': 'bob'}, {'$setOnInsert': {'data.age1': 1}}, True) self.cmp.compare.find() self.cmp.do.update_many( {'name': 'ann'}, {'$setOnInsert': {'data.age': 1}}, True) self.cmp.compare.find() def test__setOnInsert_subdocument_elemMatch(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': [{'age': 0}, {'age': 1}]}) self.cmp.do.update_many( {'name': 'bob', 'data': {'$elemMatch': {'age': 0}}}, {'$setOnInsert': {'data.$.age': 1}}) self.cmp.compare.find() def test__inc_subdocument_positional_upsert(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'data': [{'age': 0}, {'age': 1}]}) self.cmp.do.update_many( {'name': 'bob', 'data': {'$elemMatch': {'age': 0}}}, {'$setOnInsert': {'data.$.age': 1}}, True) self.cmp.compare.find() def test__set_dollar_operand(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'recordId': 1234, 'app': [ {'application': 'AppName', 'code': 1234, 'property': 'oldValue'}, {'application': 'AppName1', 'code': 1235, 'property': 'oldValue1'}]}) self.cmp.do.update_many( {'app': {'$elemMatch': {'application': 'AppName', 'code': 1234}}}, {'$set': {'app.$': {'application': 'AppName', 'code': 1234, 'property': 'newValue'}}}) def test__addToSet(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) for _ in range(3): self.cmp.do.update_many({'name': 'bob'}, {'$addToSet': {'hat': 'green'}}) self.cmp.compare.find({'name': 'bob'}) for _ in range(3): self.cmp.do.update_many({'name': 'bob'}, {'$addToSet': {'hat': 'tall'}}) self.cmp.compare.find({'name': 'bob'}) def test__addToSet_nested(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) for _ in range(3): self.cmp.do.update_many( {'name': 'bob'}, {'$addToSet': {'hat.color': 'green'}}) self.cmp.compare.find({'name': 'bob'}) for _ in range(3): self.cmp.do.update_many( {'name': 'bob'}, {'$addToSet': {'hat.color': 'tall'}}) self.cmp.compare.find({'name': 'bob'}) def test__addToSet_each(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) for _ in range(3): self.cmp.do.update_many( {'name': 'bob'}, {'$addToSet': {'hat': {'$each': ['green', 'yellow']}}}) self.cmp.compare.find({'name': 'bob'}) for _ in range(3): self.cmp.do.update_many( {'name': 'bob'}, {'$addToSet': {'shirt.color': {'$each': ['green', 'yellow']}}}) self.cmp.compare.find({'name': 'bob'}) def test__addToSet_dollar_operand(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'takes': [{'a': 2, 'tags': []}, {'a': 1, 'tags': [2]}]}) self.cmp.do.update_many( {'takes': {'$elemMatch': {'a': 1}}}, {'$addToSet': {'takes.$.tags': 3}}) def test__pop(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': ['green', 'tall']}) self.cmp.do.update_many({'name': 'bob'}, {'$pop': {'hat': 1}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': ['green', 'tall']}) self.cmp.do.update_many({'name': 'bob'}, {'$pop': {'hat': -1}}) self.cmp.compare.find({'name': 'bob'}) def test__pop_invalid_type(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': 'green'}) self.cmp.compare_exceptions.update_many({'name': 'bob'}, {'$pop': {'hat': 1}}) self.cmp.compare_exceptions.update_many({'name': 'bob'}, {'$pop': {'hat': -1}}) def test__pop_invalid_syntax(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': ['green']}) self.cmp.compare_exceptions.update_many({'name': 'bob'}, {'$pop': {'hat': 2}}) self.cmp.compare_exceptions.update_many({'name': 'bob'}, {'$pop': {'hat': '5'}}) self.cmp.compare_exceptions.update_many({'name': 'bob'}, {'$pop': {'hat.-1': 1}}) def test__pop_array_in_array(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': [['green']]}) self.cmp.do.update_many({'name': 'bob'}, {'$pop': {'hat.0': 1}}) self.cmp.compare.find({'name': 'bob'}) def test__pop_too_far_in_array(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': [['green']]}) self.cmp.do.update_many({'name': 'bob'}, {'$pop': {'hat.50': 1}}) self.cmp.compare.find({'name': 'bob'}) def test__pop_document_in_array(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': [{'hat': ['green']}]}) self.cmp.do.update_many({'name': 'bob'}, {'$pop': {'hat.0.hat': 1}}) self.cmp.compare.find({'name': 'bob'}) def test__pop_invalid_document_in_array(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': [{'hat': 'green'}]}) self.cmp.compare_exceptions.update_many({'name': 'bob'}, {'$pop': {'hat.0.hat': 1}}) def test__pop_empty(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': []}) self.cmp.do.update_many({'name': 'bob'}, {'$pop': {'hat': 1}}) self.cmp.compare.find({'name': 'bob'}) def test__pull(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$pull': {'hat': 'green'}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': ['green', 'tall']}) self.cmp.do.update_many({'name': 'bob'}, {'$pull': {'hat': 'green'}}) self.cmp.compare.find({'name': 'bob'}) def test__pull_query(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': [{'size': 5}, {'size': 10}]}) self.cmp.do.update_many( {'name': 'bob'}, {'$pull': {'hat': {'size': {'$gt': 6}}}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.delete_many({}) self.cmp.do.insert_one( {'name': 'bob', 'hat': {'sizes': [{'size': 5}, {'size': 8}, {'size': 10}]}} ) self.cmp.do.update_many( {'name': 'bob'}, {'$pull': {'hat.sizes': {'size': {'$gt': 6}}}}) self.cmp.compare.find({'name': 'bob'}) def test__pull_in_query_operator(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'sizes': [0, 1, 2, 3, 4, 5]}) self.cmp.do.update_one({'name': 'bob'}, {'$pull': {'sizes': {'$in': [1, 3]}}}) self.cmp.compare.find({'name': 'bob'}) def test__pull_in_nested_field(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'nested': {'sizes': [0, 1, 2, 3, 4, 5]}}) self.cmp.do.update_one({'name': 'bob'}, {'$pull': {'nested.sizes': {'$in': [1, 3]}}}) self.cmp.compare.find({'name': 'bob'}) def test__pull_nested_dict(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({ 'name': 'bob', 'hat': [ {'name': 'derby', 'sizes': [{'size': 'L', 'quantity': 3}, {'size': 'XL', 'quantity': 4}], 'colors': ['green', 'blue']}, {'name': 'cap', 'sizes': [{'size': 'S', 'quantity': 10}, {'size': 'L', 'quantity': 5}], 'colors': ['blue']}]}) self.cmp.do.update_many( {'hat': {'$elemMatch': {'name': 'derby'}}}, {'$pull': {'hat.$.sizes': {'size': 'L'}}}) self.cmp.compare.find({'name': 'bob'}) def test__pull_nested_list(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one( {'name': 'bob', 'hat': [{'name': 'derby', 'sizes': ['L', 'XL']}, {'name': 'cap', 'sizes': ['S', 'L']}]}) self.cmp.do.update_many( {'hat': {'$elemMatch': {'name': 'derby'}}}, {'$pull': {'hat.$.sizes': 'XL'}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.delete_many({}) self.cmp.do.insert_one( {'name': 'bob', 'hat': {'nested': ['element1', 'element2', 'element1']}}) self.cmp.do.update_many({'name': 'bob'}, {'$pull': {'hat.nested': 'element1'}}) self.cmp.compare.find({'name': 'bob'}) def test__pullAll(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$pullAll': {'hat': ['green']}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_many( {'name': 'bob'}, {'$pullAll': {'hat': ['green', 'blue']}}) self.cmp.compare.find({'name': 'bob'}) self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': ['green', 'tall', 'blue']}) self.cmp.do.update_many({'name': 'bob'}, {'$pullAll': {'hat': ['green']}}) self.cmp.compare.find({'name': 'bob'}) def test__pullAll_dollar_operand(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'takes': [ {'a': 1, 'tags': [0, 1, 2, 4]}, {'a': 2, 'tags': [0, 1, 2, 4]}, {'a': 1, 'tags': [0, 1, 4]}, {'a': 1, 'tags': [2, 3, 5]}]}) self.cmp.do.update_many( {'name': 'bob', 'takes': {'$elemMatch': {'a': 1}}}, {'$pullAll': {'takes.$.tags': [1, 2, 3]}}) self.cmp.compare.find({'name': 'bob'}) def test__push(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': ['green', 'tall']}) self.cmp.do.update_many({'name': 'bob'}, {'$push': {'hat': 'wide'}}) self.cmp.compare.find({'name': 'bob'}) def test__push_dict(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one( {'name': 'bob', 'hat': [{'name': 'derby', 'sizes': ['L', 'XL']}]}) self.cmp.do.update_many( {'name': 'bob'}, {'$push': {'hat': {'name': 'cap', 'sizes': ['S', 'L']}}}) self.cmp.compare.find({'name': 'bob'}) def test__push_each(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': ['green', 'tall']}) self.cmp.do.update_many( {'name': 'bob'}, {'$push': {'hat': {'$each': ['wide', 'blue']}}}) self.cmp.compare.find({'name': 'bob'}) def test__push_nested_dict(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({ 'name': 'bob', 'hat': [ {'name': 'derby', 'sizes': [{'size': 'L', 'quantity': 3}, {'size': 'XL', 'quantity': 4}], 'colors': ['green', 'blue']}, {'name': 'cap', 'sizes': [{'size': 'S', 'quantity': 10}, {'size': 'L', 'quantity': 5}], 'colors': ['blue']}]}) self.cmp.do.update_many( {'hat': {'$elemMatch': {'name': 'derby'}}}, {'$push': {'hat.$.sizes': {'size': 'M', 'quantity': 6}}}) self.cmp.compare.find({'name': 'bob'}) def test__push_nested_dict_each(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({ 'name': 'bob', 'hat': [ {'name': 'derby', 'sizes': [{'size': 'L', 'quantity': 3}, {'size': 'XL', 'quantity': 4}], 'colors': ['green', 'blue']}, {'name': 'cap', 'sizes': [{'size': 'S', 'quantity': 10}, {'size': 'L', 'quantity': 5}], 'colors': ['blue']}]}) self.cmp.do.update_many( {'hat': {'$elemMatch': {'name': 'derby'}}}, {'$push': {'hat.$.sizes': {'$each': [{'size': 'M', 'quantity': 6}, {'size': 'S', 'quantity': 1}]}}}) self.cmp.compare.find({'name': 'bob'}) def test__push_nested_dict_in_list(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({ 'name': 'bob', 'hat': [ {'name': 'derby', 'sizes': [{'size': 'L', 'quantity': 3}, {'size': 'XL', 'quantity': 4}], 'colors': ['green', 'blue']}, {'name': 'cap', 'sizes': [{'size': 'S', 'quantity': 10}, {'size': 'L', 'quantity': 5}], 'colors': ['blue']}]}) self.cmp.do.update_many( {'name': 'bob'}, {'$push': {'hat.1.sizes': {'size': 'M', 'quantity': 6}}}) self.cmp.compare.find({'name': 'bob'}) def test__push_nested_list_each(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({ 'name': 'bob', 'hat': [ {'name': 'derby', 'sizes': ['L', 'XL'], 'colors': ['green', 'blue']}, {'name': 'cap', 'sizes': ['S', 'L'], 'colors': ['blue']} ] }) self.cmp.do.update_many( {'hat': {'$elemMatch': {'name': 'derby'}}}, {'$push': {'hat.$.sizes': {'$each': ['M', 'S']}}}) self.cmp.compare.find({'name': 'bob'}) def test__push_nested_attribute(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': {'data': {'sizes': ['XL']}}}) self.cmp.do.update_many({'name': 'bob'}, {'$push': {'hat.data.sizes': 'L'}}) self.cmp.compare.find({'name': 'bob'}) def test__push_nested_attribute_each(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob', 'hat': {}}) self.cmp.do.update_many( {'name': 'bob'}, {'$push': {'hat.first': {'$each': ['a', 'b']}}}) self.cmp.compare.find({'name': 'bob'}) def test__push_to_absent_nested_attribute(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$push': {'hat.data.sizes': 'L'}}) self.cmp.compare.find({'name': 'bob'}) def test__push_to_absent_field(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_many({'name': 'bob'}, {'$push': {'hat': 'wide'}}) self.cmp.compare.find({'name': 'bob'}) def test__push_each_to_absent_field(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'name': 'bob'}) self.cmp.do.update_many( {'name': 'bob'}, {'$push': {'hat': {'$each': ['wide', 'blue']}}}) self.cmp.compare.find({'name': 'bob'}) def test__push_each_slice(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'scores': [40, 50, 60]}) self.cmp.do.update_one({}, {'$push': {'scores': { '$each': [80, 78, 86], '$slice': -5, }}}) self.cmp.compare.find() self.cmp.do.update_one({}, {'$push': {'scores': { '$each': [100, 20], '$slice': 3, }}}) self.cmp.compare.find() self.cmp.do.update_one({}, {'$push': {'scores': { '$each': [], '$slice': 2, }}}) self.cmp.compare.find() self.cmp.do.update_one({}, {'$push': {'scores': { '$each': [25, 15], '$slice': 0, }}}) self.cmp.compare.find() def test__update_push_slice_nested_field(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'games': [{'scores': [40, 50, 60]}, {'a': 1}]}) self.cmp.do.update_one({}, {'$push': {'games.0.scores': { '$each': [80, 78, 86], '$slice': -5, }}}) self.cmp.compare.find() self.cmp.do.update_one( {'games': {'$elemMatch': {'scores': {'$exists': True}}}}, {'$push': {'games.$.scores': {'$each': [0, 1], '$slice': -5}}}, ) self.cmp.compare.find() def test__update_push_array_of_arrays(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one({'scores': [[40, 50], [60, 20]]}) self.cmp.do.update_one( {'scores': {'$elemMatch': {'0': 60}}}, {'$push': {'scores.$': 30}}, ) self.cmp.compare.find() def test__update_push_sort(self): self.cmp.do.delete_many({}) self.cmp.do.insert_one( {'a': {'b': [{'value': 3}, {'value': 1}, {'value': 2}]}}) self.cmp.do.update_one({}, {'$push': {'a.b': { '$each': [{'value': 4}], '$sort': {'value': 1}, }}}) self.cmp.compare.find() def _compare_update_push_position(self, position): self.cmp.do.delete_many({}) self.cmp.do.insert_one( {'a': {'b': [{'value': 3}, {'value': 1}, {'value': 2}]}}) self.cmp.do.update_one({}, {'$push': {'a.b': { '$each': [{'value': 4}], '$position': position, }}}) self.cmp.compare.find() def test__update_push_position(self): self._compare_update_push_position(0) self._compare_update_push_position(1) self._compare_update_push_position(5) # TODO(pascal): Enable once we test against Mongo v3.6+ # self._compare_update_push_position(-2) def test__drop(self): self.cmp.do.insert_one({'name': 'another new'}) self.cmp.do.drop() self.cmp.compare.find({}) def test__ensure_index(self): if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.ensure_index('name') return self.cmp.compare.ensure_index('name') self.cmp.compare.ensure_index('hat', cache_for=100) self.cmp.compare.ensure_index([('name', 1), ('hat', -1)]) self.cmp.do.insert_one({}) self.cmp.compare.index_information() def test__drop_index(self): self.cmp.do.insert_one({}) self.cmp.compare.create_index([('name', 1), ('hat', -1)]) self.cmp.compare.drop_index([('name', 1), ('hat', -1)]) self.cmp.compare.index_information() def test__drop_index_by_name(self): self.cmp.do.insert_one({}) results = self.cmp.compare.create_index('name') self.cmp.compare.drop_index(results['real']) self.cmp.compare.index_information() def test__index_information(self): self.cmp.do.insert_one({}) self.cmp.compare.index_information() def test__list_indexes(self): self.cmp.do.insert_one({}) self.cmp.compare_ignore_order.sort_by(lambda i: i['name']).list_indexes() def test__empty_logical_operators(self): for operator in ('$or', '$and', '$nor'): self.cmp.compare_exceptions.find({operator: []}) def test__rename(self): input_ = {'_id': 1, 'foo': 'bar'} self.cmp.do.insert_one(input_) query = {'_id': 1} update = {'$rename': {'foo': 'bar'}} self.cmp.do.update_one(query, update=update) self.cmp.compare.find() def test__rename_collection(self): self.cmp.do.insert_one({'_id': 1, 'foo': 'bar'}) self.cmp.compare.rename('new_name') self.cmp.compare.find() def test__set_equals(self): self.cmp.do.insert_many([ {'array': ['one', 'three']}, ]) self.cmp.compare.aggregate([{'$project': { '_id': 0, 'same_array': {'$setEquals': ['$array', '$array']}, 'eq_array': {'$setEquals': [['one', 'three'], '$array']}, 'ne_array': {'$setEquals': [['one', 'two'], '$array']}, 'eq_in_another_order': {'$setEquals': [['one', 'two'], ['two', 'one']]}, 'ne_in_another_order': {'$setEquals': [['one', 'two'], ['three', 'one', 'two']]}, 'three_equal': {'$setEquals': [['one', 'two'], ['two', 'one'], ['one', 'two']]}, 'three_not_equal': {'$setEquals': [['one', 'three'], ['two', 'one'], ['two', 'one']]}, }}]) @skipIf( helpers.PYMONGO_VERSION < version.parse('4.0'), 'pymongo v4 dropped map reduce methods') def test__map_reduce_fails(self): self.cmp.compare_exceptions.map_reduce(Code(''), Code(''), 'myresults') self.cmp.compare_exceptions.inline_map_reduce(Code(''), Code('')) self.cmp.compare_exceptions.group(['a'], {'a': {'$lt': 3}}, {'count': 0}, Code(''' function(cur, result) { result.count += cur.count } ''')) @skipIf(helpers.PYMONGO_VERSION >= version.parse('4.0'), 'pymongo v4 dropped group method') @skipIf(helpers.PYMONGO_VERSION < version.parse('3.6'), 'pymongo v3.6 broke group method') def test__group_fails(self): self.cmp.compare_exceptions.group(['a'], {'a': {'$lt': 3}}, {'count': 0}, Code(''' function(cur, result) { result.count += cur.count } ''')) def test__aggregate_system_variables_generate_array(self): self.cmp.do.drop() self.cmp.do.insert_one( {'name': 'foo', 'errors': [ {'error_type': 1, 'description': 'problem 1'}, {'error_type': 2, 'description': 'problem 2'}]}) self.cmp.compare.aggregate([{'$project': {'error_type': '$$ROOT.errors.error_type'}}]) @skipIf(not helpers.HAVE_PYMONGO, 'pymongo not installed') @skipIf(not _HAVE_MAP_REDUCE, 'execjs not installed') @skipIf(helpers.PYMONGO_VERSION >= version.parse('4.0'), 'pymongo v4 dropped map reduce') class CollectionMapReduceTest(TestCase): def setUp(self): self.db = mongomock.MongoClient().map_reduce_test self.data = [{'x': 1, 'tags': ['dog', 'cat']}, {'x': 2, 'tags': ['cat']}, {'x': 3, 'tags': ['mouse', 'cat', 'dog']}, {'x': 4, 'tags': []}] for item in self.data: self.db.things.insert_one(item) self.map_func = Code(''' function() { this.tags.forEach(function(z) { emit(z, 1); }); }''') self.reduce_func = Code(''' function(key, values) { var total = 0; for(var i = 0; i= version.parse('3.6'), 'pymongo v3.6 broke group') class GroupTest(_CollectionComparisonTest): def setUp(self): _CollectionComparisonTest.setUp(self) self._id1 = ObjectId() self.data = [ {'a': 1, 'count': 4}, {'a': 1, 'count': 2}, {'a': 1, 'count': 4}, {'a': 2, 'count': 3}, {'a': 2, 'count': 1}, {'a': 1, 'count': 5}, {'a': 4, 'count': 4}, {'b': 4, 'foo': 4}, {'b': 2, 'foo': 3, 'name': 'theone'}, {'b': 1, 'foo': 2}, {'b': 1, 'foo': self._id1}, ] self.cmp.do.insert_many(self.data) def test__group1(self): key = ['a'] initial = {'count': 0} condition = {'a': {'$lt': 3}} reduce_func = Code(''' function(cur, result) { result.count += cur.count } ''') self.cmp.compare.group(key, condition, initial, reduce_func) def test__group2(self): reduce_func = Code(''' function(cur, result) { result.count += 1 } ''') self.cmp.compare.group( key=['b'], condition={'foo': {'$in': [3, 4]}, 'name': 'theone'}, initial={'count': 0}, reduce=reduce_func) def test__group3(self): reducer = Code(''' function(obj, result) {result.count+=1 } ''') conditions = {'foo': {'$in': [self._id1]}} self.cmp.compare.group( key=['foo'], condition=conditions, initial={'count': 0}, reduce=reducer) @skipIf(not helpers.HAVE_PYMONGO, 'pymongo not installed') class MongoClientAggregateTest(_CollectionComparisonTest): def setUp(self): super(MongoClientAggregateTest, self).setUp() self.data = [ {'_id': ObjectId(), 'a': 1, 'b': 1, 'count': 4, 'swallows': ['European swallow'], 'date': datetime.datetime(2015, 10, 1, 10, 0)}, {'_id': ObjectId(), 'a': 1, 'b': 1, 'count': 2, 'swallows': ['African swallow'], 'date': datetime.datetime(2015, 12, 1, 12, 0)}, {'_id': ObjectId(), 'a': 1, 'b': 2, 'count': 4, 'swallows': ['European swallow'], 'date': datetime.datetime(2014, 10, 2, 12, 0)}, {'_id': ObjectId(), 'a': 2, 'b': 2, 'count': 3, 'swallows': ['African swallow', 'European swallow'], 'date': datetime.datetime(2015, 1, 2, 10, 0)}, {'_id': ObjectId(), 'a': 2, 'b': 3, 'count': 1, 'swallows': [], 'date': datetime.datetime(2013, 1, 3, 12, 0)}, {'_id': ObjectId(), 'a': 1, 'b': 4, 'count': 5, 'swallows': ['African swallow', 'European swallow'], 'date': datetime.datetime(2015, 8, 4, 12, 0)}, {'_id': ObjectId(), 'a': 4, 'b': 4, 'count': 4, 'swallows': ['unladen swallow'], 'date': datetime.datetime(2014, 7, 4, 13, 0)}] for item in self.data: self.cmp.do.insert_one(item) def test__aggregate1(self): pipeline = [ {'$match': {'a': {'$lt': 3}}}, {'$sort': {'_id': -1}}, ] self.cmp.compare.aggregate(pipeline) def test__aggregate2(self): pipeline = [ {'$group': {'_id': '$a', 'count': {'$sum': '$count'}}}, {'$match': {'a': {'$lt': 3}}}, {'$sort': {'_id': -1, 'count': 1}}, ] self.cmp.compare.aggregate(pipeline) def test__aggregate3(self): pipeline = [ {'$group': {'_id': 'a', 'count': {'$sum': '$count'}}}, {'$match': {'a': {'$lt': 3}}}, {'$sort': {'_id': -1, 'count': 1}}, {'$skip': 1}, {'$limit': 2}] self.cmp.compare.aggregate(pipeline) def test__aggregate4(self): pipeline = [ {'$unwind': '$swallows'}, {'$sort': {'count': -1, 'swallows': -1}}] self.cmp.compare.aggregate(pipeline) def test__aggregate5(self): pipeline = [ {'$group': {'_id': {'id_a': '$a'}, 'total': {'$sum': '$count'}, 'avg': {'$avg': '$count'}}}, {'$sort': {'_id.a': 1, 'total': 1, 'avg': 1}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate6(self): pipeline = [ {'$group': {'_id': {'id_a': '$a', 'id_b': '$b'}, 'total': {'$sum': '$count'}, 'avg': {'$avg': '$count'}}}, {'$sort': {'_id.id_a': 1, '_id.id_b': 1, 'total': 1, 'avg': 1}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate7(self): pipeline = [ {'$group': {'_id': {'id_a': '$a', 'id_b': {'$year': '$date'}}, 'total': {'$sum': '$count'}, 'avg': {'$avg': '$count'}}}, {'$sort': {'_id.id_a': 1, '_id.id_b': 1, 'total': 1, 'avg': 1}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate8(self): pipeline = [ {'$group': {'_id': None, 'counts': {'$sum': '$count'}}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate9(self): pipeline = [ {'$group': {'_id': {'id_a': '$a'}, 'total': {'$sum': '$count'}, 'avg': {'$avg': '$count'}}}, {'$group': {'_id': None, 'counts': {'$sum': '$total'}}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate10(self): # group on compound index self.cmp.do.delete_many({}) data = [ {'_id': ObjectId(), 'key_1': {'sub_key_1': 'value_1'}, 'nb': 1}, {'_id': ObjectId(), 'key_1': {'sub_key_1': 'value_2'}, 'nb': 1}, {'_id': ObjectId(), 'key_1': {'sub_key_1': 'value_1'}, 'nb': 2} ] for item in data: self.cmp.do.insert_one(item) pipeline = [ {'$group': {'_id': '$key_1.sub_key_1', 'nb': {'$sum': '$nb'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate11(self): pipeline = [ {'$group': {'_id': None, 'max_count': {'$max': '$count'}, 'min_count': {'$min': '$count'}}}, ] self.cmp.compare.aggregate(pipeline) def test__aggregate12(self): pipeline = [ {'$group': {'_id': '$a', 'max_count': {'$max': '$count'}, 'min_count': {'$min': '$count'}}}, {'$sort': {'_id': 1}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate13(self): pipeline = [ {'$sort': {'date': 1}}, {'$group': {'_id': None, 'last_date': {'$last': '$date'}, 'first_date': {'$first': '$date'}}}, ] self.cmp.compare.aggregate(pipeline) def test__aggregate_on_no_data(self): pipeline = [ {'$sort': {'date': 1}}, {'$group': { '_id': None, 'last_unkown': {'$last': '$unkown_field'}, 'first_unknown': {'$first': '$unknown_field'}, }}, ] self.cmp.compare.aggregate(pipeline) def test__aggregate14(self): pipeline = [ {'$sort': {'date': 1}}, {'$group': {'_id': '$a', 'last_date': {'$last': '$date'}, 'first_date': {'$first': '$date'}}}, {'$sort': {'_id': 1}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate_group_by_dbref(self): self.cmp.do.insert_many([ {'myref': DBRef('a', '1')}, {'myref': DBRef('a', '1')}, {'myref': DBRef('a', '2')}, {'myref': DBRef('b', '1')}, ]) self.cmp.compare.aggregate([ {'$group': {'_id': '$myref'}} ]) def test__aggregate_project_include_in_inclusion(self): pipeline = [ {'$project': {'a': 1, 'b': 1}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate_project_exclude_in_exclusion(self): pipeline = [ {'$project': {'a': 0, 'b': 0}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate_project_exclude_id_in_inclusion(self): pipeline = [ {'$project': {'a': 1, '_id': 0}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate_project_with_subfields(self): self.cmp.do.insert_many([ {'a': {'b': 3}, 'other': 1}, {'a': {'c': 3}}, {'b': {'c': 3}}, {'a': 5}, ]) pipeline = [ {'$project': {'a.b': 1}} ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate_project_with_subfields_exclude(self): self.cmp.do.insert_many([ {'a': {'b': 3}, 'other': 1}, {'a': {'b': 3, 'd': 5}}, {'a': {'c': 3, 'd': 5}}, {'b': {'c': 3}}, {'a': 5}, ]) pipeline = [ {'$project': {'a.b': 0}} ] self.cmp.compare_ignore_order.aggregate(pipeline) def test_aggregate_project_with_missing_subfields(self): self.cmp.do.insert_many([ {'a': {'b': 3}, 'other': 1}, {'a': {'b': {'c': 4}, 'd': 5}}, {'a': {'c': 3, 'd': 5}}, {'b': {'c': 3}}, {'a': 5}, ]) pipeline = [ {'$project': {'_id': False, 'e': '$a.b.c'}} ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate_unwind_project_id(self): self.cmp.do.insert_one({ '_id': 'id0', 'c2': [ {'_id': 'id1', 'o': 'x'}, {'_id': 'id2', 'o': 'y'}, {'_id': 'id3', 'o': 'z'}, ], }) pipeline = [ {'$unwind': '$c2'}, {'$project': {'_id': '$c2._id', 'o': '$c2.o'}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate17(self): pipeline = [ {'$project': {'_id': 0, 'created': {'$subtract': [{'$min': ['$a', '$b']}, '$count']}}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate18(self): pipeline = [ {'$project': {'_id': 0, 'created': {'$subtract': ['$a', '$b']}}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate19(self): pipeline = [ {'$project': {'_id': 0, 'created': {'$subtract': ['$a', 1]}}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate20(self): pipeline = [{'$project': { '_id': 0, 'abs': {'$abs': '$b'}, 'add': {'$add': ['$a', 1, '$b']}, 'ceil': {'$ceil': 8.35}, 'div': {'$divide': ['$a', 1]}, 'exp': {'$exp': 2}, 'floor': {'$floor': 4.65}, 'ln': {'$ln': 100}, 'log': {'$log': [8, 2]}, 'log10': {'$log10': 1000}, 'mod': {'$mod': [46, 9]}, 'multiply': {'$multiply': [5, '$a', '$b']}, 'pow': {'$pow': [4, 2]}, 'sqrt': {'$sqrt': 100}, 'trunc': {'$trunc': 8.35}, }}] self.cmp.compare.aggregate(pipeline) def test__aggregate21(self): pipeline = [ {'$group': {'_id': '$a', 'count': {'$sum': 1}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate22(self): pipeline = [ {'$group': {'_id': {'$gte': ['$a', 2]}, 'total': {'$sum': '$count'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate23(self): # make sure we aggregate compound keys correctly pipeline = [ {'$group': {'_id': {'id_a': '$a', 'id_b': '$b'}, 'total': {'$sum': '$count'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate24(self): # make sure we aggregate zero rows correctly pipeline = [ {'$match': {'_id': '123456'}}, {'$group': {'_id': {'$eq': ['$a', 1]}, 'total': {'$sum': '$count'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate25(self): pipeline = [ {'$group': {'_id': {'$eq': [{'$year': '$date'}, 2015]}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate26(self): pipeline = [ {'$group': {'_id': {'$eq': [{'$year': '$date'}, 2015]}, 'total': {'$sum': '$count'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate27(self): # test $lookup stage pipeline = [ {'$lookup': { 'from': self.collection_name, 'localField': 'a', 'foreignField': 'b', 'as': 'lookup' }} ] self.cmp.compare.aggregate(pipeline) def test__aggregate27b(self): # test $graphLookup stage self.cmp.do.delete_many({}) data = [ {'_id': ObjectId(), 'name': 'a', 'child': 'b', 'val': 2}, {'_id': ObjectId(), 'name': 'b', 'child': 'c', 'val': 3}, {'_id': ObjectId(), 'name': 'c', 'child': None, 'val': 4}, {'_id': ObjectId(), 'name': 'd', 'child': 'a', 'val': 5} ] for item in data: self.cmp.do.insert_one(item) pipeline = [ {'$match': {'name': 'a'}}, {'$graphLookup': { 'from': self.collection_name, 'startWith': '$child', 'connectFromField': 'child', 'connectToField': 'name', 'as': 'lookup' }}, {'$unwind': '$lookup'}, {'$sort': {'lookup.name': 1}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate28(self): pipeline = [{'$group': { '_id': '$b', 'total2015': {'$sum': {'$cond': [{'$ne': [{'$year': '$date'}, 2015]}, 0, 1]}}, }}] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate29(self): # group addToSet pipeline = [ {'$group': {'_id': '$a', 'nb': {'$addToSet': '$count'}}}, {'$sort': {'_id': 1}} ] # self.cmp.compare cannot be used as addToSet returns elements in an unpredictable order aggregations = self.cmp.do.aggregate(pipeline) expected = list(aggregations['real']) result = list(aggregations['fake']) self.assertEqual(len(result), len(expected)) for expected_elt, result_elt in zip(expected, result): self.assertCountEqual(expected_elt.keys(), result_elt.keys()) for key in result_elt: if isinstance(result_elt[key], list): self.assertCountEqual(result_elt[key], expected_elt[key], msg=key) else: self.assertEqual(result_elt[key], expected_elt[key], msg=key) def test__aggregate30(self): # group addToSet dict element self.cmp.do.delete_many({}) data = [ {'a': {'c': '1', 'd': 1}, 'b': {'c': '2', 'd': 2}}, {'a': {'c': '1', 'd': 3}, 'b': {'c': '4', 'd': 4}}, {'a': {'c': '5', 'd': 1}, 'b': {'c': '6', 'd': 6}}, {'a': {'c': '5', 'd': 2}, 'b': {'c': '6', 'd': 6}} ] self.cmp.do.insert_many(data) pipeline = [ {'$group': {'_id': 'a.c', 'nb': {'$addToSet': 'b'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate31(self): # group addToSet creating dict pipeline = [ {'$group': {'_id': '$count', 'set': {'$addToSet': {'a': '$a', 'b': '$b'}}}}, ] # self.cmp.compare cannot be used as addToSet returns elements in an unpredictable order aggregations = self.cmp.do.aggregate(pipeline) expected = list(aggregations['real']) result = list(aggregations['fake']) self.assertEqual(len(result), len(expected)) set_expected = set([ tuple(sorted(e.items())) for elt in expected for e in elt['set'] ]) set_result = set([ tuple(sorted(e.items())) for elt in result for e in elt['set'] ]) self.assertEqual(set_result, set_expected) def test__aggregate_add_to_set_missing_value(self): self.cmp.do.delete_many({}) data = [ {'a': {'c': '1', 'd': 1}, 'b': 1}, {'a': {'c': '1', 'd': 2}} ] self.cmp.do.insert_many(data) pipeline = [ {'$group': {'_id': 'a.c', 'nb': {'$addToSet': 'b'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate32(self): self.cmp.do.drop() self.cmp.do.insert_many([ {'group': 'one'}, {'group': 'one'}, {'group': 'one', 'data': None}, {'group': 'one', 'data': 0}, {'group': 'one', 'data': 2}, {'group': 'one', 'data': {'a': 1}}, {'group': 'one', 'data': [1, 2]}, {'group': 'one', 'data': [3, 4]}, ]) pipeline = [{'$group': { '_id': '$group', 'count': {'$sum': 1}, 'countData': {'$sum': {'$cond': ['$data', 1, 0]}}, 'countDataExists': {'$sum': {'$cond': { 'if': {'$gt': ['$data', None]}, 'then': 1, 'else': 0, }}}, }}] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate33(self): self.cmp.do.drop() self.cmp.do.insert_one({'_id': 1, 'a': 2, 'b': 3, 'c': '$d'}) pipeline = [{'$project': { '_id': 0, 'max': {'$max': [5, 9, '$a', None]}, 'min': {'$min': [8, 2, None, 3, '$a', '$b']}, 'avg': {'$avg': [4, 2, None, 3, '$a', '$b', 4]}, 'sum': {'$sum': [4, 2, None, 3, '$a', '$b', {'$sum': [0, 1, '$b']}]}, 'maxString': {'$max': [{'$literal': '$b'}, '$c']}, 'maxNone': {'$max': [None, None]}, 'minNone': {'$min': [None, None]}, 'avgNone': {'$avg': ['a', None]}, 'sumNone': {'$sum': ['a', None]}, }}] self.cmp.compare.aggregate(pipeline) def test__aggregate34(self): self.cmp.do.drop() self.cmp.do.insert_one({'_id': 1, 'a': 'Hello', 'b': 'World'}) pipeline = [{'$project': { '_id': 0, 'concat': {'$concat': ['$a', ' Dear ', '$b']}, 'concat_none': {'$concat': ['$a', None, '$b']}, 'sub1': {'$substr': ['$a', 0, 4]}, 'lower': {'$toLower': '$a'}, 'lower_err': {'$toLower': None}, 'split_string_none': {'$split': [None, 'l']}, 'split_string_missing': {'$split': ['$missingField', 'l']}, 'split_delimiter_none': {'$split': ['$a', None]}, 'split_delimiter_missing': {'$split': ['$a', '$missingField']}, 'split': {'$split': ['$a', 'l']}, 'strcasecmp': {'$strcasecmp': ['$a', '$b']}, 'upper': {'$toUpper': '$a'}, 'upper_err': {'$toUpper': None}, }}] self.cmp.compare.aggregate(pipeline) def test__aggregate_regexpmatch(self): self.cmp.do.insert_many([ {'_id': 1, 'description': 'Single LINE description.'}, {'_id': 2, 'description': 'First lines\nsecond line'}, {'_id': 3, 'description': 'Many spaces before line'}, {'_id': 4, 'description': 'Multiple\nline descriptions'}, {'_id': 5, 'description': 'anchors, links and hyperlinks'}, {'_id': 6, 'description': u'métier work vocation'} ]) self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$description', 'regex': 'line'}}, }}]) self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$description', 'regex': 'lin(e|k)'}}, }}]) self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$description', 'regex': 'line', 'options': 'i'}}, }}]) self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$description', 'regex': Regex('line', 'i')}}, }}]) self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': { 'input': '$description', 'regex': 'line(e|k) # matches line or link', 'options': 'x', }}, }}]) self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': { 'input': '$description', 'regex': 'm.*line', 'options': 'si', }}, }}]) # Missing fields self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$missing', 'regex': 'line'}}, }}]) self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$description', 'regex': '$missing'}}, }}]) # Exceptions self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': ['$description', 'line']}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': {'inut': '$description', 'regex': 'line'}}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$description', 'regex': 'line', 'other': True}}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': 42, 'regex': 'line'}}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$description', 'regex': 'line', 'options': '?'}}, }}]) self.cmp.compare.aggregate([{'$addFields': { 'result': {'$regexMatch': { 'input': '$description', 'regex': Regex('line'), 'options': 'i'}}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': { 'input': '$description', 'regex': re.compile('line', re.U), 'options': 'i'}}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': { 'input': '$description', 'regex': re.compile('line', re.U)}}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': { 'input': '$description', 'regex': Regex('line', 'i'), 'options': 'i'}}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': { 'input': '$description', 'regex': Regex('line', 'u')}}, }}]) self.cmp.compare_exceptions.aggregate([{'$addFields': { 'result': {'$regexMatch': {'input': '$description', 'regex': 5}}, }}]) def test__aggregate35(self): self.cmp.do.drop() self.cmp.do.insert_one({ '_id': 1, 'a': 2, 'b': 3, 'c': '$d', 'd': decimal128.Decimal128('4') }) pipeline = [{'$project': { '_id': 0, 'sum': {'$sum': [4, 2, None, 3, '$a', '$b', '$d', {'$sum': [0, 1, '$b']}]}, 'sumNone': {'$sum': ['a', None]}, }}] self.cmp.compare.aggregate(pipeline) def test__aggregate_project_id_0(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'_id': 4}, {'a': 5}, {}, ]) pipeline = [{'$project': {'_id': 0}}] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate_project_array_subfield(self): self.cmp.do.insert_many([ {'_id': 1, 'a': [{'b': 1, 'c': 2, 'd': 3}], 'e': 4}, {'_id': 2, 'a': [{'c': 12, 'd': 13}], 'e': 14}, {'_id': 3, 'a': [{'b': 21, 'd': 23}], 'e': 24}, {'_id': 4, 'a': [{'b': 31, 'c': 32}], 'e': 34}, {'_id': 5, 'a': [{'b': 41}], 'e': 44}, {'_id': 6, 'a': [{'c': 51}], 'e': 54}, {'_id': 7, 'a': [{'d': 51}], 'e': 54}, {'_id': 8, 'a': [ {'b': 61, 'c': 62, 'd': 63}, 65, 'foobar', {'b': 66, 'c': 67, 'd': 68}], 'e': 64}, {'_id': 9, 'a': []}, {'_id': 10, 'a': [1, 2, 3, 4]}, {'_id': 11, 'a': 'foobar'}, {'_id': 12, 'a': 5}, ]) pipeline = [{'$project': {'a.b': 1, 'a.c': 1}}] self.cmp.compare_ignore_order.aggregate(pipeline) def test__aggregate_project_array_size_missing(self): self.cmp.do.insert_one({'_id': 1}) self.cmp.compare_exceptions.aggregate([ {'$match': {'_id': 1}}, {'$project': {'a': {'$size': '$arr'}}}, ]) def test__aggregate_bucket(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ { '_id': 1, 'title': 'The Pillars of Society', 'artist': 'Grosz', 'year': 1926, 'price': 199.99, }, { '_id': 2, 'title': 'Melancholy III', 'artist': 'Munch', 'year': 1902, 'price': 200.00, }, { '_id': 3, 'title': 'Dancer', 'artist': 'Miro', 'year': 1925, 'price': 76.04, }, { '_id': 4, 'title': 'The Great Wave off Kanagawa', 'artist': 'Hokusai', 'price': 167.30, }, { '_id': 5, 'title': 'The Persistence of Memory', 'artist': 'Dali', 'year': 1931, 'price': 483.00, }, { '_id': 6, 'title': 'Composition VII', 'artist': 'Kandinsky', 'year': 1913, 'price': 385.00, }, { '_id': 7, 'title': 'The Scream', 'artist': 'Munch', 'year': 1893, # No price }, { '_id': 8, 'title': 'Blue Flower', 'artist': "O'Keefe", 'year': 1918, 'price': 118.42, }, ]) self.cmp.compare.aggregate([{'$bucket': { 'groupBy': '$price', 'boundaries': [0, 200, 400], 'default': 'Other', 'output': { 'count': {'$sum': 1}, 'titles': {'$push': '$title'}, }, }}]) self.cmp.compare.aggregate([{'$bucket': { 'groupBy': '$price', 'boundaries': [0, 200, 400], 'default': 'Other', }}]) def test__aggregate_lookup_dot_in_local_field(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'_id': 2, 'should': {'do': 'join'}}, {'_id': 3, 'should': {'do': 'not_join'}}, {'_id': 4, 'should': 'skip'}, {'_id': 5, 'should': 'join'}, {'_id': 6, 'should': 'join'}, {'_id': 7, 'should': 'skip'}, ]) pipeline = [ {'$lookup': { 'from': self.collection_name, 'localField': 'should.do', 'foreignField': 'should', 'as': 'b' }} ] self.cmp.compare.aggregate(pipeline) def test__aggregate_count(self): self.cmp.do.insert_many([ {'_id': i} for i in range(5) ]) self.cmp.compare.aggregate([ {'$count': 'my_count'} ]) def test__aggregate_if_null(self): self.cmp.do.insert_one({'_id': 1, 'elem_a': ''}) self.cmp.compare.aggregate([ { '$project': { 'a': {'$ifNull': ['$elem_a', '']}, 'b': {'$ifNull': ['$elem_b', '']}, } } ]) def test__aggregate_if_null_multi_field(self): self.cmp.do.insert_one({'_id': 1, 'elem_a': ''}) # Multiple input expressions in $ifNull are not supported in MongoDB v4.4 and earlier. if SERVER_VERSION > version.parse('4.4'): compare = self.cmp.compare else: compare = self.cmp.compare_exceptions compare.aggregate([ { '$project': { 'a_and_b': {'$ifNull': ['$elem_a', '$elem_b', '']}, 'b_and_a': {'$ifNull': ['$elem_b', '$elem_a', '']}, 'b_and_c': {'$ifNull': ['$elem_b', '$elem_c', '']}, } } ]) def test__aggregate_is_number(self): self.cmp.do.insert_one( {'_id': 1, 'int': 3, 'big_int': 3 ** 10, 'negative': -3, 'str': 'not_a_number', 'str_numeric': '3', 'float': 3.3, 'negative_float': -3.3, 'bool': True, 'none': None}) self.cmp.compare.aggregate([ {'$project': { '_id': False, 'int': {'$isNumber': '$int'}, 'big_int': {'$isNumber': '$big_int'}, 'negative': {'$isNumber': '$negative'}, 'str': {'$isNumber': '$str'}, 'str_numeric': {'$isNumber': '$str_numeric'}, 'float': {'$isNumber': '$float'}, 'negative_float': {'$isNumber': '$negative_float'}, 'bool': {'$isNumber': '$bool'}, 'none': {'$isNumber': '$none'}, }} ]) def test__aggregate_is_array(self): self.cmp.do.insert_one( { '_id': 1, 'list': [1, 2, 3], 'tuple': (1, 2, 3), 'empty_list': [], 'empty_tuple': (), 'int': 3, 'str': '123', 'bool': True, 'none': None }) self.cmp.compare.aggregate([ {'$project': { '_id': False, 'list': {'$isArray': '$list'}, 'tuple': {'$isArray': '$tuple'}, 'empty_list': {'$isArray': '$empty_list'}, 'empty_tuple': {'$isArray': '$empty_tuple'}, 'int': {'$isArray': '$int'}, 'str': {'$isArray': '$str'}, 'bool': {'$isArray': '$bool'}, 'none': {'$isArray': '$none'} }} ]) def test__aggregate_facet(self): self.cmp.do.insert_many([ {'_id': i} for i in range(5) ]) self.cmp.compare.aggregate([ {'$facet': { 'pipeline_a': [{'$count': 'my_count'}], 'pipeline_b': [{'$group': {'_id': None}}]}} ]) def test__aggregate_project_rotate(self): self.cmp.do.insert_one({'_id': 1, 'a': 1, 'b': 2, 'c': 3}) self.cmp.compare.aggregate([ {'$project': {'a': '$b', 'b': '$a', 'c': 1}}, ]) def test__aggregate_unwind_options(self): self.cmp.do.drop() self.cmp.do.insert_many([ {'_id': 1, 'item': 'ABC', 'sizes': ['S', 'M', 'L']}, {'_id': 2, 'item': 'EFG', 'sizes': []}, {'_id': 3, 'item': 'IJK', 'sizes': 'M'}, {'_id': 4, 'item': 'LMN'}, {'_id': 5, 'item': 'XYZ', 'sizes': None}, ]) self.cmp.compare.aggregate([{'$unwind': {'path': '$sizes'}}]) self.cmp.compare.aggregate([ {'$unwind': {'path': '$sizes', 'includeArrayIndex': 'arrayIndex'}} ]) self.cmp.compare.aggregate([ {'$unwind': {'path': '$sizes', 'preserveNullAndEmptyArrays': True}}, ]) def test__aggregate_subtract_dates(self): self.cmp.compare.aggregate([{'$project': { '_id': 0, 'since': {'$subtract': ['$date', datetime.datetime(2014, 7, 4, 13, 0)]}, }}]) def test__aggregate_system_variables(self): self.cmp.do.drop() self.cmp.do.insert_many([ {'_id': 1}, {'_id': 2, 'parent_id': 1}, {'_id': 3, 'parent_id': 1}, ]) self.cmp.compare.aggregate([ {'$match': {'parent_id': {'$in': [1]}}}, {'$group': {'_id': 1, 'docs': {'$push': '$$ROOT'}}}, ]) def test__aggregate_date_operators(self): self.cmp.compare_ignore_order.aggregate([ {'$project': { 'doy': {'$dayOfYear': '$date'}, 'dom': {'$dayOfMonth': '$date'}, 'dow': {'$dayOfWeek': '$date'}, 'M': {'$month': '$date'}, 'w': {'$week': '$date'}, 'h': {'$hour': '$date'}, 'm': {'$minute': '$date'}, 's': {'$second': '$date'}, 'ms': {'$millisecond': '$date'}, }}, ]) def test__aggregate_in(self): self.cmp.compare_ignore_order.aggregate([ {'$project': { 'count': '$count', 'in': {'$in': ['$count', [1, 4, 5]]}, }}, ]) def test__aggregate_switch(self): self.cmp.compare_ignore_order.aggregate([ {'$project': { 'compare_with_3': { '$switch': { 'branches': [ {'case': {'$eq': ['$count', 3]}, 'then': 'equals 3'}, {'case': {'$gt': ['$count', 3]}, 'then': 'greater than 3'}, {'case': {'$lt': ['$count', 3]}, 'then': 'less than 3'} ], } }, 'equals_3': { '$switch': { 'branches': [ {'case': {'$eq': ['$count', 3]}, 'then': 'equals 3'}, ], 'default': 'not equal', } }, 'missing_field': { '$switch': { 'branches': [ {'case': '$missing_field', 'then': 'first case'}, {'case': True, 'then': '$missing_field'}, ], 'default': 'did not match', } }, }}, ]) def test__aggregate_switch_mongodb_to_bool(self): def build_switch(case): return { '$switch': { 'branches': [ {'case': case, 'then': 't'}, ], 'default': 'f', } } self.cmp.compare_ignore_order.aggregate([ {'$project': { 'undefined_value': build_switch('$not_existing_field'), 'false_value': build_switch(False), 'null_value': build_switch(None), 'zero_value': build_switch(0), 'true_value': build_switch(True), 'one_value': build_switch(1), 'empty_string': build_switch(''), 'empty_list': build_switch([]), 'empty_dict': build_switch({}), }}, ]) def test__aggregate_bug_473(self): """Regression test for bug https://github.com/mongomock/mongomock/issues/473.""" self.cmp.do.drop() self.cmp.do.insert_one({ 'name': 'first', 'base_value': 100, 'values_list': [ {'updated_value': 5}, {'updated_value': 15}, ], }) self.cmp.compare.aggregate([ {'$project': { 'name': 1, '_id': 0, 'sum': {'$sum': [ '$base_value', {'$arrayElemAt': ['$values_list.updated_value', -1]}, ]}, }}, ]) def test__aggregate_array_eleme_at(self): self.cmp.do.drop() self.cmp.do.insert_many([ {'values_list': [1, 2]}, {'values_list': [1, 2, 3]}, ]) self.cmp.compare.aggregate([{ '$project': { 'first_user_id': {'$arrayElemAt': ['$values_list', 2]}, 'other_user_id': {'$arrayElemAt': ['$values_list', -1]}, }, }]) def test_aggregate_bug_607(self): """Regression test for bug https://github.com/mongomock/mongomock/issues/607.""" self.cmp.do.drop() self.cmp.do.insert_one({ 'index': 2, 'values': [0, 1, 5] }) self.cmp.compare.aggregate([ {'$project': { 'values_index': {'$arrayElemAt': ['$values', '$index']} }} ]) self.cmp.compare.aggregate([ {'$project': { 'values_index': {'$arrayElemAt': ['$values', {'$add': [1, 1]}]} }} ]) def test__aggregate_first_last_in_array(self): self.cmp.do.drop() self.cmp.do.insert_one({ 'values': [0, 1, 5] }) self.cmp.compare.aggregate([ {'$project': { 'first': {'$first': '$values'}, 'last': {'$last': '$values'}, }} ]) def test__aggregate_cond_mongodb_to_bool(self): """Regression test for bug https://github.com/mongomock/mongomock/issues/650""" self.cmp.compare_ignore_order.aggregate([ {'$project': { # undefined aka KeyError 'undefined_value': {'$cond': ['$not_existing_field', 't', 'f']}, 'false_value': {'$cond': [False, 't', 'f']}, 'null_value': {'$cond': [None, 't', 'f']}, 'zero_value': {'$cond': [0, 't', 'f']}, 'true_value': {'$cond': [True, 't', 'f']}, 'one_value': {'$cond': [1, 't', 'f']}, 'empty_string': {'$cond': ['', 't', 'f']}, 'empty_list': {'$cond': [[], 't', 'f']}, 'empty_dict': {'$cond': [{}, 't', 'f']}, }}, ]) def test__aggregate_concatArrays(self): self.cmp.do.drop() self.cmp.do.insert_one({ '_id': 1, 'a': [1, 2], 'b': ['foo', 'bar', 'baz'], 'c': { 'arr1': [123] } }) pipeline = [{ '$project': { '_id': 0, 'concat': {'$concatArrays': ['$a', ['#', '*'], '$c.arr1', '$b']}, 'concat_array_expression': {'$concatArrays': '$b'}, 'concat_tuples': {'$concatArrays': ((1, 2, 3), (1,))}, 'concat_none': {'$concatArrays': None}, 'concat_missing_field': {'$concatArrays': '$foo'}, 'concat_none_item': {'$concatArrays': ['$a', None, '$b']}, 'concat_missing_field_item': {'$concatArrays': [[1, 2, 3], '$c.arr2']} } }] self.cmp.compare.aggregate(pipeline) def test__aggregate_concatArrays_exceptions(self): self.cmp.do.drop() self.cmp.do.insert_one({ '_id': 1, 'a': { 'arr1': [123] } }) self.cmp.compare_exceptions.aggregate([{ '$project': { 'concat_parameter_not_array': {'$concatArrays': 42} } }]) self.cmp.compare_exceptions.aggregate([{ '$project': { 'concat_item_not_array': {'$concatArrays': [[1, 2], '$a']} } }]) def test__aggregate_filter(self): self.cmp.do.drop() self.cmp.do.insert_many([ { '_id': 0, 'items': [ {'item_id': 43, 'quantity': 2, 'price': 10}, {'item_id': 2, 'quantity': 1, 'price': 240}, ], }, { '_id': 1, 'items': [ {'item_id': 23, 'quantity': 3, 'price': 110}, {'item_id': 103, 'quantity': 4, 'price': 5}, {'item_id': 38, 'quantity': 1, 'price': 300}, ], }, { '_id': 2, 'items': [ {'item_id': 4, 'quantity': 1, 'price': 23}, ], }, ]) self.cmp.compare.aggregate([{'$project': {'filtered_items': {'$filter': { 'input': '$items', 'as': 'item', 'cond': {'$gte': ['$$item.price', 100]}, }}}}]) self.cmp.compare.aggregate([{'$project': {'filtered_items': {'$filter': { 'input': '$items', 'cond': {'$lt': ['$$this.price', 100]}, }}}}]) def test__aggregate_map(self): self.cmp.do.insert_one({ 'array': [1, 2, 3, 4], }) self.cmp.compare.aggregate([{'$project': { '_id': 0, 'array': {'$map': { 'input': '$array', 'in': {'$multiply': ['$$this', '$$this']}, }}, 'custom_variable': {'$map': { 'input': '$array', 'as': 'self', 'in': {'$multiply': ['$$self', '$$self']}, }}, 'empty': {'$map': { 'input': [], 'in': {'$multiply': ['$$this', '$$this']}, }}, 'null': {'$map': { 'input': None, 'in': '$$this', }}, 'missing': {'$map': { 'input': '$missing.key', 'in': '$$this', }}, }}]) def test__aggregate_filter_in_arrayElemAt(self): self.cmp.do.drop() self.cmp.do.insert_many([ { '_id': 0, 'items': [ {'item_id': 11, 'category': 'book'}, {'item_id': 234, 'category': 'journal'} ] }, { '_id': 1, 'items': [ {'item_id': 23, 'category': 'book'} ] }, { '_id': 2, 'items': [ {'item_id': 232, 'category': 'book'} ] } ]) self.cmp.compare.aggregate([{'$project': {'item': { '$arrayElemAt': [ {'$filter': { 'input': '$items', 'cond': {'$eq': ['$$this.category', 'book']} }}, 0 ] }}}]) def test__aggregate_slice(self): self.cmp.do.drop() self.cmp.do.insert_many([ { '_id': 0, 'items': list(range(10)), }, { '_id': 1, 'items': list(range(10, 20)), }, { '_id': 2, 'items': list(range(20, 30)), }, ]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', 0] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', 5] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', 10] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', 0, 1] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', 0, 5] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', 5, 1] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', 5, 5] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', 0, 10000] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', -5] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', -10] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', -5, 5] }}}]) self.cmp.compare.aggregate([{'$project': {'slice': { '$slice': ['$items', -10, 5] }}}]) def test__aggregate_no_entries(self): pipeline = [ {'$match': {'a': {'$eq': 'Never going to happen'}}}, {'$out': 'new_collection'}, ] self.cmp.compare.aggregate(pipeline) cmp = self._create_compare_for_collection('new_collection') cmp.compare.find() def test__replace_root(self): self.cmp.do.drop() self.cmp.do.insert_many([ { '_id': 1, 'fruit': ['apples', 'oranges'], 'in_stock': {'oranges': 20, 'apples': 60}, 'on_order': {'oranges': 35, 'apples': 75}, }, { '_id': 2, 'vegetables': ['beets', 'yams'], 'in_stock': {'beets': 130, 'yams': 200}, 'on_order': {'beets': 90, 'yams': 145}, }, ]) self.cmp.compare.aggregate([{'$replaceRoot': {'newRoot': '$in_stock'}}]) def test__replace_root_new_document(self): self.cmp.do.drop() self.cmp.do.insert_many([ {'_id': 1, 'first_name': 'Gary', 'last_name': 'Sheffield', 'city': 'New York'}, {'_id': 2, 'first_name': 'Nancy', 'last_name': 'Walker', 'city': 'Anaheim'}, {'_id': 3, 'first_name': 'Peter', 'last_name': 'Sumner', 'city': 'Toledo'}, ]) self.cmp.compare.aggregate([{'$replaceRoot': {'newRoot': { 'full_name': {'$concat': ['$first_name', '$last_name']}, }}}]) def test__insert_date_with_timezone(self): self.cmp.do.insert_one({ 'dateNoTz': datetime.datetime(2000, 1, 1, 12, 30, 30, 12745), 'dateTz': datetime.datetime( 2000, 1, 1, 12, 30, 30, 12745, tzinfo=UTCPlus2()), }) self.cmp.compare.find_one() def test_aggregate_date_with_timezone(self): self.cmp.do.drop() self.cmp.do.insert_one({ 'start_date': datetime.datetime(2011, 11, 4, 0, 5, 23) }) pipeline = [ { '$addFields': { 'year': {'$year': { 'date': '$start_date', 'timezone': 'America/New_York'} }, 'week': {'$week': { 'date': '$start_date', 'timezone': 'America/New_York'} }, 'dayOfWeek': {'$dayOfWeek': { 'date': '$start_date', 'timezone': 'America/New_York'} }, } }, {'$project': {'_id': 0}}, ] self.cmp.compare.aggregate(pipeline) def test__aggregate_add_fields(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'a': 1, 'b': 2}, {}, {'nested': {'foo': 1}}, {'nested': 'not nested'}, ]) self.cmp.compare.aggregate([{'$addFields': { 'a': 3, 'c': {'$sum': [3, '$a', '$b']}, 'd': '$d', 'nested.foo': 5, }}]) def test__aggregate_add_fields_with_max_min(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'_id': 4, 'dates': [ datetime.datetime(2020, 1, 10), datetime.datetime(2020, 1, 5), datetime.datetime(2020, 1, 7) ]}, {'_id': 5, 'dates': []} ]) pipeline = [ {'$addFields': { 'max_date': {'$max': '$dates'}, 'min_date': {'$min': '$dates'} }} ] self.cmp.compare.aggregate(pipeline) def test__aggregate36(self): self.cmp.compare.aggregate([{'$project': {'c': {'$abs': -2}}}]) self.cmp.compare.aggregate([{'$project': {'d': {'$floor': 2.3}}}]) self.cmp.compare.aggregate([{'$project': {'e': {'$ln': None}}}]) self.cmp.compare.aggregate([{'$project': {'f': {'$exp': '$non_existent_key'}}}]) self.cmp.compare.aggregate([{'$project': {'g': {'$divide': [7, 3]}}}]) self.cmp.compare.aggregate([{'$project': {'h': {'$log': [None, 1]}}}]) self.cmp.compare.aggregate([{'$project': {'i': {'$mod': [1, None]}}}]) self.cmp.compare.aggregate([{'$project': {'j': {'$pow': [None, None]}}}]) self.cmp.compare.aggregate([{'$project': {'k': {'$subtract': [None, 1]}}}]) self.cmp.compare.aggregate([{'$project': {'k': {'$subtract': ['$non_existent_key', 1]}}}]) self.cmp.compare.aggregate([{'$project': {'o': {'$multiply': [4]}}}]) self.cmp.compare.aggregate([{'$project': {'p': {'$add': [1, 2, 3]}}}]) self.cmp.compare.aggregate([{'$project': {'s': {'$multiply': [1, None]}}}]) self.cmp.compare.aggregate([{'$project': {'t': {'$add': [None, 1]}}}]) self.cmp.compare.aggregate([{'$project': {'u': {'$multiply': ['$a', '$b', 4]}}}]) def test__aggregate_exception(self): self.cmp.compare_exceptions.aggregate([{'$project': {'c': {'$abs': [-2, 4]}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'c': {'$floor': []}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'c': {'$divide': 5}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'c': {'$log': [5]}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'c': {'$mod': [5, 3, 1]}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'c': {'$sum': []}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'c': {'$multiply': []}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'n': {'$add': '$a'}}}]) self.cmp.compare_exceptions.aggregate( [{'$project': {'q': {'$multiply': [1, '$non_existent_key']}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'r': {'$add': '$non_existent_key'}}}]) self.cmp.compare_exceptions.aggregate([{'$project': {'v': {'$multiply': '$b'}}}]) # TODO(pascal): Enable this test, for now it's not the same kind of error. # self.cmp.compare_exceptions.aggregate( # [{'$project': {'c': {'$add': ['$date', 1, '$date']}}}]) def test__aggregate_add_fields_with_sum_avg(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'_id': 4, 'values': [10, 5, 7]}, {'_id': 5, 'values': []} ]) pipeline = [ {'$addFields': { 'max_val': {'$sum': '$values'}, 'min_val': {'$avg': '$values'} }} ] self.cmp.compare.aggregate(pipeline) def test_aggregate_to_string(self): self.cmp.do.drop() self.cmp.do.insert_one({ '_id': ObjectId('5dd6a8f302c91829ef248162'), 'boolean_true': True, 'boolean_false': False, 'integer': 100, 'date': datetime.datetime(2018, 3, 27, 0, 58, 51, 538000), }) pipeline = [ { '$addFields': { '_id': {'$toString': '$_id'}, 'boolean_true': {'$toString': '$boolean_true'}, 'boolean_false': {'$toString': '$boolean_false'}, 'integer': {'$toString': '$integer'}, 'date': {'$toString': '$date'}, 'none': {'$toString': '$notexist'} } } ] self.cmp.compare.aggregate(pipeline) def test__aggregate_to_decimal(self): self.cmp.do.drop() self.cmp.do.insert_one({ 'boolean_true': True, 'boolean_false': False, 'integer': 100, 'double': 1.999, 'decimal': decimal128.Decimal128('5.5000'), 'str_base_10_numeric': '123', 'str_negative_number': '-23', 'str_decimal_number': '1.99', 'str_not_numeric': '123a123', 'datetime': datetime.datetime.utcfromtimestamp(0), }) pipeline = [ { '$addFields': { 'boolean_true': {'$toDecimal': '$boolean_true'}, 'boolean_false': {'$toDecimal': '$boolean_false'}, 'integer': {'$toDecimal': '$integer'}, 'double': {'$toDecimal': '$double'}, 'decimal': {'$toDecimal': '$decimal'}, 'str_base_10_numeric': {'$toDecimal': '$str_base_10_numeric'}, 'str_negative_number': {'$toDecimal': '$str_negative_number'}, 'str_decimal_number': {'$toDecimal': '$str_decimal_number'}, 'datetime': {'$toDecimal': '$datetime'}, 'not_exist_field': {'$toDecimal': '$not_exist_field'}, } }, {'$project': {'_id': 0}}, ] self.cmp.compare.aggregate(pipeline) def test_aggregate_to_int(self): self.cmp.do.drop() self.cmp.do.insert_one({ 'boolean_true': True, 'boolean_false': False, 'integer': 100, 'double': 1.999, 'decimal': decimal128.Decimal128('5.5000') }) pipeline = [ { '$addFields': { 'boolean_true': {'$toInt': '$boolean_true'}, 'boolean_false': {'$toInt': '$boolean_false'}, 'integer': {'$toInt': '$integer'}, 'double': {'$toInt': '$double'}, 'decimal': {'$toInt': '$decimal'}, 'not_exist': {'$toInt': '$not_exist'}, } }, { '$project': { '_id': 0 } } ] self.cmp.compare.aggregate(pipeline) def test_aggregate_to_long(self): self.cmp.do.drop() self.cmp.do.insert_one({ 'boolean_true': True, 'boolean_false': False, 'integer': 100, 'double': 1.999, 'decimal': decimal128.Decimal128('5.5000') }) pipeline = [ { '$addFields': { 'boolean_true': {'$toLong': '$boolean_true'}, 'boolean_false': {'$toLong': '$boolean_false'}, 'integer': {'$toLong': '$integer'}, 'double': {'$toLong': '$double'}, 'decimal': {'$toLong': '$decimal'}, 'not_exist': {'$toLong': '$not_exist'}, } }, { '$project': { '_id': 0 } } ] self.cmp.compare.aggregate(pipeline) def test_aggregate_date_to_string(self): self.cmp.do.drop() self.cmp.do.insert_one({ 'start_date': datetime.datetime(2011, 11, 4, 0, 5, 23) }) pipeline = [ { '$addFields': { 'start_date': { '$dateToString': {'format': '%Y/%m/%d %H:%M', 'date': '$start_date'} } } }, {'$project': {'_id': 0}}, ] self.cmp.compare.aggregate(pipeline) def test_aggregate_date_from_parts(self): self.cmp.do.drop() self.cmp.do.insert_one({ 'start_date': datetime.datetime(2022, 8, 3, 16, 6, 0) }) additional_fields_pipeline = [ { '$addFields': { 'start_date': { '$dateFromParts': { 'year': {'$year': '$start_date'}, 'month': {'$month': '$start_date'}, 'day': {'$dayOfMonth': '$start_date'}, } } } }, {'$project': {'_id': 0}}, ] self.cmp.compare.aggregate(additional_fields_pipeline) def test_aggregate_array_to_object(self): self.cmp.do.drop() self.cmp.do.insert_many([{ 'items': [['a', 1], ['b', 2], ['c', 3], ['a', 4]] }, { 'items': (['a', 1], ['b', 2], ['c', 3], ['a', 4]) }, { 'items': [('a', 1), ('b', 2), ('c', 3), ('a', 4)] }, { 'items': (('a', 1), ('b', 2), ('c', 3), ('a', 4)) }, { 'items': [['a', 1], ('b', 2), ['c', 3], ('a', 4)] }, { 'items': (['a', 1], ('b', 2), ['c', 3], ('a', 4)) }, { 'items': [{'k': 'a', 'v': 1}, {'k': 'b', 'v': 2}, {'k': 'c', 'v': 3}, {'k': 'a', 'v': 4}] }, { 'items': [] }, { 'items': () }, { 'items': None }]) pipeline = [ { '$project': { 'items': { '$arrayToObject': '$items' }, 'not_exists': { '$arrayToObject': '$nothing' } } }, {'$project': {'_id': 0}}, ] self.cmp.compare.aggregate(pipeline) # All of these items should trigger an error items = [[ {'$addFields': {'items': ''}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ], [ {'$addFields': {'items': 100}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ], [ {'$addFields': {'items': [['a', 'b', 'c'], ['d', 2]]}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ], [ {'$addFields': {'items': [['a'], ['b', 2]]}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ], [ {'$addFields': {'items': [[]]}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ], [ {'$addFields': {'items': [{'k': 'a', 'v': 1, 't': 't'}, {'k': 'b', 'v': 2}]}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ], [ {'$addFields': {'items': [{'v': 1, 't': 't'}]}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ], [ {'$addFields': {'items': [{}]}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ], [ {'$addFields': {'items': [['a', 1], {'k': 'b', 'v': 2}]}}, {'$project': {'items': {'$arrayToObject': '$items'}, '_id': 0}} ]] for item in items: self.cmp.compare_exceptions.aggregate(item) def test__create_duplicate_index(self): self.cmp.do.create_index([('value', 1)]) self.cmp.do.create_index([('value', 1)]) self.cmp.compare_exceptions.create_index([('value', 1)], unique=True) def test__partial_filter_expression_unique_index(self): self.cmp.do.delete_many({}) self.cmp.do.create_index( (('value', 1), ('partialFilterExpression_value', 1)), unique=True, partialFilterExpression={'partialFilterExpression_value': {'$exists': True}}) # We should be able to add documents with duplicated `value` if # partialFilterExpression_value isn't set. self.cmp.do.insert_one({'value': 4}) self.cmp.do.insert_one({'value': 4}) self.cmp.compare.find({'value': 4}) # We should be able to add documents with distinct `value` values and duplicated # `partialFilterExpression_value` value. self.cmp.do.insert_one({'partialFilterExpression_value': 1, 'value': 2}) self.cmp.do.insert_one({'partialFilterExpression_value': 1, 'value': 3}) self.cmp.compare.find({'partialFilterExpression_value': 1}) # We should not be able to add documents with duplicated `partialFilterExpression_value` # and `value` values. self.cmp.do.insert_one({'partialFilterExpression_value': 2, 'value': 3}) self.cmp.compare_exceptions.insert_one({'partialFilterExpression_value': 2, 'value': 3}) self.cmp.compare.find({'partialFilterExpression_value': 2, 'value': 3}) self.cmp.compare.find({}) def test_aggregate_project_with_boolean(self): self.cmp.do.drop() # Test with no items self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': []}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': []}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$not': {}}}} ]) # Tests following are with one item self.cmp.do.insert_one({ 'items': [] }) # Test with 0 arguments self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': []}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': []}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$not': {}}}} ]) # Test with one argument self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': [True]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': [True]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$not': True}}} ]) # Test with two arguments self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': [True, True]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': [False, True]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': [True, False]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': [False, False]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': [True, True]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': [False, True]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': [True, False]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': [False, False]}}} ]) # Following tests are with more than two items self.cmp.do.insert_many([ {'items': []}, {'items': []} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': []}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': []}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$not': {}}}} ]) # Test with something else than boolean self.cmp.do.insert_one({ 'items': ['foo'] }) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$and': [{'$eq': ['$items', ['foo']]}]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$or': [{'$eq': ['$items', ['foo']]}]}}} ]) self.cmp.compare.aggregate([ {'$project': {'_id': 0, 'items': {'$not': {'$eq': ['$items', ['foo']]}}}} ]) def test__aggregate_project_missing_fields(self): self.cmp.do.insert_one({'_id': 1, 'arr': {'a': 2, 'b': 3}}) self.cmp.compare.aggregate([ {'$match': {'_id': 1}}, {'$project': OrderedDict([ ('_id', False), ('rename_dot', '$arr.c'), ('a', '$arr.a') ])} ]) def test__aggregate_graph_lookup_missing_field(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'_id': ObjectId(), 'name': 'a', 'child': 'b', 'val': 2}, {'_id': ObjectId(), 'name': 'b', 'child': 'c', 'val': 3}, {'_id': ObjectId(), 'name': 'c', 'child': None, 'val': 4}, {'_id': ObjectId(), 'name': 'd', 'child': 'a', 'val': 5} ]) pipeline = [ {'$match': {'name': 'a'}}, {'$graphLookup': { 'from': self.collection_name, 'startWith': '$fieldThatDoesNotExist', 'connectFromField': 'child', 'connectToField': 'name', 'as': 'lookup' }}, {'$unwind': '$lookup'}, {'$sort': {'lookup.name': 1}} ] self.cmp.compare.aggregate(pipeline) pipeline = [ {'$match': {'name': 'a'}}, {'$graphLookup': { 'from': self.collection_name, 'startWith': {'$concat': ['a', '$fieldThatDoesNotExist']}, 'connectFromField': 'child', 'connectToField': 'name', 'as': 'lookup' }}, {'$unwind': '$lookup'}, {'$sort': {'lookup.name': 1}} ] self.cmp.compare.aggregate(pipeline) def test__aggregate_merge_objects(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'_id': ObjectId(), 'a': '1', 'b': {'c': '1', 'd': 2}}, {'_id': ObjectId(), 'a': '1', 'b': {'e': 3, 'f': '4'}}, {'_id': ObjectId(), 'a': '1', 'c': '2'}, {'_id': ObjectId(), 'a': '1', 'b': None}, {'_id': ObjectId(), 'a': 2, 'b': None}, {'_id': ObjectId(), 'a': 2, 'b': {'c': None, 'd': 6}}, {'_id': ObjectId(), 'a': 2, 'b': {'c': '7', 'd': None, 'e': 9, 'f': '10'}}, {'_id': ObjectId(), 'a': 3, 'b': None}, {'_id': ObjectId(), 'a': 3, 'b': dict()}, {'_id': ObjectId(), 'a': 4, 'b': None}, ]) pipeline = [ {'$group': { '_id': '$a', 'merged_b': {'$mergeObjects': '$b'}, }} ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__add_fields(self): self.cmp.compare.aggregate([{'$addFields': {'c': 3}}]) self.cmp.compare.aggregate([{'$addFields': {'c': 4}}]) self.cmp.compare.aggregate([{'$addFields': {'b': {'$add': ['$a', '$b', 5]}}}]) def test__aggregate_with_missing_fields1(self): self.cmp.do.delete_many({}) data = [ {'_id': ObjectId(), 'a': 0, 'b': 1}, {'_id': ObjectId(), 'a': 0}, {'_id': ObjectId()}, ] self.cmp.do.insert_many(data) pipeline = [ {'$group': {'_id': '$a', 'b': {'$sum': '$b'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__group_with_missing_fields1(self): self.cmp.do.delete_many({}) data = [ {'_id': ObjectId(), 'a': 0, 'b': 0}, {'_id': ObjectId(), 'a': 0}, {'_id': ObjectId(), 'b': 0}, {'_id': ObjectId()}, ] self.cmp.do.insert_many(data) pipeline = [ {'$group': {'_id': {'a': '$a', 'b': '$b'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__group_with_missing_fields2(self): self.cmp.do.delete_many({}) data = [ {'_id': ObjectId(), 'a': 0}, {'_id': ObjectId()}, ] self.cmp.do.insert_many(data) pipeline = [ {'$group': {'_id': {'a': '$a'}}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__group_with_missing_fields3(self): self.cmp.do.delete_many({}) data = [ {'_id': ObjectId(), 'a': 0}, {'_id': ObjectId()}, ] self.cmp.do.insert_many(data) pipeline = [ {'$group': {'_id': '$a'}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) def test__add_fields_with_missing_fields(self): self.cmp.do.delete_many({}) data = [ {'a': 0}, {}, ] self.cmp.do.insert_many(data) pipeline = [ {'$addFields': {'b': '$a'}}, ] self.cmp.compare_ignore_order.aggregate(pipeline) @skipIf(not helpers.HAVE_PYMONGO, 'pymongo not installed') class MongoClientGraphLookupTest(_CollectionComparisonTest): def setUp(self): super(MongoClientGraphLookupTest, self).setUp() self.cmp_a = self._create_compare_for_collection('data_a') self.cmp_b = self._create_compare_for_collection('data_b') def test_graphlookup_basic(self): data_a = [ {'_id': 0, 'airport': 'JFK', 'connects': ['BOS', 'ORD']}, {'_id': 1, 'airport': 'BOS', 'connects': ['JFK', 'PWM']}, {'_id': 2, 'airport': 'ORD', 'connects': ['JFK']}, {'_id': 3, 'airport': 'PWM', 'connects': ['BOS', 'LHR']}, {'_id': 4, 'airport': 'LHR', 'connects': ['PWM']}, ] data_b = [ {'_id': 1, 'name': 'Dev', 'nearestAirport': 'JFK'}, {'_id': 2, 'name': 'Eliot', 'nearestAirport': 'JFK'}, {'_id': 3, 'name': 'Jeff', 'nearestAirport': 'BOS'}, ] query = [ { '$graphLookup': { 'from': 'a', 'startWith': '$nearestAirport', 'connectFromField': 'connects', 'connectToField': 'airport', 'maxDepth': 2, 'depthField': 'numConnections', 'as': 'destinations' } } ] self.cmp_a.do.insert_many(data_a) self.cmp_b.do.insert_many(data_b) self.cmp_b.compare.aggregate(query) def test_graphlookup_nested_array(self): data_a = [ {'_id': 0, 'airport': 'JFK', 'connects': [ {'to': 'BOS', 'distance': 200}, {'to': 'ORD', 'distance': 800}]}, {'_id': 1, 'airport': 'BOS', 'connects': [ {'to': 'JFK', 'distance': 200}, {'to': 'PWM', 'distance': 2000}]}, {'_id': 2, 'airport': 'ORD', 'connects': [{'to': 'JFK', 'distance': 800}]}, {'_id': 3, 'airport': 'PWM', 'connects': [ {'to': 'BOS', 'distance': 2000}, {'to': 'LHR', 'distance': 6000}]}, {'_id': 4, 'airport': 'LHR', 'connects': [{'to': 'PWM', 'distance': 6000}]}, ] data_b = [ {'_id': 1, 'name': 'Dev', 'nearestAirport': 'JFK'}, {'_id': 2, 'name': 'Eliot', 'nearestAirport': 'JFK'}, {'_id': 3, 'name': 'Jeff', 'nearestAirport': 'BOS'}, ] query = [ { '$graphLookup': { 'from': 'a', 'startWith': '$nearestAirport', 'connectFromField': 'connects.to', 'connectToField': 'airport', 'maxDepth': 2, 'depthField': 'numConnections', 'as': 'destinations' } } ] self.cmp_a.do.insert_many(data_a) self.cmp_b.do.insert_many(data_b) self.cmp_b.compare.aggregate(query) def test_graphlookup_nested_dict(self): data_b = [ {'_id': 1, 'name': 'Dev'}, {'_id': 2, 'name': 'Eliot', 'reportsTo': { 'name': 'Dev', 'from': '2016-01-01T00:00:00.000Z'}}, {'_id': 3, 'name': 'Ron', 'reportsTo': {'name': 'Eliot', 'from': '2016-01-01T00:00:00.000Z'}}, {'_id': 4, 'name': 'Andrew', 'reportsTo': { 'name': 'Eliot', 'from': '2016-01-01T00:00:00.000Z'}}, {'_id': 5, 'name': 'Asya', 'reportsTo': { 'name': 'Ron', 'from': '2016-01-01T00:00:00.000Z'}}, {'_id': 6, 'name': 'Dan', 'reportsTo': {'name': 'Andrew', 'from': '2016-01-01T00:00:00.000Z'}}, ] data_a = [{'_id': 1, 'name': 'x'}] query = [ { '$graphLookup': { 'from': 'b', 'startWith': '$name', 'connectFromField': 'reportsTo.name', 'connectToField': 'name', 'as': 'reportingHierarchy' } } ] self.cmp_a.do.insert_many(data_a) self.cmp_b.do.insert_many(data_b) self.cmp_b.compare.aggregate(query) def test__aggregate_let(self): self.cmp.do.insert_many([ {'_id': 1, 'price': 10, 'tax': 0.50, 'applyDiscount': True}, {'_id': 2, 'price': 10, 'tax': 0.25, 'applyDiscount': False}, ]) self.cmp.compare.aggregate([{'$project': { 'finalTotal': { '$let': { 'vars': { 'total': {'$add': ['$price', '$tax']}, 'discounted': {'$cond': {'if': '$applyDiscount', 'then': 0.9, 'else': 1}}, }, 'in': {'$multiply': ['$$total', '$$discounted']}, }, }, }}]) def test__aggregate_let_errors(self): self.cmp.do.insert_many([ {'_id': 1, 'price': 10, 'tax': 0.50, 'applyDiscount': True}, {'_id': 2, 'price': 10, 'tax': 0.25, 'applyDiscount': False}, ]) self.cmp.compare_exceptions.aggregate([{'$project': { 'finalTotal': { '$let': [{'total': 3}, {'$$total'}], }, }}]) self.cmp.compare_exceptions.aggregate([{'$project': { 'finalTotal': { '$let': { 'in': {'$multiply': ['4', '3']}, }, }, }}]) self.cmp.compare_exceptions.aggregate([{'$project': { 'finalTotal': { '$let': { 'vars': ['total', 'discounted'], 'in': {'$multiply': ['$$total', '$$discounted']}, }, }, }}]) def _LIMIT(*args): return lambda cursor: cursor.limit(*args) def _SORT(*args): return lambda cursor: cursor.sort(*args) def _COUNT(cursor): return cursor.count() def _COUNT_EXCEPTION_TYPE(cursor): try: cursor.count() except Exception as error: return str(type(error)) assert False, 'Count should have failed' def _DISTINCT(*args): def sortkey(value): if isinstance(value, dict): return [(k, sortkey(v)) for k, v in sorted(value.items())] return value return lambda cursor: sorted(cursor.distinct(*args), key=sortkey) def _SKIP(*args): return lambda cursor: cursor.skip(*args) class MongoClientSortSkipLimitTest(_CollectionComparisonTest): def setUp(self): super(MongoClientSortSkipLimitTest, self).setUp() self.cmp.do.insert_many([{'_id': i, 'index': i} for i in range(30)]) def test__skip(self): self.cmp.compare(_SORT('index', 1), _SKIP(10)).find() def test__skipped_find(self): self.cmp.compare(_SORT('index', 1)).find(skip=10) def test__limit(self): self.cmp.compare(_SORT('index', 1), _LIMIT(10)).find() def test__negative_limit(self): self.cmp.compare(_SORT('index', 1), _LIMIT(-10)).find() def test__skip_and_limit(self): self.cmp.compare(_SORT('index', 1), _SKIP(10), _LIMIT(10)).find() @skipIf( helpers.PYMONGO_VERSION >= version.parse('4.0'), 'Cursor.count was removed in pymongo 4') def test__count(self): self.cmp.compare(_COUNT).find() @skipUnless( helpers.PYMONGO_VERSION >= version.parse('4.0'), 'Cursor.count was removed in pymongo 4') def test__count_fail(self): self.cmp.compare(_COUNT_EXCEPTION_TYPE).find() def test__sort_name(self): self.cmp.do.delete_many({}) for data in ({'a': 1, 'b': 3, 'c': 'data1'}, {'a': 2, 'b': 2, 'c': 'data3'}, {'a': 3, 'b': 1, 'c': 'data2'}): self.cmp.do.insert_one(data) self.cmp.compare(_SORT('a')).find() self.cmp.compare(_SORT('b')).find() def test__sort_name_nested_doc(self): self.cmp.do.delete_many({}) for data in ({'root': {'a': 1, 'b': 3, 'c': 'data1'}}, {'root': {'a': 2, 'b': 2, 'c': 'data3'}}, {'root': {'a': 3, 'b': 1, 'c': 'data2'}}): self.cmp.do.insert_one(data) self.cmp.compare(_SORT('root.a')).find() self.cmp.compare(_SORT('root.b')).find() def test__sort_name_nested_list(self): self.cmp.do.delete_many({}) for data in ({'root': [{'a': 1, 'b': 3, 'c': 'data1'}]}, {'root': [{'a': 2, 'b': 2, 'c': 'data3'}]}, {'root': [{'a': 3, 'b': 1, 'c': 'data2'}]}): self.cmp.do.insert_one(data) self.cmp.compare(_SORT('root.0.a')).find() self.cmp.compare(_SORT('root.0.b')).find() def test__sort_list(self): self.cmp.do.delete_many({}) for data in ({'a': 1, 'b': 3, 'c': 'data1'}, {'a': 2, 'b': 2, 'c': 'data3'}, {'a': 3, 'b': 1, 'c': 'data2'}): self.cmp.do.insert_one(data) self.cmp.compare(_SORT([('a', 1), ('b', -1)])).find() self.cmp.compare(_SORT([('b', 1), ('a', -1)])).find() self.cmp.compare(_SORT([('b', 1), ('a', -1), ('c', 1)])).find() def test__sort_list_nested_doc(self): self.cmp.do.delete_many({}) for data in ({'root': {'a': 1, 'b': 3, 'c': 'data1'}}, {'root': {'a': 2, 'b': 2, 'c': 'data3'}}, {'root': {'a': 3, 'b': 1, 'c': 'data2'}}): self.cmp.do.insert_one(data) self.cmp.compare(_SORT([('root.a', 1), ('root.b', -1)])).find() self.cmp.compare(_SORT([('root.b', 1), ('root.a', -1)])).find() self.cmp.compare( _SORT([('root.b', 1), ('root.a', -1), ('root.c', 1)])).find() def test__sort_list_nested_list(self): self.cmp.do.delete_many({}) for data in ({'root': [{'a': 1, 'b': 3, 'c': 'data1'}]}, {'root': [{'a': 2, 'b': 2, 'c': 'data3'}]}, {'root': [{'a': 3, 'b': 1, 'c': 'data2'}]}): self.cmp.do.insert_one(data) self.cmp.compare(_SORT([('root.0.a', 1), ('root.0.b', -1)])).find() self.cmp.compare(_SORT([('root.0.b', 1), ('root.0.a', -1)])).find() self.cmp.compare( _SORT( [('root.0.b', 1), ('root.0.a', -1), ('root.0.c', 1)])).find() def test__sort_dict(self): self.cmp.do.delete_many({}) self.cmp.do.insert_many([ {'a': 1, 'b': OrderedDict([('value', 1), ('other', True)])}, {'a': 2, 'b': OrderedDict([('value', 3)])}, {'a': 3, 'b': OrderedDict([('value', 2), ('other', False)])}, ]) self.cmp.compare(_SORT('b')).find() def test__close(self): # Does nothing - just make sure it exists and takes the right args self.cmp.do(lambda cursor: cursor.close()).find() def test__distinct_nested_field(self): self.cmp.do.insert_one({'f1': {'f2': 'v'}}) self.cmp.compare(_DISTINCT('f1.f2')).find() def test__distinct_array_field(self): self.cmp.do.insert_many( [{'f1': ['v1', 'v2', 'v1']}, {'f1': ['v2', 'v3']}]) self.cmp.compare(_DISTINCT('f1')).find() def test__distinct_array_nested_field(self): self.cmp.do.insert_one({'f1': [{'f2': 'v'}, {'f2': 'w'}]}) self.cmp.compare(_DISTINCT('f1.f2')).find() def test__distinct_array_field_with_dicts(self): self.cmp.do.insert_many([ {'f1': [{'f2': 'v2'}, {'f3': 'v3'}]}, {'f1': [{'f3': 'v3'}, {'f4': 'v4'}]}, ]) self.cmp.compare(_DISTINCT('f1')).find() class InsertedDocumentTest(TestCase): def setUp(self): super(InsertedDocumentTest, self).setUp() self.collection = mongomock.MongoClient().db.collection self.data = {'a': 1, 'b': [1, 2, 3], 'c': {'d': 4}} self.orig_data = copy.deepcopy(self.data) self.object_id = self.collection.insert_one(self.data).inserted_id def test__object_is_consistent(self): [object] = self.collection.find() self.assertEqual(object['_id'], self.object_id) def test__find_by_id(self): [object] = self.collection.find({'_id': self.object_id}) self.assertEqual(object, self.data) @skipIf( helpers.PYMONGO_VERSION and helpers.PYMONGO_VERSION >= version.parse('4.0'), 'remove was removed in pymongo v4') def test__remove_by_id(self): self.collection.remove(self.object_id) self.assertEqual(0, self.collection.count_documents({})) def test__inserting_changes_argument(self): # Like pymongo, we should fill the _id in the inserted dict # (odd behavior, but we need to stick to it) self.assertEqual(self.data, dict(self.orig_data, _id=self.object_id)) def test__data_is_copied(self): [object] = self.collection.find() self.assertEqual(dict(self.orig_data, _id=self.object_id), object) self.data.pop('a') self.data['b'].append(5) self.assertEqual(dict(self.orig_data, _id=self.object_id), object) [object] = self.collection.find() self.assertEqual(dict(self.orig_data, _id=self.object_id), object) def test__find_returns_copied_object(self): [object1] = self.collection.find() [object2] = self.collection.find() self.assertEqual(object1, object2) self.assertIsNot(object1, object2) object1['b'].append('bla') self.assertNotEqual(object1, object2) class ObjectIdTest(TestCase): def test__equal_with_same_id(self): obj1 = ObjectId() obj2 = ObjectId(str(obj1)) self.assertEqual(obj1, obj2) class MongoClientTest(_CollectionComparisonTest): """Compares a fake connection with the real mongo connection implementation This is done via cross-comparison of the results. """ def setUp(self): super(MongoClientTest, self).setUp() self.cmp = MultiCollection({'fake': self.fake_conn, 'real': self.mongo_conn}) def test__database_names(self): if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.database_names() return self.cmp.do.database_names() class DatabaseTest(_CollectionComparisonTest): """Compares a fake database with the real mongo database implementation This is done via cross-comparison of the results. """ def setUp(self): super(DatabaseTest, self).setUp() self.cmp = MultiCollection({ 'fake': self.fake_conn[self.db_name], 'real': self.mongo_conn[self.db_name], }) def test__database_names(self): if helpers.PYMONGO_VERSION >= version.parse('4.0'): self.cmp.compare_exceptions.collection_names() return self.cmp.do.collection_names()