some fixes for tests; failed because of wrong/missing metadata

Author: eosfor

PSGraph.Tests/ExportGraphViewCmdletTests.cs

@@ -153,7 +153,7 @@ public void ExportGraph_InvalidFormat_ThrowsException()
         public void ExportGraph_NoPathProvided_ReturnsOutput()
         {
             // Arrange
-            var graph = CreateSampleGraph();
+            var graph = CreateSampleGraphWithMetadata();
 
             _powershell.AddCommand("Export-Graph")
                 .AddParameter("Graph", graph)
@@ -173,7 +173,7 @@ public void ExportGraph_NoPathProvided_ReturnsOutput()
         public void ExportGraph_NoPathProvided_ReturnsVegaString()
         {
             // Arrange
-            var graph = CreateSampleGraph();
+            var graph = CreateSampleGraphWithMetadata();
 
             var records = graph.ConvertToVegaNodeLink();
 
@@ -326,6 +326,28 @@ private PsBidirectionalGraph CreateSampleGraph()
             var vertexB = new PSVertex("B");
             var vertexC = new PSVertex("C");
 
+            graph.AddVertexRange(new[] { vertexA, vertexB, vertexC });
+
+            var edgeAB = new PSEdge(vertexA, vertexB, new PSEdgeTag());
+            var edgeBC = new PSEdge(vertexB, vertexC, new PSEdgeTag());
+
+            graph.AddEdgeRange(new[] { edgeAB, edgeBC });
+
+            return graph;
+        }
+
+        private PsBidirectionalGraph CreateSampleGraphWithMetadata()
+        {
+            var graph = new PsBidirectionalGraph();
+
+            var vertexA = new PSVertex("A");
+            vertexA.Metadata.TryAdd("group", 1);
+            var vertexB = new PSVertex("B");
+            vertexB.Metadata.TryAdd("group", 1);
+            var vertexC = new PSVertex("C");
+            vertexB.Metadata.TryAdd("group", 2);
+
+
             graph.AddVertexRange(new[] { vertexA, vertexB, vertexC });
 
             var edgeAB = new PSEdge(vertexA, vertexB, new PSEdgeTag());

PSGraph.Vega.Extensions/VegaDataConverter.cs

@@ -38,11 +38,15 @@ public static (List<NodeRecord> nodes, List<LinkRecord> links) ConvertToVegaNode
                 if (vertexLookup.TryGetValue(edge.Source, out int sourceId) &&
                     vertexLookup.TryGetValue(edge.Target, out int targetId))
                 {
-                {
-                    var groupObj = ((IDictionary<string, object?>)edge.Target.Metadata)["group"];
-                    if (groupObj != null && int.TryParse(groupObj.ToString(), out int parsedGroup))
-                        value = parsedGroup;
-                }                    
+                    {
+                        if (((IDictionary<string, object?>)edge.Target.Metadata).ContainsKey("group"))
+                        {
+                            var groupObj = ((IDictionary<string, object?>)edge.Target.Metadata)["group"];
+                            if (groupObj != null && int.TryParse(groupObj.ToString(), out int parsedGroup))
+                                value = parsedGroup;
+                        }
+
+                    }
                     links.Add(new LinkRecord(sourceId, targetId, value, sourceId, targetId));
                 }
             }