Add test for connection state reset