def test_adding_with_negative_max_size_throws_exception(self):
buff = length_buffer.LengthBuffer(5)
buff.max_size = -1
with self.assertRaises(AttributeError) as context:
buff.add({'id': 'test'})
self.assertTrue(context)
def test_adding_item_forces_flush(self, mock_reducer):
obj = {'id': 'test'}
buff = length_buffer.LengthBuffer(0)
buff.register_reducer(mock_reducer)
buff.add(obj)
self.assertEqual(len(buff), 0)
mock_reducer.process.assert_called_once_with([obj])
def test_adding_multiple_reducers_forces_flush(self, mock_reducer1,
mock_reducer2):
obj = {'id': 'test'}
buff = length_buffer.LengthBuffer(0)
buff.register_reducer(mock_reducer1)
buff.register_reducer(mock_reducer2)
buff.add(obj)
mock_reducer1.process.assert_called_once_with([obj])
mock_reducer2.process.assert_called_once_with([obj])
def test_adding_multiple_objects_forces_flush(self, mock_reducer):
obj1 = {'id': 'test'}
obj2 = {'id': 'test'}
buff = length_buffer.LengthBuffer(1)
buff.register_reducer(mock_reducer)
buff.add(obj1)
buff.add(obj2)
self.assertEqual(len(buff), 0)
mock_reducer.process.assert_called_once_with([obj1, obj2])
def test_add(self, mock_reducer):
buff = length_buffer.LengthBuffer(5)
self.assertEqual(len(buff), 0)
buff.add({'id': 'test1'})
self.assertEqual(len(buff), 1)
mock_reducer.process.assert_not_called()