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