Add test for handshake