Adds tests for iterator readers

src/Reader/CountableIteratorReader.php

@@ -20,10 +20,12 @@ class CountableIteratorReader extends IteratorReader implements CountableReader
      */
     public function count()
     {
-        if ($this->iterator instanceof \Countable) {
-            return count($this->iterator);
+        $iterator = $this->getInnerIterator();
+
+        if ($iterator instanceof \Countable) {
+            return count($iterator);
         }
 
-        return iterator_count($this->iterator);
+        return iterator_count($iterator);
     }
 }

tests/Reader/CountableIteratorReaderTest.php

@@ -0,0 +1,44 @@
+<?php
+
+namespace Ddeboer\DataImport\Tests\Reader;
+
+use Ddeboer\DataImport\Reader\CountableIteratorReader;
+
+/**
+ * @author Márk Sági-Kazár <mark.sagikazar@gmail.com>
+ */
+class CountableIteratorReaderTest extends \PHPUnit_Framework_TestCase
+{
+    public function testCount()
+    {
+        $iterator = new \ArrayIterator([
+            [
+                'id'       => 1,
+                'username' => 'john.doe',
+                'name'     => 'John Doe',
+            ],
+        ]);
+
+        $reader = new CountableIteratorReader($iterator);
+
+        // We need to rewind the iterator
+        $reader->rewind();
+
+        $this->assertEquals(1, $reader->count());
+    }
+
+    public function testIteratorCount()
+    {
+        $reader = new CountableIteratorReader(new CountableIterator);
+
+        // We need to rewind the iterator
+        $reader->rewind();
+
+        $this->assertEquals(0, $reader->count());
+    }
+}
+
+class CountableIterator extends \EmptyIterator
+{
+
+}

tests/Reader/IteratorReaderTest.php

@@ -0,0 +1,32 @@
+<?php
+
+namespace Ddeboer\DataImport\Tests\Reader;
+
+use Ddeboer\DataImport\Reader\IteratorReader;
+
+/**
+ * @author Márk Sági-Kazár <mark.sagikazar@gmail.com>
+ */
+class IteratorReaderTest extends \PHPUnit_Framework_TestCase
+{
+    public function testGetFields()
+    {
+        $iterator = new \ArrayIterator([
+            [
+                'id'       => 1,
+                'username' => 'john.doe',
+                'name'     => 'John Doe',
+            ],
+        ]);
+
+        $reader = new IteratorReader($iterator);
+
+        // We need to rewind the iterator
+        $reader->rewind();
+
+        $fields = $reader->getFields();
+
+        $this->assertInternalType('array', $fields);
+        $this->assertEquals(['id', 'username', 'name'], $fields);
+    }
+}