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()