Update KeyValueGraph exporter.

loophp · Jun 13, 2019 · 0b1717d · 0b1717d
1 parent eb4efa8
commit 0b1717d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/tests/src/Exporter/KeyValueGraph.php b/tests/src/Exporter/KeyValueGraph.php
@@ -4,14 +4,15 @@
 
 namespace drupol\phptree\tests\Exporter;
 
+use drupol\phptree\Exporter\Graph;
 use drupol\phptree\Node\KeyValueNodeInterface;
 use drupol\phptree\Node\NodeInterface;
 use Fhaculty\Graph\Vertex;
 
 /**
  * Class KeyValueGraph.
  */
-class KeyValueGraph extends ValueGraph
+class KeyValueGraph extends Graph
 {
     /**
      * {@inheritdoc}
@@ -21,7 +22,7 @@ protected function createVertex(NodeInterface $node): Vertex
         $vertex = parent::createVertex($node);
 
         if ($node instanceof KeyValueNodeInterface) {
-            $vertex->setAttribute('graphviz.label', $node->getValue());
+            $vertex->setAttribute('graphviz.label', $node->getKey() . $vertex->getAttribute('graphviz.label'));
         }
 
         return $vertex;