def test_rollback_last(self):
"""Test transactional_update.rollback with last snapshot"""
salt_mock = {
"cmd.run_all": MagicMock(return_value={"stdout": "output", "retcode": 0})
}
with patch.dict(tu.__salt__, salt_mock):
assert tu.rollback("last") == "output"
salt_mock["cmd.run_all"].assert_called_with(
["transactional-update", "rollback", "last"]
)
def test_rollback_snapshot_number():
"""Test transactional_update.rollback with numeric snapshot"""
salt_mock = {
"cmd.run_all": MagicMock(return_value={
"stdout": "output",
"retcode": 0
})
}
with patch.dict(tu.__salt__, salt_mock):
assert tu.rollback(10) == "output"
salt_mock["cmd.run_all"].assert_called_with(
["transactional-update", "rollback", 10])
def test_rollback_error():
"""Test transactional_update.rollback with wrong snapshot"""
with pytest.raises(CommandExecutionError):
tu.rollback("error")
