Enhanced unit test support class