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))