#pragma once
// Copyright (c) .NET Foundation and contributors. All rights reserved. Licensed under the Microsoft Reciprocal License. See LICENSE.TXT file in the project root for full license information.


namespace Microsoft
{
namespace Tools
{
namespace WindowsInstallerXml
{
namespace Test
{
namespace Bootstrapper
{
    using namespace System;
    using namespace WixInternal::TestSupport;

    public ref struct BurnTestException : public SucceededException
    {
    public:
        BurnTestException(HRESULT error, String^ message)
            : SucceededException(error, message)
        {
        }

        property Int32 ErrorCode
        {
            Int32 get()
            {
              return this->HResult;
            }
        }

    };
}
}
}
}
}

// this class is used by __TestThrowOnFailure_Format() below to deallocate
// the string created after the function call has returned
class __TestThrowOnFailure_StringFree
{
    LPWSTR m_scz;

public:
    __TestThrowOnFailure_StringFree(LPWSTR scz)
    {
        m_scz = scz;
    }

    ~__TestThrowOnFailure_StringFree()
    {
        ReleaseStr(m_scz);
    }

    operator LPCWSTR()
    {
        return m_scz;
    }
};

// used by the TestThrowOnFailure macros to format the error string and
// return an LPCWSTR that can be used to initialize a System::String
#pragma warning (push)
#pragma warning (disable : 4793)
inline __TestThrowOnFailure_StringFree __TestThrowOnFailure_Format(LPCWSTR wzFormat, ...)
{
    Assert(wzFormat && *wzFormat);

    HRESULT hr = S_OK;
    LPWSTR scz = NULL;
    va_list args;

    va_start(args, wzFormat);
    hr = StrAllocFormattedArgs(&scz, wzFormat, args);
    va_end(args);
    ExitOnFailure(hr, "Failed to format message string.");

LExit:
    return scz;
}
#pragma warning (pop)

#define TestThrowOnFailure(hr, s) if (FAILED(hr)) { throw gcnew Microsoft::Tools::WindowsInstallerXml::Test::Bootstrapper::BurnTestException(hr, gcnew System::String(s)); }
#define TestThrowOnFailure1(hr, s, p) if (FAILED(hr)) { throw gcnew Microsoft::Tools::WindowsInstallerXml::Test::Bootstrapper::BurnTestException(hr, gcnew System::String(__TestThrowOnFailure_Format(s, p))); }
#define TestThrowOnFailure2(hr, s, p1, p2) if (FAILED(hr)) { throw gcnew Microsoft::Tools::WindowsInstallerXml::Test::Bootstrapper::BurnTestException(hr, gcnew System::String(__TestThrowOnFailure_Format(s, p1, p2))); }