Merge pull request #14 from irfanh94/fix_word_collection_iterator

fix word collection iterator and add tests

Author: irfanh94

src/Collection/WordCollection.php

@@ -29,12 +29,12 @@ public function next(): void {
         ++$this->iteratorPosition;
     }
 
-    public function key(): mixed {
+    public function key(): int {
         return $this->iteratorPosition;
     }
 
     public function valid(): bool {
-        return isset($this->iteratorPosition);
+        return isset($this->items[$this->iteratorPosition]);
     }
 
     public function rewind(): void {

tests/Unit/Collection/WordCollectionTest.php

@@ -0,0 +1,40 @@
+<?php
+
+declare(strict_types=1);
+
+namespace WordCounter\Tests\Unit\Collection;
+
+use PHPUnit\Framework\TestCase;
+use WordCounter\Collection\Word;
+use WordCounter\Collection\WordCollection;
+
+class WordCollectionTest extends TestCase {
+
+    public function testCanIterateWords(): void {
+        $expected = [
+            new Word('first', 5),
+            new Word('second', 6),
+        ];
+
+        $wordCollection = new WordCollection($expected);
+
+        $actual = [];
+        foreach ($wordCollection as $word) {
+            $actual[] = $word;
+        }
+
+        self::assertEquals($expected, $actual);
+    }
+
+    public function testCanOutputArray(): void {
+        $words = [
+            new Word('first', 5),
+            new Word('second', 6),
+        ];
+
+        $wordCollection = new WordCollection($words);
+
+        self::assertEquals(['first', 'second'], $wordCollection->toArray());
+    }
+
+}