Martin O'Neill leaving Celtic would not surprise Paul Lambert, but the former club captain would be "shocked and disappointed" if the veteran manager was to be sacked.