throw new RuntimeException("Failed to inject mock CIFSContext", e);
        }

        when(cifsContext.getTransportPool()).thenReturn(transportPool);
        when(cifsContext.getNameServiceClient()).thenReturn(nameServiceClient);
        lenient().when(cifsContext.getConfig()).thenReturn(configuration);
        lenient().when(configuration.getDefaultDomain()).thenReturn("TEST_DOMAIN");

        // Create a fresh pool instance for each test
        pool = new SmbTransportPoolImpl();

        // Setup default mock behaviors
        when(ctx.getConfig()).thenReturn(config);
        when(ctx.getNameServiceClient()).thenReturn(nameSvc);
        when(ctx.getCredentials()).thenReturn(creds);
        when(ctx.getTransportPool()).thenReturn(pool);

        // Default config values

        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) -> {

            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

            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();