def test_batch_update(self):
self._perform_batch_insertion()
batch = Batch(self.collection, self.connection)
batch.add_update(
{
'batch_insert_1': 3
},
{
"$inc": {
"bar": 1
}
}
)
batch.add_update(
{
'batch_insert_2': 'banana'
},
{
'$set': {
'banana': 'tasty'
}
}
)
batch.execute()
retrieved_document_1, retrieved_document_2 = self._retrieve_documents()
self.assertEqual(3, retrieved_document_1['bar'])
self.assertEqual('tasty', retrieved_document_2['banana'])
def test_batch_update_with_invalid_characters(self):
self._perform_batch_insertion()
batch = Batch(self.collection, self.connection)
batch.add_update(
{
'batch_insert_1': 3
},
{
"$inc": {
"bar": 1
}
}
)
exception = False
try:
batch.execute()
except Exception:
exception = True
self.assertFalse(exception)
