@Test
    void testSuccessfulRegistration() throws Exception {
        // Setup mock RPC response
        WitnessRegisterResponse mockResponse = mock(WitnessRegisterResponse.class);
        lenient().when(mockResponse.isSuccess()).thenReturn(true);
        lenient().when(mockResponse.getRegistrationId()).thenReturn("test-reg-123");

        WitnessRpcClient mockRpc = mock(WitnessRpcClient.class);

 */
public class DirectoryLeaseManagerTest {

    @Mock
    private CIFSContext context;

    @Mock
    private Configuration config;

    @Mock
    private LeaseManager baseLeaseManager;

    @Mock
    private SmbFile mockFile1;

    @Mock
    private SmbFile mockFile2;

    private DirectoryLeaseManager directoryLeaseManager;
    private AutoCloseable mocks;

    @BeforeEach
    public void setUp() {

class SmbTransportPoolImplTest {

    private SmbTransportPoolImpl pool;

    @Mock
    private CIFSContext ctx;
    @Mock
    private Configuration config;
    @Mock
    private NameServiceClient nameSvc;
    @Mock
    private Credentials creds;
    @Mock
    private Address address;
    @Mock
    private SmbNegotiationResponse negotiationResponse;

    @BeforeEach
    void setUp() {

        serverThread = new Thread(this::runServer, "MockWitnessService");
        serverThread.setDaemon(true);
        serverThread.start();

        log.info("Mock witness service started on port {}", serverSocket.getLocalPort());
    }

    /**
     * Stop the mock witness service
     */
    public void stop() {
        running.set(false);

        if (serverSocket != null && !serverSocket.isClosed()) {
            try {

/**
 * Unit tests for ChannelLoadBalancer
 */
@ExtendWith(MockitoExtension.class)
class ChannelLoadBalancerTest {

    @Mock
    private ChannelManager mockChannelManager;

    @Mock
    private SmbTransport mockTransport1;

    @Mock
    private SmbTransport mockTransport2;

    @Mock
    private CommonServerMessageBlock mockMessage;

    private ChannelLoadBalancer loadBalancer;
    private ChannelInfo channel1;

            data = new byte[128];
            Arrays.fill(data, (byte) 0x00);

            // Set up mock messages
            request = mock(CommonServerMessageBlock.class);
            response = mock(CommonServerMessageBlock.class);
        }

        @Test
        @DisplayName("Should sign data correctly with zero signature field")

class NameServiceClientImplTest {

    @Mock
    private CIFSContext mockContext;

    @Mock
    private Configuration mockConfig;

    @Mock
    private NetbiosAddress mockNetbiosAddress;

    @Mock
    private Address mockAddress;

    private NameServiceClientImpl nameServiceClient;

    @BeforeEach
    void setUp() throws UnknownHostException {
        // Configure mock context with fast timeouts

    @BeforeEach
    public void setUp() {
        mockContext = Mockito.mock(CIFSContext.class);
        mockTransport = Mockito.mock(SmbTransportImpl.class);
        mockConfig = Mockito.mock(Configuration.class);
        mockCredentials = Mockito.mock(Credentials.class);
        mockCredentialsInternal = Mockito.mock(CredentialsInternal.class);

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

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

    @BeforeEach
    public void setUp() {
        mockContext = Mockito.mock(CIFSContext.class);
        mockTransport = Mockito.mock(SmbTransportImpl.class);
        mockConfig = Mockito.mock(Configuration.class);
        mockCredentials = Mockito.mock(Credentials.class);
        mockCredentialsInternal = Mockito.mock(CredentialsInternal.class);

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