/**
     * Tests the writeSetupWireFormat method.
     */
    @Test
    void testWriteSetupWireFormat() {
        // Given
        Trans2QueryPathInformation trans = new Trans2QueryPathInformation("test.txt", 0x0101);
        byte[] dst = new byte[2];

        // When
        int bytesWritten = trans.writeSetupWireFormat(dst, 0);

        // Then
        assertEquals(2, bytesWritten, "Should write 2 bytes.");

        // Name pointer (0xC00C)
        src[srcIndex] = (byte) 0xC0;
        src[srcIndex + 1] = 0x0C;
        // Record type (NBSTAT = 0x0021)
        src[srcIndex + 2] = 0x00;
        src[srcIndex + 3] = 0x21;
        // Record class (IN = 0x0001)
        src[srcIndex + 4] = 0x00;
        src[srcIndex + 5] = 0x01;
        // TTL (4 bytes)
        src[srcIndex + 6] = 0x00;
        src[srcIndex + 7] = 0x00;
        src[srcIndex + 8] = 0x00;

        // Then
        assertEquals(messageSize, transformHeader.getOriginalMessageSize());
    }

    @Test
    @DisplayName("Should set and get flags")
    void testFlags() {
        // Given
        int flags = 0x0001; // Encrypted flag

        // When
        transformHeader.setFlags(flags);

        // Then
        assertEquals(flags, transformHeader.getFlags());
    }

    @Test

    class ToStringTests {

        @Test
        @DisplayName("toString with valid data produces expected format")
        void testToStringWithValidData() {
            byte[] challenge = new byte[] { (byte) 0x01, (byte) 0x02, (byte) 0x03, (byte) 0xFF };
            UniAddress dc = mock(UniAddress.class);
            when(dc.toString()).thenReturn("SERVER123");

            NtlmChallenge nc = new NtlmChallenge(challenge, dc);

        NtTransQuerySecurityDesc desc2 = new NtTransQuerySecurityDesc(mockConfig, 0x2222, GROUP_SECURITY_INFORMATION);
        NtTransQuerySecurityDesc desc3 = new NtTransQuerySecurityDesc(mockConfig, 0x3333, DACL_SECURITY_INFORMATION);

        // Verify each instance maintains its own state
        assertEquals(0x1111, desc1.fid);

    private LeaseV1CreateContextRequest leaseContext;
    private Smb2LeaseKey testKey;
    private int testState;

    @BeforeEach
    void setUp() {
        byte[] keyBytes = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06, 0x07, 0x08, 0x09, 0x0A, 0x0B, 0x0C, 0x0D, 0x0E, 0x0F, 0x10 };
        testKey = new Smb2LeaseKey(keyBytes);
        testState = Smb2LeaseState.SMB2_LEASE_READ_WRITE;

	version(3.0)
]
interface netdfs
{
	import "../rpc.idl";

	const uint32_t DFS_VOLUME_FLAVOR_STANDALONE = 0x100;
	const uint32_t DFS_VOLUME_FLAVOR_AD_BLOB = 0x200;

	const uint32_t DFS_STORAGE_STATE_OFFLINE = 0x0001;
	const uint32_t DFS_STORAGE_STATE_ONLINE = 0x0002;
	const uint32_t DFS_STORAGE_STATE_ACTIVE = 0x0004;

	typedef struct {
		[string] wchar_t *entry_path;
	} DfsInfo1;

	typedef struct {
		uint32_t count;

        // Check FID
        int actualFid = SMBUtil.readInt2(buffer, 0);
        assertEquals(TEST_FID, actualFid);

        // Check information level (FILE_BASIC_INFO maps to 0x0101)
        int actualInfoLevel = SMBUtil.readInt2(buffer, 2);
        assertEquals(0x0101, actualInfoLevel);

        // Check reserved bytes (should be 0)
        int reserved = SMBUtil.readInt2(buffer, 4);
        assertEquals(0, reserved);
    }

        return transformCount > 0 && rdmaTransformId == 0x0001;
    }

    /**
     * Get the transform count
     *
     * @return number of transforms
     */
    public int getTransformCount() {
        return transformCount;
    }

    /**
     * Get the RDMA transform ID
     *
     * @return transform ID (should be 0x0001 for V1)
     */
    public int getRdmaTransformId() {

        // Test successful reading of 6 bytes from the input stream.
        // The content of these bytes is not critical for this test, only that 6 bytes are read.
        byte[] data = { 0x01, 0x02, 0x03, 0x04, 0x05, 0x06 }; // Sample 6 bytes
        ByteArrayInputStream bais = new ByteArrayInputStream(data);
        SessionRetargetResponsePacket packet = new SessionRetargetResponsePacket();