if (this.loadBalance) {
                dc = new UniAddress(getTransportContext().getNameServiceClient().getNbtByName(this.domainController, 0x1C, null));
            } else {
                dc = getTransportContext().getNameServiceClient().getByName(this.domainController, true);
            }
            NtlmPasswordAuthentication ntlm;
            if (msg.startsWith("NTLM ")) {

     */
    Configuration getConfig();

    /**
     * Get the name service client for NetBIOS name resolution
     *
     * @return the name server client
     */
    NameServiceClient getNameServiceClient();

    /**
     * Get the buffer cache for efficient memory management
     *
     * @return the buffer cache
     */
    BufferCache getBufferCache();

    /**

        }

        @Override
        public BufferCache getBufferCache() {
            return null; // Not relevant for AbstractCIFSContext tests
        }

        @Override
        public NameServiceClient getNameServiceClient() {
            return null; // Not relevant for AbstractCIFSContext tests
        }

        @Override
        public Configuration getConfig() {

        lenient().when(mockConfig.getMachineId()).thenReturn(machineId);
        lenient().when(mockCtx.getConfig()).thenReturn(mockConfig);
        lenient().when(mockCtx.getNameServiceClient()).thenReturn(mockNameServiceClient);
        lenient().when(mockNameServiceClient.getLocalHost()).thenReturn(mockHost);
        lenient().when(mockHost.getHostName()).thenReturn("TEST_HOSTNAME");

        return mockCtx;

    void setUp() {
        // Setup mock context with lenient stubbing to avoid UnnecessaryStubbingException
        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockContext.getNameServiceClient()).thenReturn(mockNameServiceClient);
        lenient().when(mockNameServiceClient.getLocalHost()).thenReturn(mockLocalHost);
        lenient().when(mockLocalHost.getHostName()).thenReturn("localhost");

        }
    };

    @BeforeEach
    void setup() {
        when(ctx.getConfig()).thenReturn(config);
        when(ctx.getCredentials()).thenReturn(creds);
        when(ctx.getNameServiceClient()).thenReturn(nsc);
    }

    private static URL smbUrl(String spec) {
        try {
            return new URL(null, spec, SMB_HANDLER);
        } catch (Exception e) {

                log.debug("Failed to get domain bytes", ex);
            }
        }
        final int domainLength = domain.length;
        byte[] server = {};
        final String host = tc.getNameServiceClient().getLocalHost().getHostName();
        if (host != null) {
            try {
                server = host.getBytes(UNI_ENCODING);
            } catch (final IOException ex) {

     * environment.
     *
     * @param tc
     *            context to use
     */
    public Type1Message(final CIFSContext tc) {
        this(tc, getDefaultFlags(tc), tc.getConfig().getDefaultDomain(), tc.getNameServiceClient().getLocalHost().getHostName());
    }

    /**
     * Creates a Type-1 message with the specified parameters.
     *
     * @param tc
     *            context to use
     * @param flags

            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