Add a couple more tests for connection state handling.