lenient().when(mockConfig.isUseUnicode()).thenReturn(true);

        // Mock NameServiceClient for Type1Message test
        lenient().when(mockCifsContext.getNameServiceClient()).thenReturn(mockNameServiceClient);
        lenient().when(mockNameServiceClient.getLocalHost()).thenReturn(mockAddress);
        lenient().when(mockAddress.getHostName()).thenReturn("WORKSTATION");
    }

        when(mockCifsContext.getCredentials()).thenReturn(creds);
        when(mockCifsContext.getConfig()).thenReturn(new PropertyConfiguration(System.getProperties()));
        when(mockCifsContext.getNameServiceClient()).thenReturn(mockNameServiceClient);

        ntlmConnection = new NtlmHttpURLConnection(mockConnection, mockCifsContext);
    }

    /**

    public SmbTransportImpl getSmbTransport(final CIFSContext tf, final String name, final int port, final boolean exclusive,
            final boolean forceSigning) throws IOException {

        final Address[] addrs = tf.getNameServiceClient().getAllByName(name, true);

        if (addrs == null || addrs.length == 0) {
            throw new UnknownHostException(name);
        }

        Arrays.sort(addrs, (o1, o2) -> {

            if (msg.startsWith("NTLM ")) {
                byte[] challenge;
                final NameServiceClient nameServiceClient = getTransportContext().getNameServiceClient();
                if (pathInfo == null || server == null) {
                    final String mb = nameServiceClient.getNbtByName(NbtAddress.MASTER_BROWSER_NAME, 0x01, null).getHostAddress();

            this.scope = scope;
            this.svr = svr;
            this.tc = tc;
        }

        @Override
        public void run() {
            try {
                this.ans = this.tc.getNameServiceClient().getNbtAllByName(this.host, this.type, this.scope, this.svr);
            } catch (final UnknownHostException ex) {
                this.uhe = ex;
            } catch (final Exception ex) {

                final Type2Message type2 = (Type2Message) message;
                message = new Type3Message(this.transportContext, type2, null, password, domain, user,
                        this.transportContext.getNameServiceClient().getLocalHost().getHostName(), 0);
            }
            return message;
        }
    }

    private void reconnect() throws IOException {
        final int readTimeout = getReadTimeout();

    public Type3Message(final CIFSContext tc) {
        setFlags(getDefaultFlags(tc));
        setDomain(tc.getConfig().getDefaultDomain());
        setUser(tc.getConfig().getDefaultUsername());
        setWorkstation(tc.getNameServiceClient().getLocalHost().getHostName());
    }

    /**
     * Creates a Type-3 message in response to the given Type-2 message.
     *
     * @param tc
     *            context to use
     * @param type2

                    // Cache workgroup resolution for better performance
                    getContext().getNameServiceClient().getByName(this.url.getHost(), true);
                } else {
                    // Cache server name resolution
                    getContext().getNameServiceClient().getByName(this.url.getHost()).getHostName();
                }
            } else {

            this.out = this.socket.getOutputStream();
            this.in = this.socket.getInputStream();

            final SessionServicePacket ssp = new SessionRequestPacket(tc.getConfig(), calledName, tc.getNameServiceClient().getLocalName());
            this.out.write(this.sbuf, 0, ssp.writeWireFormat(this.sbuf, 0));
            if (readn(this.in, this.sbuf, 0, 4) < 4) {
                try {
                    this.socket.close();