Ejemplo n.º 1
0
def test_resource_set_should_ignore_duplicate_executions_on_rollback():
    mock1 = ResourceSet('boo')
    mock1.rollback = Mock()

    res = ResourceSet('foo')
    res.add_item(mock1)

    res.rollback()
    res.rollback()
    res.rollback()
    mock1.rollback.assert_called_once_with()
Ejemplo n.º 2
0
def test_rollback_exec_sub_stets():
    mock1 = Mock(spec=ResourceSet)
    mock1.is_executed.return_value = False
    mock2 = Mock(spec=ResourceSet, return_value=False)
    mock2.is_executed.return_value = False
    mock3 = Mock(spec=ResourceSet, return_value=False)
    mock3.is_executed.return_value = False

    res = ResourceSet('foo')
    res.add_item(mock1)
    res.add_item(mock2)
    res.add_item(mock3)

    res.rollback()
    mock1.rollback.assert_called_once_with()
    mock2.rollback.assert_called_once_with()
    mock3.rollback.assert_called_once_with()