// Copyright (c) .NET Foundation. All rights reserved.
// Licensed under the MIT License. See License.txt in the project root for license information.

#include "stdafx.h"
#include "CppUnitTest.h"

using namespace Microsoft::VisualStudio::CppUnitTestFramework;

namespace AspNetCoreModuleTests
{
    TEST_CLASS(HOSTFXR_UTILITY_TESTS)
    {
    public:

        TEST_METHOD(ParseHostfxrArguments_BasicHostFxrArguments)
        {
            DWORD retVal = 0;
            BSTR* bstrArray;
            PCWSTR exeStr = L"C:/Program Files/dotnet.exe";

            HRESULT hr = HOSTFXR_UTILITY::ParseHostfxrArguments(
                L"exec \"test.dll\"", // args
                exeStr,  // exe path
                L"invalid",  // physical path to application
                NULL, // event log
                &retVal, // arg count
                &bstrArray); // args array.

            Assert::AreEqual(hr, S_OK);
            Assert::AreEqual(DWORD(3), retVal);
            Assert::AreEqual(exeStr, bstrArray[0]);
            Assert::AreEqual(L"exec", bstrArray[1]);
            Assert::AreEqual(L"test.dll", bstrArray[2]);
        }

        TEST_METHOD(ParseHostfxrArguments_NoExecProvided)
        {
            DWORD retVal = 0;
            BSTR* bstrArray;
            PCWSTR exeStr = L"C:/Program Files/dotnet.exe";

            HRESULT hr = HOSTFXR_UTILITY::ParseHostfxrArguments(
                L"test.dll", // args
                exeStr,  // exe path
                L"ignored",  // physical path to application
                NULL, // event log
                &retVal, // arg count
                &bstrArray); // args array.

            Assert::AreEqual(hr, S_OK);
            Assert::AreEqual(DWORD(2), retVal);
            Assert::AreEqual(exeStr, bstrArray[0]);
            Assert::AreEqual(L"test.dll", bstrArray[1]);
        }

        TEST_METHOD(ParseHostfxrArguments_ConvertDllToAbsolutePath)
        {
            DWORD retVal = 0;
            BSTR* bstrArray;
            PCWSTR exeStr = L"C:/Program Files/dotnet.exe";

            HRESULT hr = HOSTFXR_UTILITY::ParseHostfxrArguments(
                L"exec \"test.dll\"", // args
                exeStr,  // exe path
                L"C:/test",  // physical path to application
                NULL, // event log
                &retVal, // arg count
                &bstrArray); // args array.

            Assert::AreEqual(hr, S_OK);
            Assert::AreEqual(DWORD(3), retVal);
            Assert::AreEqual(exeStr, bstrArray[0]);
            Assert::AreEqual(L"exec", bstrArray[1]);
            Assert::AreEqual(L"C:\\test\\test.dll", bstrArray[2]);
        }

        TEST_METHOD(ParseHostfxrArguments_ProvideNoArgs_InvalidArgs)
        {
            DWORD retVal = 0;
            BSTR* bstrArray;
            PCWSTR exeStr = L"C:/Program Files/dotnet.exe";

            HRESULT hr = HOSTFXR_UTILITY::ParseHostfxrArguments(
                L"", // args
                exeStr,  // exe path
                L"ignored",  // physical path to application
                NULL, // event log
                &retVal, // arg count
                &bstrArray); // args array.

            Assert::AreEqual(E_INVALIDARG, hr);
        }

        TEST_METHOD(GetAbsolutePathToDotnetFromProgramFiles_BackupWorks)
        {
            STRU struAbsolutePathToDotnet;
            HRESULT hr = S_OK;
            BOOL fDotnetInProgramFiles;
            BOOL is64Bit;
            BOOL fIsWow64 = FALSE;
            SYSTEM_INFO systemInfo;
            IsWow64Process(GetCurrentProcess(), &fIsWow64);
            if (fIsWow64)
            {
                is64Bit = FALSE;
            }
            else
            {
                GetNativeSystemInfo(&systemInfo);
                is64Bit = systemInfo.wProcessorArchitecture == PROCESSOR_ARCHITECTURE_AMD64;
            }

            if (is64Bit)
            {
                fDotnetInProgramFiles = UTILITY::CheckIfFileExists(L"C:/Program Files/dotnet/dotnet.exe");
            }
            else
            {
                fDotnetInProgramFiles = UTILITY::CheckIfFileExists(L"C:/Program Files (x86)/dotnet/dotnet.exe");
            }
            
            hr = HOSTFXR_UTILITY::GetAbsolutePathToDotnetFromProgramFiles(&struAbsolutePathToDotnet);
            if (fDotnetInProgramFiles)
            {
                Assert::AreEqual(hr, S_OK);
            }
            else
            {
                Assert::AreNotEqual(hr, S_OK);
                Assert::IsTrue(struAbsolutePathToDotnet.IsEmpty());
            }
        }

        TEST_METHOD(GetHostFxrArguments_InvalidParams)
        {
            DWORD retVal = 0;
            BSTR* bstrArray;
            STRU  struHostFxrDllLocation;

            HRESULT hr = HOSTFXR_UTILITY::GetHostFxrParameters(
                INVALID_HANDLE_VALUE,
                L"bogus", // processPath
                L"",  // application physical path, ignored.
                L"ignored",  //arguments
                NULL, // event log
                &retVal, // arg count
                &bstrArray); // args array.

            Assert::AreEqual(E_INVALIDARG, hr);
        }
    };
}