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

#include "stdafx.h"
#include "file_utility.h"

TEST(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
        &retVal, // arg count
        &bstrArray); // args array.

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

TEST(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
        &retVal, // arg count
        &bstrArray); // args array.

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

TEST(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
        &retVal, // arg count
        &bstrArray); // args array.

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

TEST(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
        &retVal, // arg count
        &bstrArray); // args array.

    EXPECT_EQ(E_INVALIDARG, hr);
}

TEST(GetAbsolutePathToDotnetFromProgramFiles, BackupWorks)
{
    STRU struAbsolutePathToDotnet;
    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 = FILE_UTILITY::CheckIfFileExists(L"C:/Program Files/dotnet/dotnet.exe");
    }
    else
    {
        fDotnetInProgramFiles = FILE_UTILITY::CheckIfFileExists(L"C:/Program Files (x86)/dotnet/dotnet.exe");
    }

    auto dotnetPath = HOSTFXR_UTILITY::GetAbsolutePathToDotnetFromProgramFiles();
    if (fDotnetInProgramFiles)
    {
        EXPECT_TRUE(dotnetPath.has_value());
    }
    else
    {
        EXPECT_FALSE(dotnetPath.has_value());
    }
}

TEST(GetHostFxrArguments, InvalidParams)
{
    DWORD retVal = 0;
    BSTR* bstrArray;
    STRU  struHostFxrDllLocation;
    STRU  struExeLocation;

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

    EXPECT_EQ(HRESULT_FROM_WIN32(ERROR_FILE_NOT_FOUND), hr);
}