def test_create_cell_recalculator_should(self, mock_recalculate): unrecalculated_queue = Queue() unrecalculated_queue.put(1) unrecalculated_queue.put(1) unrecalculated_queue.task_done = Mock() leaf_queue = Queue() leaf_queue.put(sentinel.one) leaf_queue.put(sentinel.two) leaf_queue.task_done = Mock() target = create_cell_recalculator(leaf_queue, unrecalculated_queue, sentinel.graph, sentinel.context) target() self.assertTrue(unrecalculated_queue.empty()) self.assertEquals( unrecalculated_queue.task_done.call_args_list, [ ((), {}), ((), {}), ] ) self.assertTrue(leaf_queue.empty()) self.assertEquals( mock_recalculate.call_args_list, [ ((sentinel.one, leaf_queue, sentinel.graph, sentinel.context), {}), ((sentinel.two, leaf_queue, sentinel.graph, sentinel.context), {}) ] ) self.assertEquals( leaf_queue.task_done.call_args_list, [ ((), {}), ((), {}), ] )
def test_create_cell_recalculator_should(self, mock_recalculate): unrecalculated_queue = Queue() unrecalculated_queue.put(1) unrecalculated_queue.put(1) unrecalculated_queue.task_done = Mock() leaf_queue = Queue() leaf_queue.put(sentinel.one) leaf_queue.put(sentinel.two) leaf_queue.task_done = Mock() target = create_cell_recalculator(leaf_queue, unrecalculated_queue, sentinel.graph, sentinel.context) target() self.assertTrue(unrecalculated_queue.empty()) self.assertEquals(unrecalculated_queue.task_done.call_args_list, [ ((), {}), ((), {}), ]) self.assertTrue(leaf_queue.empty()) self.assertEquals(mock_recalculate.call_args_list, [ ((sentinel.one, leaf_queue, sentinel.graph, sentinel.context), {}), ((sentinel.two, leaf_queue, sentinel.graph, sentinel.context), {}) ]) self.assertEquals(leaf_queue.task_done.call_args_list, [ ((), {}), ((), {}), ])
def test_create_cell_recalculator_should_handle_exceptions_from_recalc_cell(self, mock_recalculate): mock_recalculate.side_effect = die() unrecalculated_queue = Queue() unrecalculated_queue.put(1) unrecalculated_queue.task_done = Mock() leaf_queue = Queue() leaf_queue.put(sentinel.one) leaf_queue.task_done = Mock() target = create_cell_recalculator(leaf_queue, unrecalculated_queue, sentinel.graph, sentinel.context) self.assertRaises(AssertionError, target) self.assertTrue(unrecalculated_queue.empty()) self.assertCalledOnce(unrecalculated_queue.task_done) self.assertTrue(leaf_queue.empty()) self.assertCalledOnce(mock_recalculate, sentinel.one, leaf_queue, sentinel.graph, sentinel.context) self.assertCalledOnce(leaf_queue.task_done)
def test_create_cell_recalculator_should_handle_exceptions_from_recalc_cell( self, mock_recalculate): mock_recalculate.side_effect = die() unrecalculated_queue = Queue() unrecalculated_queue.put(1) unrecalculated_queue.task_done = Mock() leaf_queue = Queue() leaf_queue.put(sentinel.one) leaf_queue.task_done = Mock() target = create_cell_recalculator(leaf_queue, unrecalculated_queue, sentinel.graph, sentinel.context) self.assertRaises(AssertionError, target) self.assertTrue(unrecalculated_queue.empty()) self.assertCalledOnce(unrecalculated_queue.task_done) self.assertTrue(leaf_queue.empty()) self.assertCalledOnce(mock_recalculate, sentinel.one, leaf_queue, sentinel.graph, sentinel.context) self.assertCalledOnce(leaf_queue.task_done)