Add additional tests for connection loss exception handling.