/** * @param string $label * @param boolean $withAttribute * * @return ColumnInfoInterface */ protected function getColumnInfoMock($label, $withAttribute = true) { $column = parent::getColumnInfoMock($label); $column->expects($this->any())->method('getName')->will($this->returnValue($label . '_name')); $column->expects($this->any())->method('getLocale')->will($this->returnValue('locale')); $column->expects($this->any())->method('getScope')->will($this->returnValue('scope')); if ($withAttribute) { $attribute = $this->getMock('Pim\\Bundle\\CatalogBundle\\Entity\\Attribute'); $attribute->expects($this->any())->method('getCode')->will($this->returnValue($label . '_code')); $column->expects($this->any())->method('getAttribute')->will($this->returnValue($attribute)); } return $column; }
/** * {@inheritdoc} */ protected function setUp() { parent::setUp(); $this->importValidator = new ImportValidator($this->validator); $this->entity = $this->getMock('Pim\\Bundle\\CatalogBundle\\Model\\ReferableInterface'); }