Make countable

team-acaisia · Apr 23, 2023 · 08e878e · 08e878e
1 parent 497dc4f
commit 08e878e
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 1 deletion.
diff --git a/src/ByteArray.php b/src/ByteArray.php
@@ -9,7 +9,7 @@
 /**
  * A class to hold an array of bytes
  */
-class ByteArray implements \Stringable //@todo might implement ArrayAccess or Traversable
+class ByteArray implements \Stringable, \Countable //@todo might implement ArrayAccess or Traversable
 {
     private $array = [];
 
@@ -89,4 +89,9 @@ private static function assertValidByte($value): void
             );
         }
     }
+
+    public function count(): int
+    {
+        return count($this->array);
+    }
 }
diff --git a/tests/ByteArrayTest.php b/tests/ByteArrayTest.php
@@ -101,4 +101,22 @@ public static function provideAppend() : array
             [ByteArray::fromString('🚀'), ByteArray::fromString('🐸'), ByteArray::fromString('🚀🐸')],
         ];
     }
+
+    /**
+     * @dataProvider provideCount
+     */
+    public function testCount(ByteArray $array, int $expectedCount) : void
+    {
+        $this->assertSame($expectedCount, $array->count());
+        $this->assertSame($expectedCount, count($array));
+    }
+
+    public static function provideCount() : array
+    {
+        return [
+            [ByteArray::fromArray([]), 0],
+            [ByteArray::fromArray([123]), 1],
+            [ByteArray::fromString('🚀'), 4],
+        ];
+    }
 }