Tests: Misc aslts error fixes for Identity2MS test.

Fixes some problems with the test itself, reducing runtime errors.
Chao Guan.