void testWithDifferentBufferSizes() {
        // Test with buffers of different sizes
        byte[] smallBuffer = new byte[] { 0x01 };
        byte[] mediumBuffer = new byte[] { 0x01, 0x02, 0x03, 0x04, 0x05 };

        SmbNegotiation negotiationDifferentSizes = new SmbNegotiation(mockRequest, mockResponse, smallBuffer, mediumBuffer);

        assertEquals(1, negotiationDifferentSizes.getRequestRaw().length);

            EncryptionNegotiateContext context = new EncryptionNegotiateContext(mockConfig, new int[] { 1, 2 });
            byte[] smallBuffer = new byte[100]; // Make buffer large enough
            byte[] largeBuffer = new byte[1000];

            // Act
            int smallEncodedSize = context.encode(smallBuffer, 0);
            int largeEncodedSize = context.encode(largeBuffer, 500);

            // Assert

        // Given
        trans2GetDfsReferral = new Trans2GetDfsReferral(mockConfig, "test");

        // Test with minimum size buffer
        byte[] smallBuffer = new byte[2];
        assertEquals(2, trans2GetDfsReferral.writeSetupWireFormat(smallBuffer, 0));

        // Test with exact size buffer
        byte[] exactBuffer = new byte[2];
        assertEquals(2, trans2GetDfsReferral.writeSetupWireFormat(exactBuffer, 0));

        void testBufferBoundaries() {
            byte[] smallBuffer = new byte[10];
            int[] ciphers = { 1, 2 };

            EncryptionNegotiateContext context = new EncryptionNegotiateContext(mockConfig, ciphers);

            // Should encode successfully at the beginning
            int encoded = context.encode(smallBuffer, 0);
            assertEquals(6, encoded);

        // Create buffer smaller than required size
        byte[] smallBuffer = new byte[request.size() - 1];

        assertThrows(ArrayIndexOutOfBoundsException.class, () -> {
            request.encode(smallBuffer, 0);
        });
    }

    @Test
    @DisplayName("Test encode with offset exceeding buffer throws exception")

        void testBufferOverflowProtection() {
            request.setReadLength(65536);

            byte[] smallBuffer = new byte[48]; // Smaller than required 49 bytes

            // Should not overflow buffer
            assertThrows(ArrayIndexOutOfBoundsException.class, () -> request.writeBytesWireFormat(smallBuffer, 0));
        }

        @Test
        @DisplayName("Should handle write at buffer boundary")

        Smb2TreeConnectRequest req = new Smb2TreeConnectRequest(mockConfig, longPath);
        byte[] smallBuffer = new byte[20]; // Too small for header + path

        // When & Then
        assertThrows(ArrayIndexOutOfBoundsException.class, () -> {
            req.encode(smallBuffer, 0);
        });
    }

    @Test
    @DisplayName("Should correctly inherit from ServerMessageBlock2Request")

    @DisplayName("Test buffer overflow protection")
    void testBufferOverflowProtection() {
        byte[] smallBuffer = new byte[23]; // One byte too small

        // Should throw ArrayIndexOutOfBoundsException
        assertThrows(ArrayIndexOutOfBoundsException.class, () -> {
            request.writeBytesWireFormat(smallBuffer, 0);
        });
    }

    @Test
    @DisplayName("Test file ID boundary values")

            byte[] largeData = new byte[1000];
            request.setData(largeData, 0, largeData.length);

            byte[] smallBuffer = new byte[100];

            IllegalArgumentException exception =
                    assertThrows(IllegalArgumentException.class, () -> request.writeBytesWireFormat(smallBuffer, 0));

            assertTrue(exception.getMessage().contains("Data exceeds buffer size"));
        }

        @Test

            Smb2LockRequest req = new Smb2LockRequest(mockConfig, testFileId, locks);

            byte[] smallBuffer = new byte[50]; // Smaller than required

            // Should not overflow buffer
            assertThrows(ArrayIndexOutOfBoundsException.class, () -> req.writeBytesWireFormat(smallBuffer, 0));
        }

        @Test
        @DisplayName("Should handle write at buffer boundary")