def test_batch_one_remove(self):
self._perform_batch_insertion()
batch = Batch(self.collection, self.connection)
batch.add_remove({
'batch_insert_1': 3
})
batch.execute()
self.assertFalse(self.collection.find({'batch_insert_1': 3}))
self.assertTrue(self.collection.find({'batch_insert_3': 29}))
def test_batch_find_with_invalid_characters(self):
self._perform_batch_insertion()
batch = Batch(self.collection, self.connection)
batch.add_find({
'bar': 'ben & company'
})
self.assertEqual(0, len(batch.find()[0]))
def test_batch_insert_with_invalid_characters(self):
batch = Batch(self.collection, self.connection)
batch.add_insert({
'batch_insert_5': 'Ben & Company',
'bar': 2
})
batch.execute()
self.assertTrue(self.collection.find_one({'batch_insert_5': 'Ben & Company'}))
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)
def _perform_batch_insertion(self):
batch = Batch(self.collection, self.connection)
batch.add_insert({
'batch_insert_1': 3,
'bar': 2
})
batch.add_insert({
'batch_insert_2': 'banana',
'banana': 'apple'
})
batch.add_insert({
'batch_insert_3': 29,
'bar': 2
})
batch.execute()
return batch
def test_batch_find_with_no_results(self):
self._perform_batch_insertion()
batch = Batch(self.collection, self.connection)
batch.add_find({
'bar': 23434
})
batch.add_find({
'batch_insert_2': 'abananasdf'
})
result = batch.find()
self.assertFalse(result[0])
self.assertFalse(result[1])
def test_batch_find(self):
self._perform_batch_insertion()
batch = Batch(self.collection, self.connection)
batch.add_find({
'bar': 2
})
batch.add_find({
'batch_insert_2': 'banana'
})
result = batch.find()
self.assertEqual(2, len(result))
self.assertEqual(2, len(result[0]))
self.assertEqual('apple', result[1][0]['banana'])
def test_batch_find_with_no_jobs(self):
self._perform_batch_insertion()
batch = Batch(self.collection, self.connection)
result = batch.find()
self.assertFalse(result)
