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

        when(mockConnection.getRequestMethod()).thenReturn("GET");
        when(mockConnection.getAllowUserInteraction()).thenReturn(true);
        when(mockConnection.getDoInput()).thenReturn(true);
        when(mockConnection.getDoOutput()).thenReturn(true);
        when(mockConnection.getIfModifiedSince()).thenReturn(12345L);
        when(mockConnection.getUseCaches()).thenReturn(false);

    @BeforeEach
    void setUp() throws Exception {
        parent = mock(SmbFile.class);
        when(parent.getLocator()).thenReturn(locator);
        when(treeHandle.getConfig()).thenReturn(config);
        when(treeHandle.acquire()).thenReturn(treeHandle);
        when(treeHandle.getOEMDomainName()).thenReturn("TESTDOMAIN");
    }

    @Test
    @DisplayName("Constructor should throw SmbException for non-workgroup type with host")

        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

        lenient().when(mockSmbPipeHandle.unwrap(SmbPipeHandleInternal.class)).thenReturn(mockSmbPipeHandleInternal);
        lenient().when(mockSmbNamedPipe.getContext()).thenReturn(mockContext);
        lenient().when(mockSmbNamedPipe.getLocator()).thenReturn(mockSmbResourceLocator);
        lenient().when(mockDcerpcBinding.getServer()).thenReturn(TEST_SERVER);

    }

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

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

        // Assert
        assertNotNull(actualAddress);

        lenient().when(mockContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.getWitnessHeartbeatTimeout()).thenReturn(120000L);
        lenient().when(mockConfig.getWitnessRegistrationTimeout()).thenReturn(300000L);
        lenient().when(mockConfig.getWitnessReconnectDelay()).thenReturn(1000L);
        lenient().when(mockConfig.isWitnessServiceDiscovery()).thenReturn(true);
    }

    @Test

        when(mockConfig.getNetbiosLocalAddress()).thenReturn(InetAddress.getByName("127.0.0.1"));
        when(mockConfig.getBroadcastAddress()).thenReturn(InetAddress.getByName("255.255.255.255"));
        when(mockConfig.getNetbiosSndBufSize()).thenReturn(576);
        when(mockConfig.getNetbiosRcvBufSize()).thenReturn(576);
        when(mockConfig.getResolveOrder()).thenReturn(Arrays.asList(ResolverType.RESOLVER_DNS)); // Only DNS, no broadcast

    void testGetKey_KeyExists() throws LoginException {
        when(key1.getKeyType()).thenReturn(KEY_TYPE_1);
        when(key2.getKeyType()).thenReturn(KEY_TYPE_2);
        Set<Object> privateCredentials = new HashSet<>();
        privateCredentials.add(key1);
        privateCredentials.add(key2);
        when(subject.getPrivateCredentials()).thenReturn(privateCredentials);

    @BeforeEach
    void setUp() {
        mockDelegate = mock(CIFSContext.class);
        mockConfiguration = mock(Configuration.class);
        when(mockDelegate.getConfig()).thenReturn(mockConfiguration);
        when(mockConfiguration.isTraceResourceUsage()).thenReturn(false); // Default to false or true based on expected behavior
        cifsContextWrapper = new CIFSContextWrapper(mockDelegate);
    }

    @Test
    void testConstructor() {