Test fixes

Since this is a test case, isn't it better to somehow inform about the failure, rather than handling failure gracefully? I assume you might have got this as a usual application logic or am I missing something here?