1 files changed, 98 insertions, 0 deletions

diff --git a/src/libs/dutil/test/DUtilUnitTest/UriUtilTest.cpp b/src/libs/dutil/test/DUtilUnitTest/UriUtilTest.cpp
new file mode 100644
index 00000000..b3bf87a2
--- /dev/null
+++ b/src/libs/dutil/test/DUtilUnitTest/UriUtilTest.cpp
@@ -0,0 +1,98 @@
+// 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();
+                }
+        };
+}