// 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.

#include "precomp.h"

using namespace System;
using namespace System::Text;
using namespace System::Collections::Generic;
using namespace Xunit;

namespace CfgTests
{
	public ref class UriUtil
	{
	public:
		[Fact]
		void UriProtocolTest()
		{
			HRESULT hr = S_OK;

			DutilInitialize(&DutilTestTraceError);

			LPCWSTR uri = L"https://localhost/";
			URI_PROTOCOL uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_HTTPS, (int)uriProtocol);

			uri = L"HTTPS://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_HTTPS, (int)uriProtocol);

			uri = L"HtTpS://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_HTTPS, (int)uriProtocol);

			uri = L"HTTP://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_HTTP, (int)uriProtocol);

			uri = L"http://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_HTTP, (int)uriProtocol);

			uri = L"HtTp://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_HTTP, (int)uriProtocol);

			uri = L"file://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_FILE, (int)uriProtocol);

			uri = L"FILE://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_FILE, (int)uriProtocol);

			uri = L"FiLe://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_FILE, (int)uriProtocol);

			uri = L"FTP://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_FTP, (int)uriProtocol);

			uri = L"ftp://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_FTP, (int)uriProtocol);

			uri = L"FtP://localhost/";
			uriProtocol = URI_PROTOCOL::URI_PROTOCOL_UNKNOWN;
			hr = UriProtocol(uri, &uriProtocol);
			ExitOnFailure(hr, "Failed to determine UriProtocol");
			Assert::Equal((int)URI_PROTOCOL::URI_PROTOCOL_FTP, (int)uriProtocol);

		LExit:
			DutilUninitialize();
		}
	};
}