add chained test

src/Resources/LazyCollection.php

@@ -8,6 +8,8 @@
 /**
  * @template TKey of array-key
  * @template TValue
+ *
+ * @implements IteratorAggregate<TKey, TValue>
  */
 class LazyCollection implements IteratorAggregate
 {
@@ -37,8 +39,8 @@ public function all(): array
     /**
      * Get an item from the collection by key.
      *
-     * @param int|string  $key
-     * @return mixed|null
+     * @param TKey $key
+     * @return TValue|null
      */
     public function get($key)
     {
@@ -113,7 +115,7 @@ public function map(callable $callback): self
     }
 
     /**
-     * Take the first or last {$limit} items.
+     * Take the first {$limit} items.
      *
      * @param int $limit
      * @return static

tests/Mollie/API/Resources/LazyCollectionTest.php

@@ -84,4 +84,17 @@ public function testEvery()
             return $value > 1;
         }));
     }
+
+    public function testChainedUsage()
+    {
+        $result = $this->collection
+            ->filter(function ($value) {
+                return $value > 1;
+            })->map(function ($value) {
+                return $value * 2;
+            })->take(1);
+
+        $this->assertEquals(1, $result->count());
+        $this->assertEquals(4, $result->first());
+    }
 }