Enhance WhitespaceAroundCheck

Enhance WhitespaceAroundCheck to check DO_WHILE and LITERAL_SWITCH
tokens.