Added unit test for #338

Author: ezimuel

test/unit/TableGateway/AbstractTableGatewayTest.php

@@ -84,6 +84,8 @@ protected function setUp()
                 ->getMock()
         ));
 
+        $this->mockFeatureSet = $this->getMockBuilder('Zend\Db\TableGateway\Feature\FeatureSet')->getMock();
+
         $this->table = $this->getMockForAbstractClass(
             'Zend\Db\TableGateway\AbstractTableGateway'
             //array('getTable')
@@ -104,6 +106,9 @@ protected function setUp()
                 case 'sql':
                     $tgPropReflection->setValue($this->table, $this->mockSql);
                     break;
+                case 'featureSet':
+                    $tgPropReflection->setValue($this->table, $this->mockFeatureSet);
+                    break;
             }
         }
     }
@@ -369,6 +374,30 @@ public function testGetLastInsertValue()
         self::assertEquals(10, $this->table->getLastInsertValue());
     }
 
+    public function testInitializeBuildsAResultSet()
+    {
+        $stub = $this->getMockForAbstractClass(AbstractTableGateway::class);
+
+        $tgReflection = new \ReflectionClass('Zend\Db\TableGateway\AbstractTableGateway');
+        foreach ($tgReflection->getProperties() as $tgPropReflection) {
+            $tgPropReflection->setAccessible(true);
+            switch ($tgPropReflection->getName()) {
+                case 'table':
+                    $tgPropReflection->setValue($stub, 'foo');
+                    break;
+                case 'adapter':
+                    $tgPropReflection->setValue($stub, $this->mockAdapter);
+                    break;
+                case 'featureSet':
+                    $tgPropReflection->setValue($stub, $this->mockFeatureSet);
+                    break;
+            }
+        }
+
+        $stub->initialize();
+        $this->assertInstanceOf(ResultSet::class, $stub->getResultSetPrototype());
+    }
+
     /**
      * @covers \Zend\Db\TableGateway\AbstractTableGateway::__get
      */