def test_insert_returns_errors_if_error_ocurred_during_insertion(self): Collection.collection.create_index('test', unique=True) Collection.insert_one({'test': 'test'}) errors = Collection({'test': 'test'}).insert() self.assertEqual(errors['error_code'], 0) self.assertEqual(errors['error_type'], 'PyMongoError') Collection.drop_index('test_1')
def test_save_returns_errors_if_error_ocurred_during_save(self): Collection.collection.create_index('test', unique=True) Collection.insert_one({'test': 'test'}) document = Document(Collection) document.test = 'test' errors = document.save() self.assertEqual(errors['error_code'], 0) self.assertEqual(errors['error_type'], 'PyMongoError') Collection.drop_index('test_1')
def test_update_updates_calls_cascade_and_returns_updated_document_if_document_is_valid( self, after_update): _id = Collection.insert_one({'test': 'test1'}) updated = Collection({'_id': _id, 'test': 'test2'}).update() self.assertEqual(updated, {'_id': _id, 'test': 'test2'}) self.assertEqual(after_update.call_count, 1)
def test_update_returns_errors_if_error_ocurred_during_save(self): Collection.collection.create_index('test', unique=True) Collection.insert_one({'test': 'test1'}) _id = Collection.insert_one({'test': 'test2'}) errors = Document(Collection, {'_id': _id, 'test': 'test1'}).update() errors.pop('error_message') self.assertEqual( errors, { 'error_code': 0, 'error_type': 'PyMongoError', 'operation': 'update', 'collection': 'Collection', 'document': {'_id': _id, 'test': 'test1'} } ) Collection.drop_index('test_1')
def test_update_returns_errors_if_error_ocurred_during_update(self): Collection.collection.create_index('test', unique=True) Collection.insert_one({'test': 'test1'}) _id = Collection.insert_one({'test': 'test2'}) errors = Collection({'_id': _id, 'test': 'test1'}).update() errors.pop('error_message') self.assertEqual( errors, { 'error_code': 0, 'error_type': 'PyMongoError', 'operation': 'update', 'collection': 'Collection', 'document': { '_id': _id, 'test': 'test1' } }) Collection.drop_index('test_1')
def test_update_updates_calls_cascade_and_returns_updated_document_if_document_is_valid(self, after_update): _id = Collection.insert_one({'test': 'test1'}) updated = Collection({'_id': _id, 'test': 'test2'}).update() self.assertEqual(updated, {'_id': _id, 'test': 'test2'}) self.assertEqual(after_update.call_count, 1)