#include <stdio.h>
#include <winpr/crt.h>
#include <winpr/path.h>
#include <winpr/tchar.h>
#include <winpr/winpr.h>

static const TCHAR testBasePathBackslash[] = _T("C:\\Program Files\\");
static const TCHAR testBasePathNoBackslash[] = _T("C:\\Program Files");
static const TCHAR testMorePathBackslash[] = _T("\\Microsoft Visual Studio 11.0");
static const TCHAR testMorePathNoBackslash[] = _T("Microsoft Visual Studio 11.0");
static const TCHAR testPathOut[] = _T("C:\\Program Files\\Microsoft Visual Studio 11.0");
static const TCHAR testPathOutMorePathBackslash[] = _T("C:\\Microsoft Visual Studio 11.0");

int TestPathAllocCombine(int argc, char* argv[])
{
	HRESULT status;
	LPTSTR PathOut;

	/* Base Path: Backslash, More Path: No Backslash */

	status = PathAllocCombine(testBasePathBackslash, testMorePathNoBackslash, 0, &PathOut);

	if (status != S_OK)
	{
		_tprintf(_T("PathAllocCombine status: 0x%08")_T(PRIX32)_T("\n"), status);
		return -1;
	}

	if (_tcscmp(PathOut, testPathOut) != 0)
	{
		_tprintf(_T("Path Mismatch 1: Actual: %s, Expected: %s\n"), PathOut, testPathOut);
		return -1;
	}

	HeapFree(GetProcessHeap(), 0, PathOut);

	/* Base Path: Backslash, More Path: Backslash */

	status = PathAllocCombine(testBasePathBackslash, testMorePathBackslash, 0, &PathOut);

	if (status != S_OK)
	{
		_tprintf(_T("PathAllocCombine status: 0x%08")_T(PRIX32)_T("\n"), status);
		return -1;
	}

	if (_tcscmp(PathOut, testPathOutMorePathBackslash) != 0)
	{
		_tprintf(_T("Path Mismatch 2: Actual: %s, Expected: %s\n"), PathOut, testPathOutMorePathBackslash);
		return -1;
	}

	HeapFree(GetProcessHeap(), 0, PathOut);

	/* Base Path: No Backslash, More Path: Backslash */

	status = PathAllocCombine(testBasePathNoBackslash, testMorePathBackslash, 0, &PathOut);

	if (status != S_OK)
	{
		_tprintf(_T("PathAllocCombine status: 0x%08")_T(PRIX32)_T("\n"), status);
		return -1;
	}

	if (_tcscmp(PathOut, testPathOutMorePathBackslash) != 0)
	{
		_tprintf(_T("Path Mismatch 3: Actual: %s, Expected: %s\n"), PathOut, testPathOutMorePathBackslash);
		return -1;
	}

	HeapFree(GetProcessHeap(), 0, PathOut);

	/* Base Path: No Backslash, More Path: No Backslash */

	status = PathAllocCombine(testBasePathNoBackslash, testMorePathNoBackslash, 0, &PathOut);

	if (status != S_OK)
	{
		_tprintf(_T("PathAllocCombine status: 0x%08")_T(PRIX32)_T("\n"), status);
		return -1;
	}

	if (_tcscmp(PathOut, testPathOut) != 0)
	{
		_tprintf(_T("Path Mismatch 4: Actual: %s, Expected: %s\n"), PathOut, testPathOut);
		return -1;
	}

	HeapFree(GetProcessHeap(), 0, PathOut);

	return 0;
}