Add first() and last() functions to Collection

stripe · Jun 11, 2020 · e862605 · e862605
1 parent 2468717
commit e862605
Show file tree

Hide file tree

Showing 2 changed files with 44 additions and 0 deletions.
diff --git a/lib/Collection.php b/lib/Collection.php
@@ -241,6 +241,26 @@ public function previousPage($params = null, $opts = null)
         return $this->all($params, $opts);
     }
 
+    /**
+     * Get the first item from the collection.
+     *
+     * @return null|\Stripe\StripeObject
+     */
+    public function first()
+    {
+        return \reset($this->data);
+    }
+
+    /**
+     * Get the last item from the collection.
+     *
+     * @return null|\Stripe\StripeObject
+     */
+    public function last()
+    {
+        return \end($this->data);
+    }
+
     private function extractPathAndUpdateParams($params)
     {
         $url = \parse_url($this->url);

diff --git a/tests/Stripe/CollectionTest.php b/tests/Stripe/CollectionTest.php
@@ -312,4 +312,28 @@ public function testPreviousPage()
         $previousPage = $this->fixture->previousPage();
         static::assertSame([], $previousPage->data);
     }
+
+    public function testFirst()
+    {
+        $collection = Collection::constructFrom([
+            'data' => [
+                ['content' => 'first'],
+                ['content' => 'middle'],
+                ['content' => 'last'],
+            ],
+        ]);
+        static::assertSame('first', $collection->first()['content']);
+    }
+
+    public function testLast()
+    {
+        $collection = Collection::constructFrom([
+            'data' => [
+                ['content' => 'first'],
+                ['content' => 'middle'],
+                ['content' => 'last'],
+            ],
+        ]);
+        static::assertSame('last', $collection->last()['content']);
+    }
 }