SMBUtil.writeInt4(32, dst, dstIndex); // DataLength
        dstIndex += 4;

        // Write context name
        System.arraycopy(CONTEXT_NAME_BYTES, 0, dst, dstIndex, 4);
        dstIndex += 4;

        // Padding to align data to 8-byte boundary
        dstIndex += 4;

        // Write lease V1 data (32 bytes total)
        leaseKey.encode(dst, dstIndex); // LeaseKey (16 bytes)
        dstIndex += 16;

	return &pooledFlateWriter{fw: fw}
}

type pooledFlateWriter struct {
	mu sync.Mutex // guards Close and Write
	fw *flate.Writer
}

func (w *pooledFlateWriter) Write(p []byte) (n int, err error) {
	w.mu.Lock()
	defer w.mu.Unlock()
	if w.fw == nil {
		return 0, errors.New("Write after Close")
	}
	return w.fw.Write(p)
}

func (w *pooledFlateWriter) Close() error {
	w.mu.Lock()
	defer w.mu.Unlock()
	var err error

// If nn < len(p), it also returns an error explaining
// why the write is short.
func (b *Writer) Write(p []byte) (nn int, err error) {
	for len(p) > b.Available() && b.err == nil {
		var n int
		if b.Buffered() == 0 {
			// Large write, empty buffer.
			// Write directly from p to avoid copy.
			n, b.err = b.wr.Write(p)
		} else {
			n = copy(b.buf[b.n:], p)
			b.n += n
			b.Flush()
		}
		nn += n

            // Read the response
            int bytesRead = echoResponse.readBytesWireFormat(buffer, bufferIndex);
            assertEquals(0, bytesRead);

            // Write response (echo responses don't write data)
            int bytesWritten = echoResponse.writeBytesWireFormat(buffer, 0);
            assertEquals(0, bytesWritten);

            // Mark as received
            echoResponse.received();

        
        writeInt4(buffer, offset + 12, dataOffset - offset);  // DataOffset
        
        // Write context name
        System.arraycopy(getName().getBytes(), 0, buffer, offset + 16, getName().length());
        
        // Write lease data
        System.arraycopy(leaseKey.getKey(), 0, buffer, dataOffset, 16);  // LeaseKey

        }

        @Test
        @DisplayName("Should handle write at buffer boundary")
        void testWriteAtBufferBoundary() {
            request.setReadLength(1024);

            byte[] buffer = new byte[100];

            // Try to write at position that would exceed buffer

        // Cannot test getConfig() as it's protected
    }

    @Nested
    @DisplayName("writeBytesWireFormat tests")
    class WriteBytesWireFormatTests {

        @Test
        @DisplayName("Should return 0 for write bytes")
        void testWriteBytesWireFormat() {
            // Given
            byte[] dst = new byte[100];
            int dstIndex = 0;

            // When

            DataOutputStream dos = new DataOutputStream(baos);

            // Write minimal PAC header
            dos.writeInt(Integer.reverseBytes(3)); // 3 buffers
            dos.writeInt(Integer.reverseBytes(PacConstants.PAC_VERSION));

            // Write buffer entries
            dos.writeInt(Integer.reverseBytes(PacConstants.LOGON_INFO));

            throw e;
        }
    }

    /**
     * Performs RDMA write operation.
     *
     * @param buffer the buffer to write from
     * @param remoteAddress remote memory address
     * @param remoteKey remote access key
     * @param length number of bytes to write
     * @return number of bytes written
     * @throws IOException if operation fails
     */

func TestDeadlineWriter(t *testing.T) {
	w := NewDeadlineWriter(&sleepWriter{timeout: 500 * time.Millisecond}, 450*time.Millisecond)
	_, err := w.Write([]byte("1"))
	if err != context.DeadlineExceeded {
		t.Error("DeadlineWriter shouldn't be successful - should return context.DeadlineExceeded")
	}
	_, err = w.Write([]byte("1"))
	if err != context.DeadlineExceeded {
		t.Error("DeadlineWriter shouldn't be successful - should return context.DeadlineExceeded")