Refactor the usage message unit tests