feat: make MessagesFlattener countable

CuyZ · Jul 26, 2022 · 2c1c7cf · 2c1c7cf
1 parent 0b37b48
commit 2c1c7cf
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/src/Mapper/Tree/Message/MessagesFlattener.php b/src/Mapper/Tree/Message/MessagesFlattener.php
@@ -4,12 +4,14 @@
 
 namespace CuyZ\Valinor\Mapper\Tree\Message;
 
+use Countable;
 use CuyZ\Valinor\Mapper\Tree\Node;
 use CuyZ\Valinor\Mapper\Tree\NodeTraverser;
 use IteratorAggregate;
 use Traversable;
 
 use function array_filter;
+use function count;
 
 /**
  * Will recursively flatten messages of a node and all its children.
@@ -34,7 +36,7 @@
  *
  * @implements IteratorAggregate<NodeMessage>
  */
-final class MessagesFlattener implements IteratorAggregate
+final class MessagesFlattener implements IteratorAggregate, Countable
 {
     /** @var array<NodeMessage> */
     private array $messages = [];
@@ -58,6 +60,11 @@ public function errors(): self
         return $clone;
     }
 
+    public function count(): int
+    {
+        return count($this->messages);
+    }
+
     /**
      * @return Traversable<NodeMessage>
      */

diff --git a/tests/Unit/Mapper/Tree/Message/MessagesFlattenerTest.php b/tests/Unit/Mapper/Tree/Message/MessagesFlattenerTest.php
@@ -36,6 +36,8 @@ public function test_messages_are_filtered_and_can_be_iterated_through(): void
 
         $messages = [...(new MessagesFlattener($node))->errors()];
 
+        self::assertCount(2, $messages);
+
         self::assertSame('some error message B', (string)$messages[0]);
         self::assertSame('some error message A', (string)$messages[1]);
     }