protected function addColumn($name, $addTransformer = true, $addField = true, $skip = false, $failing = false, $suffixes = array()) { $label = implode('_', array_merge(array($name), $suffixes)); $column = parent::addColumn($label, $addTransformer, $skip, $failing, $suffixes); $column->expects($this->any())->method('getName')->will($this->returnValue($name)); if ($addField) { $this->fields[] = $column->getPropertyPath(); } else { $column->expects($this->any())->method('setAttribute')->with($this->equalTo($this->attributes[$label])); } }