projectmodelserver

test/dotnet-projectmodel-server.Tests/Helpers/JArrayExtensions.cs

@@ -0,0 +1,85 @@
+// 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.
+
+using System;
+using Newtonsoft.Json.Linq;
+using Xunit;
+
+namespace Microsoft.DotNet.ProjectModel.Server.Tests
+{
+    public static class JArrayExtensions
+    {
+        public static JArray AssertJArrayEmpty(this JArray array)
+        {
+            Assert.NotNull(array);
+            Assert.Empty(array);
+
+            return array;
+        }
+
+        public static JArray AssertJArrayNotEmpty(this JArray array)
+        {
+            Assert.NotNull(array);
+            Assert.NotEmpty(array);
+
+            return array;
+        }
+
+        public static JArray AssertJArrayCount(this JArray array, int expectedCount)
+        {
+            Assert.NotNull(array);
+            Assert.Equal(expectedCount, array.Count);
+
+            return array;
+        }
+
+        public static JArray AssertJArrayElement<T>(this JArray array, int index, T expectedElementValue)
+        {
+            Assert.NotNull(array);
+
+            var element = array[index];
+            Assert.NotNull(element);
+            Assert.Equal(expectedElementValue, element.Value<T>());
+
+            return array;
+        }
+
+        public static JArray AssertJArrayContains<T>(this JArray array, T value)
+        {
+            AssertJArrayContains<T>(array, element => object.Equals(element, value));
+
+            return array;
+        }
+
+        public static JArray AssertJArrayContains<T>(this JArray array, Func<T, bool> critiera)
+        {
+            bool contains = false;
+            foreach (var element in array)
+            {
+                var value = element.Value<T>();
+
+                contains = critiera(value);
+                if (contains)
+                {
+                    break;
+                }
+            }
+
+            Assert.True(contains, "JArray doesn't contains the specified element.");
+
+            return array;
+        }
+
+        public static T RetrieveArraryElementAs<T>(this JArray json, int index)
+            where T : JToken
+        {
+            Assert.NotNull(json);
+            Assert.True(index >= 0 && index < json.Count, "Index out of range");
+
+            var element = json[index];
+            DthMessageExtension.AssertType<T>(element, $"Element at {index}");
+
+            return (T)element;
+        }
+    }
+}