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

        doNothing().when(messageBlock).reset();

        when(messageBlock.getMid()).thenReturn(mid);
        when(messageBlock.getCommand()).thenReturn(command);
        when(messageBlock.getDigest()).thenReturn(digest);
        when(messageBlock.getResponse()).thenReturn(response);

        // When - simulate a typical usage sequence
        messageBlock.setMid(mid);

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

        // Test different cases
        when(mockNetbiosName.getName()).thenReturn(upperCase);
        assertEquals(upperCase, mockNetbiosName.getName());

        when(mockNetbiosName.getName()).thenReturn(lowerCase);
        assertEquals(lowerCase, mockNetbiosName.getName());

        when(mockNetbiosName.getName()).thenReturn(mixedCase);
        assertEquals(mixedCase, mockNetbiosName.getName());
    }

        when(project.getGroupId()).thenReturn("org.apache.maven");
        when(project.getArtifactId()).thenReturn("maven-embedder");
        when(project.getPackaging()).thenReturn("jar");
        when(project.getName()).thenReturn("Apache Maven Embedder");
        when(project.getVersion()).thenReturn("3.5.4-SNAPSHOT");
        when(project.getFile()).thenReturn(new File(basedir, "maven-embedder/pom.xml"));

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

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

        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

    void setUp() {
        lenient().when(mockFileEntry1.getName()).thenReturn("file1.txt");
        lenient().when(mockFileEntry2.getName()).thenReturn("file2.txt");
        lenient().when(mockFileEntry3.getName()).thenReturn("file3.txt");

        lenient().when(mockResource1.getName()).thenReturn("file1.txt");
        lenient().when(mockResource2.getName()).thenReturn("file2.txt");

            // 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