when(mockConfig.getMaximumVersion()).thenReturn(DialectVersion.SMB311);
        when(mockConfig.getMachineId()).thenReturn(testMachineId);
        when(mockConfig.getRandom()).thenReturn(mockRandom);
        when(mockContext.getConfig()).thenReturn(mockConfig);

        // Mock random for predictable salt generation
        doAnswer(invocation -> {
            byte[] buffer = invocation.getArgument(0);

        void testConstructorWithConfig() {
            ServerMessageBlock2 msg = new TestServerMessageBlock2(mockConfig);
            assertNotNull(msg);
            assertEquals(mockConfig, msg.getConfig());
            assertEquals(0, msg.getCommand());
        }

        @Test
        @DisplayName("Should create with configuration and command")
        void testConstructorWithConfigAndCommand() {

    @Override
    public WebConfigDbm asDBMeta() {
        return WebConfigDbm.getInstance();
    }

    @Override
    public String asTableDbName() {
        return "web_config";
    }

    // ===================================================================================
    //                                                                              Source

        ntlmSsp = new NtlmSsp();

        // Use lenient stubbing to avoid UnnecessaryStubbing errors for tests that don't need all mocks
        lenient().when(mockCifsContext.getConfig()).thenReturn(mockConfig);
        lenient().when(mockConfig.getDefaultDomain()).thenReturn("DOMAIN");
        lenient().when(mockConfig.isUseUnicode()).thenReturn(true);

        void testConstructor() {
            Smb2EchoResponse response = new Smb2EchoResponse(mockConfig);
            assertNotNull(response);
            assertEquals(mockConfig, response.getConfig());
        }

        @Test
        @DisplayName("Should extend ServerMessageBlock2Response")
        void testInheritance() {
            assertTrue(echoResponse instanceof ServerMessageBlock2Response);
        }

        Provider<CopyRestApiTask> copyCompatYamlSpecTask = project.getTasks()
            .register("copyRestCompatApiTask", CopyRestApiTask.class, task -> {
                task.dependsOn(bwcMinorConfig);
                task.setConfig(bwcMinorConfig);
                task.setAdditionalConfig(bwcMinorConfig);
                task.getInclude().set(extension.getRestApi().getInclude());

        tasks.register("checkstyleApi", Checkstyle.class, task -> {
            task.source(extension.getDocumentedSource());
            // TODO: This is ugly
            task.setConfig(project.getResources().getText().fromFile(checkstyle.getConfigDirectory().file("checkstyle-api.xml")));
            task.setClasspath(layout.files());

    @BeforeEach
    void setUp() {
        // 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);

        final UniAddress dc = new UniAddress(addr);
        try (SmbTransportInternal trans = tf.getTransportPool()
                .getSmbTransport(tf, dc, 0, false, tf.hasDefaultCredentials() && tf.getConfig().isIpcSigningEnforced())
                .unwrap(SmbTransportInternal.class)) {
            if (!tf.hasDefaultCredentials()) {
                trans.ensureConnected();
                log.warn("""

        mockContext = Mockito.mock(CIFSContext.class);
        mockConfig = Mockito.mock(Configuration.class);
        mockCredentials = Mockito.mock(Credentials.class);

        Mockito.when(mockContext.getConfig()).thenReturn(mockConfig);
        Mockito.when(mockContext.getCredentials()).thenReturn(mockCredentials);

        // Mock CredentialsInternal for session tests