@Mock
    private NameServiceClient nameServiceClient;

    @BeforeEach
    void setUp() {
        MockitoAnnotations.openMocks(this);
    }

    @Test
    void testGetLocalHost() {
        // Arrange
        NetbiosAddress expectedAddress = mock(NetbiosAddress.class);
        when(nameServiceClient.getLocalHost()).thenReturn(expectedAddress);

        // Act

        // Arrange
        byte[] dst = new byte[100];
        int dstIndex = 0;

        // Act
        int result = response.writeParametersWireFormat(dst, dstIndex);

        // Assert
        assertEquals(0, result);
    }

    @Test
    @DisplayName("writeParametersWireFormat with offset should return 0")
    void testWriteParametersWireFormatWithOffset() {
        // Arrange

    @Mock
    SmbTreeHandleImpl tree;

    @Mock
    SmbNamedPipe pipe;

    @Mock
    SmbFileHandleImpl fileHandle;

    private SmbPipeOutputStream newStream() throws CIFSException {
        // Arrange common constructor collaborators to avoid touching network/state
        when(handle.getPipe()).thenReturn(pipe);
        when(tree.isSMB2()).thenReturn(true);
        when(tree.getSendBufferSize()).thenReturn(4096);

    void constructor_shouldSetHandleCorrectly() {
        // Arrange
        short level = 1;

        // Act
        queryPolicy = new MsrpcQueryInformationPolicy(mockPolicyHandle, level, mockNdrObject);

        // Assert
        assertSame(mockPolicyHandle, queryPolicy.handle);
    }

    @Test
    void constructor_shouldSetLevelCorrectly() {
        // Arrange
        short level = 7;

        // Act

            // Arrange
            SamrPolicyHandle ph = createMockPolicyHandle();

            // Act & Assert
            assertDoesNotThrow(() -> new MsrpcSamrConnect2(null, 0, ph));
        }

        @Test
        @DisplayName("should accept negative access mask values")
        void testNegativeAccessMaskAccepted() {
            // Arrange

        verify(tree).connectLogon(null);
    }

    @Test
    @DisplayName("connectLogon propagates SmbException from implementation")
    void connectLogon_throws_propagates() throws Exception {
        // Arrange
        doThrow(new SmbException("login failed")).when(tree).connectLogon(context);

        // Act + Assert
        SmbException ex = assertThrows(SmbException.class, () -> tree.connectLogon(context));

    void setUp() {
        // MockitoAnnotations.openMocks(this); // Not needed with @ExtendWith(MockitoExtension.class)
    }

    @Test
    void constructor_shouldOpenPolicySuccessfully() throws IOException {
        // Arrange
        String server = "testServer";
        int access = 123;

        // Mock the behavior of sendrecv for MsrpcLsarOpenPolicy2
        doAnswer(invocation -> {

        when(tree.acquire()).thenReturn(tree);
    }

    @Test
    @DisplayName("Getters return constructor values and acquire is called")
    void gettersAndAcquire() throws Exception {
        // Arrange
        stubAcquireReturnsSelf();
        String wildcard = "*.*";
        int attrs = 123;
        FileEntry initial = entry("first");
        List<FileEntry[]> pages = List.of(new FileEntry[][] { new FileEntry[] {} });

            // Arrange
            Smb2LogoffRequest req = newRequest();

            // Act/Assert
            assertThrows(NullPointerException.class, () -> req.createResponse(null, req));
        }

        @Test
        @DisplayName("createResponse tolerates null request parameter")
        void createResponse_nullRequest_ok() {
            // Arrange

        closeHandleCaptor = ArgumentCaptor.forClass(MsrpcSamrCloseHandle.class);
    }

    @Test
    void constructor_shouldOpenDomainSuccessfully() throws IOException {
        // Arrange
        int access = 0x01; // Example access value
        // Simulate successful RPC call
        doAnswer(invocation -> {
            MsrpcSamrOpenDomain rpc = invocation.getArgument(0);