Leaving the world and re-entering dont give you protection
MC-278867 Damage immunity for 3 seconds after spawning no longer works
MC-279029 Spawn Immunity Removed
MC-212 Fall damage is ignored for a couple of seconds when reloading into LAN or singleplayer worlds