SmbResourceLocator context2 = mock(SmbResourceLocator.class);
        when(context2.getShare()).thenReturn("share");
        when(context2.getDfsReferral()).thenReturn(null);
        when(context2.getUNCPath()).thenReturn("\\");
        when(context2.getURLPath()).thenReturn("/share/");
        when(context2.getServer()).thenReturn("server");

        base2.resolveInContext(context2, "sub/child");

        expect:
        type.rawType.signature == 'org.gradle.SomeType[][]'
    }

    def rawTypeForVarargsType() {
        when:
        type.setVarargs()

        then:
        type.rawType.signature == 'org.gradle.SomeType[]'

        when:
        type.arrayDimensions = 1

        then:
        type.rawType.signature == 'org.gradle.SomeType[][]'
    }

    def rawTypeForParameterizedArrayType() {

            int result = 0;
            when(mockBuffer.dec_ndr_short()).thenReturn(4096, 4096, 10, result, 0);
            when(mockBuffer.dec_ndr_long()).thenReturn(12345);
            when(mockBuffer.dec_ndr_small()).thenReturn(1);
            lenient().doNothing().when(mockBuffer).advance(anyInt());
            lenient().when(mockBuffer.align(anyInt())).thenReturn(0);

            // When
            bind.decode_out(mockBuffer);

        lenient().when(request.getSession()).thenReturn(session);
        lenient().when(request.getSession(false)).thenReturn(session);
        lenient().when(request.getSession(true)).thenReturn(session);

        // Setup default session attribute behavior
        lenient().when(session.getAttribute(anyString())).thenReturn(null);

        // Setup SmbFile mock
        lenient().when(smbFile.getLocator()).thenReturn(locator);

            return null;
        }).when(mockDcerpcHandle).sendrecv(any(MsrpcSamrOpenDomain.class));

        SamrDomainHandle handle = new SamrDomainHandle(mockDcerpcHandle, mockPolicyHandle, access, mockSid);

        IOException expectedException = new IOException("Network error during close");
        // Simulate IOException during close RPC call

        when(mockSearchResponse.getScrollId()).thenReturn("scroll1", "scroll1");
        when(mockSearchResponse.getHits()).thenReturn(mockSearchHits);
        when(mockSearchHits.getHits()).thenReturn(new SearchHit[] { mockHit }, new SearchHit[0]);
        when(mockHit.getIndex()).thenReturn("test-index");
        when(mockHit.getId()).thenReturn("doc1");

        when(readResponse.getFlags2()).thenReturn(SmbConstants.FLAGS2_SECURITY_SIGNATURES);
        when(readResponse.getSignSeq()).thenReturn(10);
        when(readResponse.getCommand()).thenReturn((int) ServerMessageBlock.SMB_COM_READ_ANDX);
        when(readResponse.getLength()).thenReturn(100);
        when(readResponse.getDataLength()).thenReturn(50);
        when(readResponse.getData()).thenReturn(readData);

        // Set up mock behaviors
        when(mockContext.getConfig()).thenReturn(mockConfig);
        when(mockContext.getCredentials()).thenReturn(mockCredentials);
        when(mockContext.getTransportPool()).thenReturn(mockTransportPool);
        when(mockConfig.getDfsTtl()).thenReturn(300L);

        // Mock transport to throw IOException - simulating connection failure

        when(file.ensureTreeConnected()).thenReturn(tree);
        when(tree.getConfig()).thenReturn(cfg);
        when(tree.getReceiveBufferSize()).thenReturn(1024);
        when(tree.getSendBufferSize()).thenReturn(1024);
        when(tree.hasCapability(anyInt())).thenAnswer(inv -> ntSmbsCap && inv.getArgument(0).equals(SmbConstants.CAP_NT_SMBS));
        when(tree.areSignaturesActive()).thenReturn(false);

        byte[] dst = new byte[42];

        // When
        int result = trans2SetFileInformation.writeDataWireFormat(dst, 0);

        // Then
        assertEquals(40, result); // Returns 40 bytes written
    }

    @Test
    void testReadSetupWireFormat() {
        // When
        int result = trans2SetFileInformation.readSetupWireFormat(new byte[0], 0, 0);

        // Then
        assertEquals(0, result);
    }