// Simulate different encode return values
            when(mockChunk1.encode(any(byte[].class), anyInt())).thenReturn(20);
            when(mockChunk2.encode(any(byte[].class), anyInt())).thenReturn(24);
            when(mockChunk3.encode(any(byte[].class), anyInt())).thenReturn(22);

            SrvCopychunkCopy copy = new SrvCopychunkCopy(sourceKey, mockChunk1, mockChunk2, mockChunk3);

            byte[] expectedName = "TEST_CONTEXT".getBytes(StandardCharsets.UTF_8);
            CreateContextRequest request = mock(CreateContextRequest.class);
            when(request.getName()).thenReturn(expectedName);

            byte[] actualName = request.getName();
            assertArrayEquals(expectedName, actualName);
            verify(request, times(1)).getName();
        }

        @Test

        NetbiosName mockNetbiosName = mock(NetbiosName.class);
        when(mockNetbiosName.getName()).thenReturn("MOCKNAME");
        when(mockNetbiosName.getNameType()).thenReturn(0x20);
        when(mockNetbiosName.getScope()).thenReturn("mock.scope");

        Name name = new Name(mockConfig, mockNetbiosName);

        assertEquals("MOCKNAME", name.name);

    void setUp() {
        // Setup default mock behavior
        lenient().when(mockConfig.getNetbiosScope()).thenReturn("DEFAULT.SCOPE");
        lenient().when(mockConfig.getOemEncoding()).thenReturn("UTF-8");

        // Setup mock name
        lenient().when(mockName.writeWireFormat(any(byte[].class), anyInt())).thenReturn(34);
        mockName.hexCode = 0x20;
        mockName.name = "TEST";
        mockName.scope = "test.scope";
    }

        transport.server = serverData;

        smbtStatic = mockStatic(SmbTransport.class);
        smbtStatic.when(() -> SmbTransport.getSmbTransport(addr, 445, inet, 0, null)).thenReturn(transport);
        smbtStatic.when(() -> SmbTransport.getSmbTransport(addr, 445)).thenReturn(transport);
    }

    @AfterEach
    void tearDown() {
        if (smbtStatic != null) {
            smbtStatic.close();
        }
    }

    @Test

        failedChannel = new ChannelInfo("failed-channel", mockTransport, localInterface, remoteInterface);
        failedChannel.setState(ChannelState.ESTABLISHED);

        when(mockChannelManager.getLoadBalancer()).thenReturn(mockLoadBalancer);
    }

    @Test
    void testHandleFailure() {
        IOException error = new IOException("Connection failed");

        failover.handleFailure(failedChannel, error);

                        Mockito.any(PathScope.class)))
                .thenReturn(resolverResult);
        Node node = Mockito.mock(Node.class);
        Mockito.when(resolverResult.getRoot()).thenReturn(node);
        Node child = Mockito.mock(Node.class);
        Mockito.when(node.getChildren()).thenReturn(List.of(child));

        services.addAll(List.of(
                new DefaultRequestCacheFactory(),

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockConfig.getPid()).thenReturn(1234);
        when(mockConfig.getMaximumBufferSize()).thenReturn(65535);
        when(mockConfig.getMinimumVersion()).thenReturn(DialectVersion.SMB1);
        when(mockConfig.getMaximumVersion()).thenReturn(DialectVersion.SMB311);
        response = new TestSmbComNtTransactionResponse(mockConfig);
    }

    @Test

        // Given
        byte[] buffer = new byte[10];
        int bufferIndex = 0;
        int len = 10;
        when(mockDecodable.decode(buffer, bufferIndex, len)).thenReturn(10);

        // When
        int decodedLength = mockDecodable.decode(buffer, bufferIndex, len);

        // Then
        assertEquals(10, decodedLength);

        }
    }

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
        when(mockConfig.getTransactionBufferSize()).thenReturn(65535);
        when(mockConfig.getOemEncoding()).thenReturn("ASCII");
        transaction = new TestSmbComTransaction(mockConfig, SmbComTransaction.SMB_COM_TRANSACTION, SmbComTransaction.TRANS2_FIND_FIRST2);
    }

    @Test