SitemapSet sitemapSet = new SitemapSet();
        sitemapSet.addSitemap(sitemap);

        when(crawlerContainer.getComponent("sitemapsHelper")).thenReturn(sitemapsHelper);
        try {
            when(sitemapsHelper.parse(any(InputStream.class))).thenReturn(sitemapSet);
        } catch (Exception e) {
            fail("Should not throw exception in test setup");
        }

        when(ctx.getConfig()).thenReturn(config);
        when(ctx.getNameServiceClient()).thenReturn(nameSvc);
        when(ctx.getCredentials()).thenReturn(creds);
        when(ctx.getTransportPool()).thenReturn(pool);

        // Default config values
        when(config.getLocalAddr()).thenReturn(null);
        when(config.getLocalPort()).thenReturn(0);
        when(config.getSessionLimit()).thenReturn(10);

        when(mockConfig.isDfsDisabled()).thenReturn(false);
        when(mockConfig.isEncryptionEnabled()).thenReturn(true);
        when(mockConfig.getMinimumVersion()).thenReturn(DialectVersion.SMB202);
        when(mockConfig.getMaximumVersion()).thenReturn(DialectVersion.SMB311);
        when(mockConfig.getMachineId()).thenReturn(testMachineId);
        when(mockConfig.getRandom()).thenReturn(mockRandom);

        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

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

            // Given
            when(mockResource.openOutputStream()).thenReturn(mockOutputStream);
            when(mockResource.openOutputStream(anyBoolean())).thenReturn(mockOutputStream);
            when(mockResource.openOutputStream(anyBoolean(), anyInt())).thenReturn(mockOutputStream);
            when(mockResource.openOutputStream(anyBoolean(), anyInt(), anyInt(), anyInt())).thenReturn(mockOutputStream);

            // When

            when(mockStrategies.get(0).isApplicable(context)).thenReturn(true);
            when(mockStrategies.get(0).apply(Mockito.eq(context), Mockito.any()))
                    .thenReturn(UpgradeResult.empty());

            when(mockStrategies.get(1).isApplicable(context)).thenReturn(false);
            when(mockStrategies.get(2).isApplicable(context)).thenReturn(false);

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