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)