def test_exceptions1(self): mongo_connection = MongoProxy( MongoConnection(exceptions_to_raise=[ValueError]), methods_needing_retry={MongoConnection: ['find', ]} ) with self.assertRaises(ValueError): mongo_connection.find() self.assertEqual(mongo_connection.call_count, 1)
def test_exceptions1(self): mongo_connection = MongoProxy( MongoConnection(exceptions_to_raise=[ValueError]), methods_needing_retry={MongoConnection: [ 'find', ]}) with self.assertRaises(ValueError): mongo_connection.find() self.assertEqual(mongo_connection.call_count, 1)
def test_exceptions2(self): mongo_connection = MongoProxy( MongoConnection(exceptions_to_raise=[AutoReconnect, AutoReconnect]), methods_needing_retry={MongoConnection: ['find', ]} ) self.assertEqual(mongo_connection.find(), "Finished.") self.assertEqual(mongo_connection.call_count, 3)
def test_no_exceptions1(self): mongo_connection = MongoProxy( MongoConnection(), methods_needing_retry={MongoConnection: [ 'find', ]}) self.assertEqual(mongo_connection.find(), "Finished.") self.assertEqual(mongo_connection.call_count, 1)
def test_exceptions2(self): mongo_connection = MongoProxy( MongoConnection( exceptions_to_raise=[AutoReconnect, AutoReconnect]), methods_needing_retry={MongoConnection: [ 'find', ]}) self.assertEqual(mongo_connection.find(), "Finished.") self.assertEqual(mongo_connection.call_count, 3)
def test_no_exceptions1(self): mongo_connection = MongoProxy(MongoConnection(), methods_needing_retry={MongoConnection: ['find', ]}) self.assertEqual(mongo_connection.find(), "Finished.") self.assertEqual(mongo_connection.call_count, 1)