Added method and tests for GetQueryKeys

Author: Jimmy Merrild Krag

UriBuilderExtended.Test/UriBuilderExtensionQueryTests.cs

@@ -137,6 +137,34 @@ public void SetQuery()
             Assert.IsFalse(uriBuilder.HasQuery("key", "value3"), "Unexpected value found");
         }
 
+        [TestMethod]
+        public void GetQueryKeys()
+        {
+            string urlString = "http://www.test.com/";
+
+            UriBuilder uriBuilder = new UriBuilder(urlString);
+
+            HashSet<string> inputKeys = new HashSet<string>
+            {
+                "key1",
+                "key2",
+                "key3",
+                "key4",
+                "key5",
+            };
+
+            foreach (string key in inputKeys)
+            {
+                uriBuilder.AddQuery(key, "value");
+            }
+
+            HashSet<string> retrievedKeys = new HashSet<string>(uriBuilder.GetQueryKeys());
+
+            Assert.AreEqual<int>(uriBuilder.GetQueryKeys().Count, inputKeys.Count, "Count does not match input count");
+
+            Assert.IsTrue(inputKeys.SetEquals(retrievedKeys), "Retrieved result does not match set keys");
+        }
+
         [TestMethod]
         public void GetQueryValues()
         {
@@ -160,7 +188,7 @@ public void GetQueryValues()
 
             HashSet<string> retrievedValues = new HashSet<string>(uriBuilder.GetQueryValues("key"));
 
-            Assert.AreEqual<int>(uriBuilder.GetQueryValues("key").Count, inputValues.Count, "Count for key does not match input count");
+            Assert.AreEqual<int>(uriBuilder.GetQueryValues("key").Count, inputValues.Count, "Count does not match input count");
 
             Assert.IsTrue(inputValues.SetEquals(retrievedValues), "Retrieved result does not match set values");
 

UriBuilderExtended/UriBuilderExtensions.cs

@@ -120,11 +120,29 @@ public static UriBuilder AddQuery(this UriBuilder uri, string key, params string
         }
 
         /// <summary>
-        /// Get all values for the given key.
+        /// Get all keys in the query
+        /// </summary>
+        /// <param name="uri">The <see cref="Uri"/></param>
+        /// <returns>A collection with all keys in the current query</returns>
+        public static ICollection<string> GetQueryKeys(this UriBuilder uri)
+        {
+            HashSet<string> result = new HashSet<string>();
+
+            NameValueCollection queryValues = uri.ParseQuery();
+            foreach (string key in queryValues.AllKeys)
+            {
+                result.Add(key);
+            }
+
+            return result;
+        }
+
+        /// <summary>
+        /// Get all values for the given key
         /// </summary>
         /// <param name="uri">The <see cref="Uri"/></param>
         /// <param name="key">The key to look up on</param>
-        /// <returns>A collection of values or an empty list for non-existing keys.</returns>
+        /// <returns>A collection of values or an empty list if the key does not exist</returns>
         public static ICollection<string> GetQueryValues(this UriBuilder uri, string key)
         {
             HashSet<string> result = new HashSet<string>();
@@ -135,7 +153,6 @@ public static ICollection<string> GetQueryValues(this UriBuilder uri, string key
             }
 
             NameValueCollection queryValues = uri.ParseQuery();
-
             foreach (string value in queryValues.GetValues(key))
             {
                 result.Add(value);