在《西游记》的第96到100回中，唐僧一行人到达了火焰山，面对烈火的考验，孙悟空化身火神，八戒化身水神，成功将火焰扑灭。接着，他们遇到了牛魔王，孙悟空与牛魔王大战三百回合，最终化解了彼此的仇恨，成为好友。