buffer.size();
            });
        }

        @ParameterizedTest
        @DisplayName("Should handle various referral levels")
        @ValueSource(ints = { 0, 1, 2, 3, 4, 5, 255, 256, 32767, 65535 })
        void testConstructorWithVariousReferralLevels(int maxReferralLevel) {
            String path = "\\\\test";

import com.google.common.base.CharMatcher;
import com.google.common.base.MoreObjects;
import com.google.common.hash.Hashing;
import com.google.common.io.ByteStreams;
import com.google.common.primitives.Ints;
import com.google.errorprone.annotations.CanIgnoreReturnValue;
import java.math.BigInteger;
import java.net.Inet4Address;
import java.net.Inet6Address;
import java.net.InetAddress;
import java.net.NetworkInterface;

    }

    @Test
    @DisplayName("Test hasMoreElements becomes false when all data is sent")
    void testHasMoreElementsBecomeFalse() {
        // Set small amounts so everything fits in one message
        transaction.setParameterBytesWritten(10);
        transaction.setDataBytesWritten(10);

        // First call processes all data
        transaction.nextElement();

import com.google.common.annotations.J2ktIncompatible;
import com.google.common.annotations.VisibleForTesting;
import com.google.common.base.Preconditions;
import com.google.common.primitives.Ints;
import com.google.errorprone.annotations.CanIgnoreReturnValue;
import com.google.errorprone.annotations.concurrent.LazyInit;
import com.google.j2objc.annotations.WeakOuter;
import java.io.IOException;

        assertEquals("DFS resolution failed", ex.getMessage());
        verify(handle, times(1)).ensureDFSResolved();
    }

    @ParameterizedTest
    @ValueSource(ints = { 0, 1, -1, Integer.MIN_VALUE, Integer.MAX_VALUE })
    @DisplayName("hasCapability(cap): returns configured value and captures argument across edge caps")

    void zeroCodeMapsToZero() {
        // Act
        int actual = findNtStatusOrMinusOne(0x00000000);

        // Assert
        assertEquals(0x00000000, actual);
    }

    @ParameterizedTest
    @ValueSource(ints = { -1, 1, 0x00FFFF00, 0x0BADF00D })
    @DisplayName("Invalid or unknown DOS codes return not-found sentinel (-1)")
    void unknownCodesReturnMinusOne(int dosCode) {
        // Act

        // Then
        assertNotNull(exception);
        assertEquals(message, exception.getMessage());
        assertEquals(cause, exception.getCause());
    }

    @ParameterizedTest
    @ValueSource(ints = { NtStatus.NT_STATUS_SUCCESS, NtStatus.NT_STATUS_ACCESS_DENIED, NtStatus.NT_STATUS_OBJECT_NAME_NOT_FOUND,
            NtStatus.NT_STATUS_SHARING_VIOLATION, NtStatus.NT_STATUS_INVALID_PARAMETER })

     *
     * @return byte array containing associated data
     */
    public byte[] getAssociatedData() {
        final byte[] aad = new byte[52]; // Use full header size to ensure all data fits
        int index = 0;

        // Protocol ID
        SMBUtil.writeInt4(TRANSFORM_PROTOCOL_ID, aad, index);
        index += 4;

        // Skip signature (16 bytes of zeros for AAD)
        index += 16;

   * type of the returned array is that of the specified array. If the queue fits in the specified
   * array, it is returned therein. Otherwise, a new array is allocated with the runtime type of the
   * specified array and the size of this queue.
   *
   * <p>If this queue fits in the specified array with room to spare (i.e., the array has more

            // Then
            assertEquals(HEADER_SIZE + CHUNK_SIZE, size);
        }

        @ParameterizedTest
        @DisplayName("Should return correct size with multiple chunks")
        @ValueSource(ints = { 1, 2, 3, 5, 10, 100 })
        void testSizeWithMultipleChunks(int chunkCount) {
            // Given
            byte[] sourceKey = new byte[SOURCE_KEY_SIZE];