Wardens have been made immune to lava however they can't swim in it properly. This isn't in parity with Java.

Steps to reproduce:

1. Place lava on the ground.
2. Place a warden in the lava.
3. Have a mob or player (must be in survival) make enough vibrations to anger the warden.
4. Observe the warden not move in the lava after being angered.