});
    }

    @Test
    @DisplayName("Should get server")
    void testGetServer() {
        // Given
        String server = "testserver";
        when(mockReferralData.getServer()).thenReturn(server);

        // When
        String result = mockReferralData.getServer();

        // Then
        assertEquals(server, result);
        verify(mockReferralData).getServer();
    }

    @Test

    @BeforeEach
    void setUp() throws Exception {
        when(ctx.getConfig()).thenReturn(cfg);
        when(cfg.isSigningEnforced()).thenReturn(false);
        when(cfg.getSessionTimeout()).thenReturn(30_000);
        when(cfg.getResponseTimeout()).thenReturn(5_000);
        when(address.getHostAddress()).thenReturn("127.0.0.1");
        when(address.getHostName()).thenReturn("localhost");

    }

    @Test
    void testGetLocalHost() {
        // Arrange
        NetbiosAddress expectedAddress = mock(NetbiosAddress.class);
        when(nameServiceClient.getLocalHost()).thenReturn(expectedAddress);

        // Act
        NetbiosAddress actualAddress = nameServiceClient.getLocalHost();

        // Assert
        assertNotNull(actualAddress);

        mockConfig = mock(Configuration.class);
        when(mockContext.getConfig()).thenReturn(mockConfig);
    }

    @Test
    void testGetByNameWithNullLmHostsFile() {
        // Test when lmhosts file is not configured
        when(mockConfig.getLmHostsFileName()).thenReturn(null);

        NbtAddress result = lmhosts.getByName("TEST_HOST", mockContext);

        assertNull(result);
    }

    @Test
    @DisplayName("isSigningOptional and isSigningEnforced reflect configuration")
    void signingModes() throws Exception {
        when(transport.isSigningOptional()).thenReturn(true);
        when(transport.isSigningEnforced()).thenReturn(false);

        assertTrue(transport.isSigningOptional());
        assertFalse(transport.isSigningEnforced());
        verify(transport).isSigningOptional();

        when(pipe.ensureTreeConnected()).thenReturn(tree);
        when(tree.acquire()).thenReturn(tree);
        when(tree.isSMB2()).thenReturn(true);
        when(pipe.openUnshared(eq("\\\\pipe\\\\foo"), eq(0), anyInt(), anyInt(), eq(SmbConstants.ATTR_NORMAL), eq(0))).thenReturn(fh);
        when(fh.acquire()).thenReturn(fh);
        when(fh.isValid()).thenReturn(true);

        // Act

        when(mockDelegate.getOemEncoding()).thenReturn("UTF-8");
        when(mockDelegate.getNetbiosHostname()).thenReturn("TESTHOST");
        when(mockDelegate.getDefaultDomain()).thenReturn("WORKGROUP");
        when(mockDelegate.getDefaultUsername()).thenReturn("testuser");
        when(mockDelegate.getDefaultPassword()).thenReturn("testpass");
        when(mockDelegate.getNativeLanman()).thenReturn("jCIFS");

        // When

        mockNdrBuffer.deferred = mockDeferredBuffer;
        lenient().when(mockNdrBuffer.derive(anyInt())).thenReturn(mockDeferredBuffer);
        lenient().when(mockDeferredBuffer.derive(anyInt())).thenReturn(mockDeferredBuffer);
    }

    @Test
    void testGetSyntax() {
        String syntax = srvsvc.getSyntax();

            // Arrange
            when(mockCtx.getSigningKey()).thenReturn(new byte[] { 7 });
            when(mockCtx.isEstablished()).thenReturn(true);
            when(mockCtx.initSecContext(any(byte[].class), anyInt(), anyInt())).thenReturn(new byte[] { 9 });
            when(mockCtx.getNetbiosName()).thenReturn("NB");
            when(mockCtx.isSupported(any())).thenReturn(true);
            when(mockCtx.isPreferredMech(any())).thenReturn(false);

    void setUp() {
        Arrays.fill(TEST_FILE_ID, (byte) 0x42);
    }

    private void setupMockConfig() {
        when(mockConfig.getTransactionBufferSize()).thenReturn(65536);
    }

    private void setupMockContext() {
        when(mockContext.getConfig()).thenReturn(mockConfig);
    }

    @Test
    @DisplayName("Test constructor with config and control code")
    void testConstructorWithConfigAndControlCode() {