/**
  * @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);
 }