Change the Yarn-RM state mapping after NEW_SAVING is added on the Yarn
part.
Simplify the test.
Fix 80 char line.