Improving code coverage

thecodingmachine · Sep 30, 2019 · 4525540 · 4525540
1 parent 09b4dab
commit 4525540
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/tests/EmptyInnerResultIteratorTest.php b/tests/EmptyInnerResultIteratorTest.php
@@ -38,7 +38,10 @@ public function testIterate()
         foreach ($iterator as $elem) {
             $this->fail('Iterator should be empty');
         }
-        $this->assertTrue(true);
+
+        $iterator->next();
+        $this->assertNull($iterator->current());
+        $this->assertNull($iterator->key());
     }
 
     public function testOffsetGet()

diff --git a/tests/MapIteratorTest.php b/tests/MapIteratorTest.php
@@ -21,10 +21,31 @@
 
 namespace TheCodingMachine\TDBM;
 
+use ArrayIterator;
+use IteratorAggregate;
 use PHPUnit\Framework\TestCase;
 
 class MapIteratorTest extends TestCase
 {
+    public function testIteratorAggregate(): void
+    {
+        $mapIterator = new MapIterator(new class implements IteratorAggregate
+            {
+                public $property1 = "Public property one";
+                public $property2 = "Public property two";
+                public $property3 = "Public property three";
+
+                public function getIterator()
+                {
+                    return new ArrayIterator($this);
+                }
+            }, function ($item) {
+                return $item;
+            });
+
+        self::assertCount(3, $mapIterator);
+    }
+
     public function testConstructorException1(): void
     {
         $this->expectException('TheCodingMachine\TDBM\TDBMException');