Add 'labeled' method to 'Node' class

neo4j-php · Dec 13, 2022 · 27633fa · 27633fa
1 parent 8808a32
commit 27633fa
Show file tree

Hide file tree

Showing 2 changed files with 28 additions and 0 deletions.
diff --git a/src/Patterns/Node.php b/src/Patterns/Node.php
@@ -9,6 +9,7 @@
  */
 namespace WikibaseSolutions\CypherDSL\Patterns;
 
+use WikibaseSolutions\CypherDSL\Expressions\Label;
 use WikibaseSolutions\CypherDSL\Expressions\Literals\Map;
 use WikibaseSolutions\CypherDSL\Traits\ErrorTrait;
 use WikibaseSolutions\CypherDSL\Traits\EscapeTrait;
@@ -81,6 +82,16 @@ public function getLabels(): array
         return $this->labels;
     }
 
+    /**
+     * Returns a label with the variable in this node.
+     *
+     * @param string ...$labels The labels to attach to the variable in this node
+     */
+    public function labeled(string ...$labels): Label
+    {
+        return new Label($this->getVariable(), ...$labels);
+    }
+
     /**
      * @inheritDoc
      */

diff --git a/tests/unit/Patterns/NodeTest.php b/tests/unit/Patterns/NodeTest.php
@@ -11,6 +11,7 @@
 
 use PHPUnit\Framework\TestCase;
 use TypeError;
+use WikibaseSolutions\CypherDSL\Expressions\Label;
 use WikibaseSolutions\CypherDSL\Expressions\Literals\Float_;
 use WikibaseSolutions\CypherDSL\Expressions\Literals\Integer;
 use WikibaseSolutions\CypherDSL\Expressions\Literals\List_;
@@ -234,6 +235,22 @@ public function testRelationshipUni(): void
         $this->assertSame("(:City)-[:LIVES_IN]-(:City {city: 'Amsterdam'})", $relationship->toQuery());
     }
 
+    public function testLabeledSingleLabel(): void
+    {
+        $node = new Node();
+        $labeled = $node->labeled('German');
+
+        $this->assertEquals(new Label($node->getVariable(), 'German'), $labeled);
+    }
+
+    public function testLabeledMultipleLabels(): void
+    {
+        $node = new Node();
+        $labeled = $node->labeled('German', 'Swedish');
+
+        $this->assertEquals(new Label($node->getVariable(), 'German', 'Swedish'), $labeled);
+    }
+
     public function provideOnlyLabelData(): array
     {
         return [