}

    void setUid(int uid) {
        this.uid = uid;
    }

    void setSessionSetup(Smb2SessionSetupResponse response) {
        this.extendedSecurity = true;
        this.connectionState.set(2);
        this.sessionId = response.getSessionId();
    }

    void setSessionSetup(SmbComSessionSetupAndXResponse response) {
        this.extendedSecurity = response.isExtendedSecurity();

    void testConnectionAndFailureStatus() {
        SmbSessionImpl session = newSession();

        // Make session look connected via setSessionSetup
        Smb2SessionSetupResponse resp = mock(Smb2SessionSetupResponse.class);
        when(resp.getSessionId()).thenReturn(42L);
        session.setSessionSetup(resp);

        when(transport.isDisconnected()).thenReturn(false);
        when(transport.isFailed()).thenReturn(false);