Add test for disconnected client