} catch (final IOException ex) {}
        return connection.getResponseCode();
    }

    @Override
    public String getResponseMessage() throws IOException {
        try {
            handshake();
        } catch (final IOException ex) {}
        return connection.getResponseMessage();
    }

    @Override
    public void disconnect() {
        connection.disconnect();
        handshakeComplete = false;

        handshake();
        return this.connection.getResponseCode();
    }

    @Override
    public String getResponseMessage() throws IOException {
        handshake();
        return this.connection.getResponseMessage();
    }

    @Override
    public void disconnect() {
        this.connection.disconnect();
        this.handshakeComplete = false;

        if (conn == null)
            return;

        String statusLine = "HTTP/1.1 " + code + " " + message;
        when(conn.getResponseCode()).thenReturn(code);
        when(conn.getResponseMessage()).thenReturn(message);
        when(conn.getHeaderField(0)).thenReturn(statusLine);

        // Mock getHeaderFields to return our headers