Merge pull request #125 from Flashky/feature/56-add-labelededge-class-for-jgrapht-graphs

feat: add LabeledEdge class #56

Author: Flashky

src/main/java/com/adventofcode/flashk/common/jgrapht/LabeledEdge.java

@@ -0,0 +1,34 @@
+package com.adventofcode.flashk.common.jgrapht;
+
+import org.jgrapht.graph.DefaultEdge;
+import org.jgrapht.nio.Attribute;
+import org.jgrapht.nio.AttributeType;
+
+/// Labeled edge class based on [JGraphT LabeledEdges](https://jgrapht.org/guide/LabeledEdges)
+///
+/// Allows creating an edge with label.
+///
+/// It also implements the JGraphT Attribute interface to allow exporting the labels in DOT representation
+public class LabeledEdge extends DefaultEdge implements Attribute {
+
+    private final String label;
+
+    public LabeledEdge(String label) {
+        this.label = label;
+    }
+
+    @Override
+    public String getValue() {
+        return label;
+    }
+
+    @Override
+    public AttributeType getType() {
+        return AttributeType.STRING;
+    }
+
+    @Override
+    public String toString() {
+        return "(" + getSource() + " : " + getTarget() + " : " + label + ")";
+    }
+}

src/test/java/com/adventofcode/flashk/common/jgrapht/LabeledEdgeTest.java

@@ -0,0 +1,21 @@
+package com.adventofcode.flashk.common.jgrapht;
+
+import org.jgrapht.nio.AttributeType;
+import org.junit.jupiter.api.Test;
+
+import static org.junit.jupiter.api.Assertions.*;
+
+class LabeledEdgeTest {
+
+    @Test
+    void getValue() {
+        LabeledEdge labeledEdge = new LabeledEdge("expected");
+        assertEquals("expected", labeledEdge.getValue());
+    }
+
+    @Test
+    void getType() {
+        LabeledEdge labeledEdge = new LabeledEdge("expected");
+        assertEquals(AttributeType.STRING, labeledEdge.getType());
+    }
+}