def test_ensure_indexes_method_is_called_for_all_managers(self):
from catalog.mongomodels import ensure_all_indexes
mocker_manager = self.mocker.mock()
mocker_manager._ensure_indexes()
self.mocker.result(None)
self.mocker.replay()
self.assertIsNone(ensure_all_indexes(managers=[mocker_manager]))
def rebuild_collection(self, collection, new_data):
"""
Drops the collection ``collection`` and rebuild it
using ``new_data``.
``collection`` is a string of the collection name
``new_data`` is an iterable containing the data to
be loaded after the collection is recreated.
"""
self._mongoconn.db.drop_collection(collection)
col = self._mongoconn.db[collection]
mongomodels.ensure_all_indexes()
for data in new_data:
col.insert(data, w=1)
return None
def test_autodiscover_is_used_when_managers_is_not_provided(self):
from catalog.mongomodels import ensure_all_indexes
mocker_autodiscover = self.mocker.mock()
mocker_manager = self.mocker.mock()
mocker_autodiscover()
self.mocker.result([mocker_manager])
mocker_manager._ensure_indexes()
self.mocker.result(None)
self.mocker.replay()
self.assertIsNone(ensure_all_indexes(autodiscover=mocker_autodiscover))
