cleaning up

Author: 123kiran17

src/Neo4jRequestFactory.php

@@ -8,9 +8,6 @@
 use Psr\Http\Message\RequestInterface;
 use Psr\Http\Message\StreamFactoryInterface;
 
-/**
- * @api
- */
 class Neo4jRequestFactory
 {
     public function __construct(

src/OGM.php

@@ -8,9 +8,6 @@
 use Neo4j\QueryAPI\Objects\Path;
 use InvalidArgumentException;
 
-/**
- *  @api
- */
 class OGM
 {
     /**
@@ -57,14 +54,14 @@ private function mapNode(array $nodeData): Node
     {
         return new Node(
             labels: $nodeData['_labels'] ?? [],
-            properties: $this->mapProperties($nodeData['_properties'] ?? []) // ✅ Fix: Ensure properties exist
+            properties: $this->mapProperties($nodeData['_properties'] ?? [])
         );
     }
 
     private function mapRelationship(array $relationshipData): Relationship
     {
         return new Relationship(
-            type: $relationshipData['_type'] ?? 'UNKNOWN',  // ✅ Fix: Default to 'UNKNOWN'
+            type: $relationshipData['_type'] ?? 'UNKNOWN',
             properties: $this->mapProperties($relationshipData['_properties'] ?? [])
         );
     }

src/Objects/Node.php

@@ -30,15 +30,6 @@ public function __construct(array $labels, array $properties)
         $this->properties = $properties;
     }
 
-    /**
-     * Get the labels of the node.
-     * @return string[] Array of labels.
-     */
-    public function getLabels(): array
-    {
-        return $this->labels;
-    }
-
     /**
      * Get the properties of the node.
      * @return array<string, mixed> Associative array of properties.

src/Results/ResultSet.php

@@ -12,7 +12,6 @@
 use Traversable;
 
 /**
- * @api
  * @template TValue
  * @implements IteratorAggregate<int, ResultRow>
  */

tests/Integration/Neo4jOGMTest.php

@@ -5,8 +5,12 @@
 use Neo4j\QueryAPI\OGM;
 use PHPUnit\Framework\TestCase;
 
-final class Neo4jOGMTest extends TestCase
+/**
+ *  @api
+ */
+class Neo4jOGMTest extends TestCase
 {
+    /** @psalm-suppress PropertyNotSetInConstructor */
     private OGM $ogm;
 
     #[\Override]
@@ -31,9 +35,9 @@ public function testWithNode(): void
         $this->assertEquals('Ayush', $node->getProperties()['name']['_value']);
     }
 
-
     public function testWithSimpleRelationship(): void
     {
+
         $relationshipData = [
             '$type' => 'Relationship',
             '_value' => [
@@ -43,10 +47,9 @@ public function testWithSimpleRelationship(): void
         ];
 
         $relationship = $this->ogm->map($relationshipData);
-        $this->assertEquals('FRIENDS', $relationship->getType());
+        $this->assertEquals('FRIENDS', $relationship->type);
     }
 
-    // More tests...
     public function testWithPath(): void
     {
         $pathData = [
@@ -82,10 +85,10 @@ public function testWithPath(): void
 
         $path = $this->ogm->map($pathData);
 
-        $this->assertCount(2, $path->getNodes());
-        $this->assertCount(1, $path->getRelationships());
-        $this->assertEquals('A', $path->getNodes()[0]->getProperties()['name']['_value']);
-        $this->assertEquals('B', $path->getNodes()[1]->getProperties()['name']['_value']);
+        $this->assertCount(2, $path->nodes);
+        $this->assertCount(1, $path->relationships);
+        $this->assertEquals('A', $path->nodes[0]->getProperties()['name']['_value']);
+        $this->assertEquals('B', $path->nodes[1]->getProperties()['name']['_value']);
     }
 
 }