/** * @param array $map * @param BaseElementModel|MockObject $mockElement */ private function setAttributesMockElementCriteria(array $map, BaseElementModel $mockElement) { $mockElementCriteria = $this->getMockElementCriteria(); if (array_key_exists('parent', $map)) { $mockElementCriteria->expects($this->exactly(1))->method('first')->willReturn('parent'); } if (array_key_exists('ancestors', $map)) { $mockElementCriteria->expects($this->exactly(1))->method('find')->willReturn(array('ancestor1', 'ancestor2')); } $mockElement->expects($this->any())->method('getAncestors')->willReturn($mockElementCriteria); }