Sunderland manager Martin O'Neill has warned James McClean over his behaviour following the Twitter rant that landed the winger in hot water with his international manager.