/** * Create source adapter mock and set it into model object which tested in this class * * @param array $columns value which will be returned by method getColNames() * @return Mage_ImportExport_Model_Import_SourceAbstract|PHPUnit_Framework_MockObject_MockObject */ protected function _createSourceAdapterMock(array $columns) { /** @var $source Mage_ImportExport_Model_Import_SourceAbstract|PHPUnit_Framework_MockObject_MockObject */ $source = $this->getMockForAbstractClass('Mage_ImportExport_Model_Import_SourceAbstract', array(), '', false, true, true, array('getColNames')); $source->expects($this->any())->method('getColNames')->will($this->returnValue($columns)); $this->_model->setSource($source); return $source; }
/** * Source model setter * * @param Mage_ImportExport_Model_Import_SourceAbstract $source * @return Mage_ImportExport_Model_Import_EntityAbstract */ public function setSource(Mage_ImportExport_Model_Import_SourceAbstract $source) { $this->_customerEntity->setSource($source); $this->_addressEntity->setSource($source); return parent::setSource($source); }